test/prototype/mx_formats/test_kernels.py

zxd1997066 · zxd1997066 · commit 8d57626d7ad0 · 2025-12-01T16:57:58.000+08:00
diff --git a/test/prototype/mx_formats/test_kernels.py b/test/prototype/mx_formats/test_kernels.py
@@ -561,7 +561,7 @@ def test_rearrange(shape):
 
 
 @pytest.mark.skipif(
-    torch.cuda.is_available() and not is_sm_at_least_100(),
+    not is_sm_at_least_100(),
     reason="MXFP8 requires CUDA capability 10.0 or greater",
 )
 @pytest.mark.parametrize("M", (32, 64, 2048))
@@ -609,7 +609,7 @@ def test_cuda_mx_dim1_numerics(M, K, input_dtype, scaling_mode):
 
 
 @pytest.mark.skipif(
-    torch.cuda.is_available() and not is_sm_at_least_100(),
+    not is_sm_at_least_100(),
     reason="MXFP8 requires CUDA capability 10.0 or greater",
 )
 def test_cuda_mx_dim0_not_supported():
@@ -633,7 +633,7 @@ def test_cuda_mx_dim0_not_supported():
 
 
 @pytest.mark.skipif(
-    torch.cuda.is_available() and not is_sm_at_least_100(),
+    not is_sm_at_least_100(),
     reason="MXFP8 requires CUDA capability 10.0 or greater",
 )
 def test_cuda_mx_dim1_invalid_block_size():

Original file line number	Diff line number	Diff line change
`@@ -561,7 +561,7 @@ def test_rearrange(shape):`
`561`	`561`
`562`	`562`
`563`	`563`	`@pytest.mark.skipif(`
`564`		`- torch.cuda.is_available() and not is_sm_at_least_100(),`
	`564`	`+ not is_sm_at_least_100(),`
`565`	`565`	`reason="MXFP8 requires CUDA capability 10.0 or greater",`
`566`	`566`	`)`
`567`	`567`	`@pytest.mark.parametrize("M", (32, 64, 2048))`
`@@ -609,7 +609,7 @@ def test_cuda_mx_dim1_numerics(M, K, input_dtype, scaling_mode):`
`609`	`609`
`610`	`610`
`611`	`611`	`@pytest.mark.skipif(`
`612`		`- torch.cuda.is_available() and not is_sm_at_least_100(),`
	`612`	`+ not is_sm_at_least_100(),`
`613`	`613`	`reason="MXFP8 requires CUDA capability 10.0 or greater",`
`614`	`614`	`)`
`615`	`615`	`def test_cuda_mx_dim0_not_supported():`
`@@ -633,7 +633,7 @@ def test_cuda_mx_dim0_not_supported():`
`633`	`633`
`634`	`634`
`635`	`635`	`@pytest.mark.skipif(`
`636`		`- torch.cuda.is_available() and not is_sm_at_least_100(),`
	`636`	`+ not is_sm_at_least_100(),`
`637`	`637`	`reason="MXFP8 requires CUDA capability 10.0 or greater",`
`638`	`638`	`)`
`639`	`639`	`def test_cuda_mx_dim1_invalid_block_size():`