Update

[ghstack-poisoned]
pytorch · Mar 5, 2025 · 8f82ae7 · 8f82ae7
1 parent ee76770
commit 8f82ae7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/prototype/mx_formats/test_mx_linear.py b/test/prototype/mx_formats/test_mx_linear.py
@@ -186,6 +186,10 @@ def test_linear_compile(recipe_name, bias):
         if not is_sm_at_least_89():
             pytest.skip("CUDA capability >= 8.9 required for float8 in triton")
 
+    if recipe_name in ["mxfp8_cutlass", "mxfp4_cutlass"]:
+        if not is_sm_at_least_100():
+            pytest.skip("CUDA capability >= 10.0 required for MX gemms")
+
     if bias and recipe_name in ["mxfp8_cutlass", "mxfp4_cutlass"]:
         # TODO(future PR): fix this, things are clearly broken with bias=True
         pytest.skip("this test is broken for cutlass recipes with bias=True")