File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 1717from torch ._inductor .test_operators import realize
1818from torch ._inductor .utils import sympy_index_symbol
1919from 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
2121from torch .testing ._internal .inductor_utils import HAS_CUDA
2222from torch .utils ._pytree import tree_map
2323from 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
You can’t perform that action at this time.
0 commit comments