frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Inductor] Fix cooperative reduction tests broken in recent refactor (#143135)

Browse Source 
These tests were broken by https://github.com/pytorch/pytorch/pull/142020. This PR updates the fixed configs accordingly.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143135
Approved by: https://github.com/jansel, https://github.com/huydhn
This commit is contained in:
Blaine Burton Rister
2024-12-13 02:03:43 +00:00
committed by PyTorch MergeBot
parent 9f5ebf3fc6
commit dbe4b69df0
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/inductor/test_cooperative_reductions.py
View File

@ -146,12 +146,12 @@ class TestFixedConfigs(TestCase):
@parametrize(
"persistent,cooperative,cfg",
[
(False, False, {"XBLOCK": 1, "RBLOCK": 128}),
(False, False, {"XBLOCK": 2, "RBLOCK": 128}),
(False, False, {"XBLOCK": 1, "R0_BLOCK": 128}),
(False, False, {"XBLOCK": 2, "R0_BLOCK": 128}),
(True, False, {"XBLOCK": 1}),
(True, False, {"XBLOCK": 2}),
(False, True, {"XBLOCK": 1, "RBLOCK": 128, "RSPLIT": 16}),
(False, True, {"XBLOCK": 2, "RBLOCK": 128, "RSPLIT": 16}),
(False, True, {"XBLOCK": 1, "R0_BLOCK": 128, "RSPLIT": 16}),
(False, True, {"XBLOCK": 2, "R0_BLOCK": 128, "RSPLIT": 16}),
(True, True, {"XBLOCK": 1, "RSPLIT": 16}),
(True, True, {"XBLOCK": 2, "RSPLIT": 16}),
],