Skip to content

Commit 6b05842

Browse files
davidberard98pytorchmergebot
authored andcommitted
[test][inductor] fix test_conv_cat failure (pytorch#155852)
This test is currently failing because triton_poi_fused_cat_2 has changed to triton_poi_fused_cat_3. I have not investigated why the extra kernel is generated, but this test has been failing on trunk for a while (and I verified locally that it is failing). Pull Request resolved: pytorch#155852 Approved by: https://github.com/FindHao, https://github.com/Skylion007
1 parent 2c76f31 commit 6b05842

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/inductor/test_max_autotune.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -783,7 +783,7 @@ def forward(self, x):
783783
self.assertEqual(out, m(input_tensor))
784784

785785
if not TEST_WITH_ROCM:
786-
FileCheck().check("triton_poi_fused_cat_2.run").run(code[0])
786+
FileCheck().check("def triton_poi_fused_cat_").run(code[0])
787787

788788
def test_conv3d(self):
789789
fn = torch.nn.functional.conv3d

0 commit comments

Comments
 (0)