Skip to content

Commit 6cdc70b

Browse files
praguptapytorchmergebot
authored andcommitted
[ROCm] skip test_fp8_cast_and_t on non-MI300 machines (pytorch#135917)
Fixes #ISSUE_NUMBER Pull Request resolved: pytorch#135917 Approved by: https://github.com/malfet
1 parent e6b6835 commit 6cdc70b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/inductor/test_loop_ordering.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from torch._inductor.test_operators import realize
1818
from torch._inductor.utils import sympy_index_symbol
1919
from torch._inductor.virtualized import ops, V
20-
from torch.testing._internal.common_cuda import SM90OrLater
20+
from torch.testing._internal.common_cuda import PLATFORM_SUPPORTS_FP8
2121
from torch.testing._internal.inductor_utils import HAS_CUDA
2222
from torch.utils._pytree import tree_map
2323
from torch.utils._sympy.functions import ModularIndexing
@@ -371,7 +371,7 @@ def f(x):
371371
self.do_acc_test(f, x)
372372
self.assertEqual(1, metrics.generated_kernel_count)
373373

374-
@unittest.skipIf(not SM90OrLater, "FP8 requires H100+")
374+
@unittest.skipIf(not PLATFORM_SUPPORTS_FP8, "FP8 requires H100+ and MI300+")
375375
def test_fp8_cast_and_t(self):
376376
"""
377377
This test repros the not able to fuses issue in

0 commit comments

Comments
 (0)