From afc94523c965a61bd6869287c798b2a45f2c205e Mon Sep 17 00:00:00 2001 From: liuzhenwei Date: Mon, 27 Jul 2026 15:28:02 +0800 Subject: [PATCH] [XPU][CI] Use platform device in InputBatch V2 test (#49939) Signed-off-by: zhenwei-intel --- tests/v1/worker/test_gpu_input_batch_v2.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/v1/worker/test_gpu_input_batch_v2.py b/tests/v1/worker/test_gpu_input_batch_v2.py index 2c43cd72147..f517abb2b14 100644 --- a/tests/v1/worker/test_gpu_input_batch_v2.py +++ b/tests/v1/worker/test_gpu_input_batch_v2.py @@ -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(