Skip failing UT

pytorch · Mar 27, 2024 · f882600 · f882600
1 parent 65996d3
commit f882600
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/inductor/test_triton_heuristics.py b/test/inductor/test_triton_heuristics.py
@@ -5,7 +5,7 @@
 
 import torch
 
-from torch.testing._internal.common_utils import IS_LINUX
+from torch.testing._internal.common_utils import IS_LINUX, skipIfRocm
 from torch.testing._internal.inductor_utils import HAS_GPU
 
 try:
@@ -71,6 +71,7 @@ def forward(primals_1, primals_2, primals_5):
         ]
         self.assertEqual(forward(*args), foo_c(*args))
 
+    @skipIfRocm
     def test_artificial_zgrid(self):
         self._test_artificial_zgrid()