test/prototype/mx_formats/test_kernels.py

zxd1997066 · zxd1997066 · commit 360945ca5a47 · 2025-12-02T09:49:31.000+08:00
diff --git a/test/prototype/mx_formats/test_kernels.py b/test/prototype/mx_formats/test_kernels.py
@@ -528,7 +528,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, 256))
@@ -576,7 +576,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():
@@ -600,7 +600,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
`@@ -528,7 +528,7 @@ def test_rearrange(shape):`
`528`	`528`
`529`	`529`
`530`	`530`	`@pytest.mark.skipif(`
`531`		`- torch.cuda.is_available() and not is_sm_at_least_100(),`
	`531`	`+ not is_sm_at_least_100(),`
`532`	`532`	`reason="MXFP8 requires CUDA capability 10.0 or greater",`
`533`	`533`	`)`
`534`	`534`	`@pytest.mark.parametrize("M", (32, 256))`
`@@ -576,7 +576,7 @@ def test_cuda_mx_dim1_numerics(M, K, input_dtype, scaling_mode):`
`576`	`576`
`577`	`577`
`578`	`578`	`@pytest.mark.skipif(`
`579`		`- torch.cuda.is_available() and not is_sm_at_least_100(),`
	`579`	`+ not is_sm_at_least_100(),`
`580`	`580`	`reason="MXFP8 requires CUDA capability 10.0 or greater",`
`581`	`581`	`)`
`582`	`582`	`def test_cuda_mx_dim0_not_supported():`
`@@ -600,7 +600,7 @@ def test_cuda_mx_dim0_not_supported():`
`600`	`600`
`601`	`601`
`602`	`602`	`@pytest.mark.skipif(`
`603`		`- torch.cuda.is_available() and not is_sm_at_least_100(),`
	`603`	`+ not is_sm_at_least_100(),`
`604`	`604`	`reason="MXFP8 requires CUDA capability 10.0 or greater",`
`605`	`605`	`)`
`606`	`606`	`def test_cuda_mx_dim1_invalid_block_size():`