[XPU][CI] enable prompt embeds tests on XPU (#58283)
Signed-off-by: Lin, Fanli <fanli.lin@intel.com> Signed-off-by: Fanli Lin <fanli.lin@intel.com> Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
Fanli Lin committed
Sep 25, 2026 at 06:56 UTC
7edb27f68e23410e73d74066e4e3ce9ba27d4421
1 file changed
+7
-3
tests/v1/worker/test_prompt_embeds_state.py
+7
-3
@@ -15,16 +15,20 @@ import pytest
15
import torch
16
17
pytest.importorskip("triton")
18
-if not torch.cuda.is_available():
18
+
19
+from vllm.platforms import current_platform
20
+
21
+if not (current_platform.is_cuda_alike() or current_platform.is_xpu()):
22
pytest.skip(
20
- "CUDA required for prompt-embeds overlay tests", allow_module_level=True
23
+ "An accelerator is required for prompt-embeds overlay tests",
24
+ allow_module_level=True,
25
)
26
27
from vllm.v1.worker.gpu.model_states.prompt_embeds import PromptEmbedsState
28
29
HIDDEN = 24
30
MAX_NUM_REQS = 8
27
-DEVICE = torch.device("cuda")
31
+DEVICE = torch.device(current_platform.device_type)
32
33
34
@dataclass