[XPU][CI] Use platform device in InputBatch V2 test (#49939)

Signed-off-by: zhenwei-intel <zhenwei.liu@intel.com>
2026-07-27 15:28:02 +08:00
parent 8061dc26bd
commit afc94523c9
1 changed files with 2 additions and 1 deletions
@@ -5,9 +5,10 @@
 import pytest
 import torch

+from vllm.platforms import current_platform
 from vllm.v1.worker.gpu.input_batch import InputBatch, InputBuffers

-DEVICE = "cuda"
+DEVICE = current_platform.device_type


@pytest.mark.parametrize(