Skip to content

Commit dbe4b69

Browse files
blaine-risterpytorchmergebot
authored andcommitted
[Inductor] Fix cooperative reduction tests broken in recent refactor (pytorch#143135)
These tests were broken by pytorch#142020. This PR updates the fixed configs accordingly. Pull Request resolved: pytorch#143135 Approved by: https://github.com/jansel, https://github.com/huydhn
1 parent 9f5ebf3 commit dbe4b69

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/inductor/test_cooperative_reductions.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -146,12 +146,12 @@ class TestFixedConfigs(TestCase):
146146
@parametrize(
147147
"persistent,cooperative,cfg",
148148
[
149-
(False, False, {"XBLOCK": 1, "RBLOCK": 128}),
150-
(False, False, {"XBLOCK": 2, "RBLOCK": 128}),
149+
(False, False, {"XBLOCK": 1, "R0_BLOCK": 128}),
150+
(False, False, {"XBLOCK": 2, "R0_BLOCK": 128}),
151151
(True, False, {"XBLOCK": 1}),
152152
(True, False, {"XBLOCK": 2}),
153-
(False, True, {"XBLOCK": 1, "RBLOCK": 128, "RSPLIT": 16}),
154-
(False, True, {"XBLOCK": 2, "RBLOCK": 128, "RSPLIT": 16}),
153+
(False, True, {"XBLOCK": 1, "R0_BLOCK": 128, "RSPLIT": 16}),
154+
(False, True, {"XBLOCK": 2, "R0_BLOCK": 128, "RSPLIT": 16}),
155155
(True, True, {"XBLOCK": 1, "RSPLIT": 16}),
156156
(True, True, {"XBLOCK": 2, "RSPLIT": 16}),
157157
],

0 commit comments

Comments
 (0)