Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 465a23b

Browse files
committedApr 3, 2025·
Add test checks
1 parent 1c24eb0 commit 465a23b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed
 

‎tests/kernels/test_cutlass_mla_decode.py

+5
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,11 @@
55
from torch import Tensor
66

77
import vllm._custom_ops as ops
8+
from vllm.platforms import current_platform
9+
10+
if not current_platform.has_device_capability(100):
11+
pytest.skip(reason="Cutlass MLA Requires compute capability of 10 or above.",
12+
allow_module_level=True)
813

914
def ref_mla(
1015
out: Tensor, # (bs, num_heads, v_head_dim)

0 commit comments

Comments
 (0)
Please sign in to comment.