[XPU][CI] Use platform device in InputBatch V2 test (#49939)
Signed-off-by: zhenwei-intel <zhenwei.liu@intel.com>
liuzhenwei committed
Jul 27, 2026 at 15:28 UTC
afc94523c965a61bd6869287c798b2a45f2c205e
1 file changed
+2
-1
tests/v1/worker/test_gpu_input_batch_v2.py
+2
-1
@@ -5,9 +5,10 @@
5
import pytest
6
import torch
7
8
+from vllm.platforms import current_platform
9
from vllm.v1.worker.gpu.input_batch import InputBatch, InputBuffers
10
10
-DEVICE = "cuda"
11
+DEVICE = current_platform.device_type
12
13
14
@pytest.mark.parametrize(