[Bugfix] Gate sm_100-only kernel tests on the capability family, not >= (#54306)
Signed-off-by: bojiang3 <bli314159@gmail.com>
Bojiang Li committed
Sep 1, 2026 at 08:29 UTC
a232e29e9d717562cf9cf7a8d086d52a57056615
5 files changed
+6
-6
tests/kernels/attention/test_cutlass_mla_decode.py
+2
-2
@@ -42,7 +42,7 @@ CUTLASS_MLA_UNSUPPORTED_REASON = (
42
43
44
@pytest.mark.skipif(
45
- not current_platform.has_device_capability(100),
45
+ not current_platform.is_device_capability_family(100),
46
reason=CUTLASS_MLA_UNSUPPORTED_REASON,
47
)
48
@pytest.mark.parametrize("b", [128])
@@ -217,7 +217,7 @@ def test_cutlass_mla_decode(
217
218
219
@pytest.mark.skipif(
220
- not current_platform.has_device_capability(100),
220
+ not current_platform.is_device_capability_family(100),
221
reason=CUTLASS_MLA_UNSUPPORTED_REASON,
222
)
223
@torch.inference_mode()
tests/kernels/moe/test_cutedsl_moe.py
+1
-1
@@ -6,7 +6,7 @@ import pytest
6
7
from vllm.platforms import current_platform
8
9
-if not current_platform.has_device_capability(100):
9
+if not current_platform.is_device_capability_family(100):
10
pytest.skip(
11
reason="Nvfp4 Requires compute capability of 10 or above.",
12
allow_module_level=True,
tests/kernels/moe/test_routed_experts_capture_monolithic.py
+1
-1
@@ -51,7 +51,7 @@ if not has_flashinfer_trtllm_fused_moe() or not current_platform.is_cuda():
51
allow_module_level=True,
52
)
53
54
-if not current_platform.has_device_capability(100):
54
+if not current_platform.is_device_capability_family(100):
55
pytest.skip(
56
"TRT-LLM fused MoE kernels require SM100+",
57
allow_module_level=True,
tests/kernels/moe/test_trtllm_bf16_moe.py
+1
-1
@@ -39,7 +39,7 @@ from vllm.utils.torch_utils import set_random_seed
39
40
if pytest and (
41
not has_flashinfer_trtllm_fused_moe()
42
- or not current_platform.has_device_capability(100)
42
+ or not current_platform.is_device_capability_family(100)
43
):
44
pytest.skip(
45
"Requires flashinfer TRTLLM fused MoE BF16 backend (SM100)",
tests/kernels/moe/test_trtllm_nvfp4_moe.py
+1
-1
@@ -44,7 +44,7 @@ from vllm.utils.torch_utils import set_random_seed
44
45
if pytest and (
46
not has_flashinfer_trtllm_fused_moe()
47
- or not current_platform.has_device_capability(100)
47
+ or not current_platform.is_device_capability_family(100)
48
):
49
pytest.skip(
50
"Requires flashinfer TRTLLM fused MoE and NvFP4 (SM100)",