[Test] Skip ROCm AITER MLA prefill tests on non-ROCm platforms (#49945)
Signed-off-by: Liangliang-Ma <liangliang.ma@intel.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Liangliang Ma committed
Jul 28, 2026 at 16:26 UTC
88402a41c4ab272ebbbd33f4a77fbbac0431cbb9
1 file changed
+6
tests/v1/attention/test_mla_prefill_selector.py
+6
@@ -8,6 +8,7 @@ import pytest
8
import torch
9
10
from vllm.config import AttentionConfig, ModelConfig, VllmConfig
11
+from vllm.platforms import current_platform
12
from vllm.platforms.interface import DeviceCapability
13
from vllm.v1.attention.backends.mla.prefill.base import MLADimensions
14
from vllm.v1.attention.backends.mla.prefill.registry import MLAPrefillBackendEnum
@@ -287,6 +288,11 @@ class TestBackendValidation:
288
assert invalid_reasons == []
289
290
291
+@pytest.mark.skipif(
292
+ not current_platform.is_cuda_alike(),
293
+ reason="Imports vllm.platforms.rocm, whose module init requires a CUDA or "
294
+ "ROCm torch build; not importable on XPU/CPU/TPU.",
295
+)
296
class TestROCmAiterFAPrefillSelection:
297
"""Tests for the ROCm AITER FlashAttention MLA prefill backend."""
298