[XPU][CI] Enable shared loader test (#45977)
Signed-off-by: Chaojun Zhang <chaojun.zhang@intel.com>
Chaojun Zhang committed
Jun 30, 2026 at 19:20 UTC
ea9ddf59fc9d262da7467699959d8c84600c073c
2 files changed
+28
-1
.buildkite/intel_jobs/models_distributed_intel.yaml
new
+27
@@ -0,0 +1,27 @@
1
+group: Models - Distributed
2
+depends_on:
3
+ - image-build-xpu
4
+steps:
5
+- label: Distributed Model Tests (2 GPUs)
6
+ key: distributed-model-tests-2-gpus
7
+ timeout_in_minutes: 50
8
+ device: intel_gpu
9
+ agent_tags:
10
+ label: production
11
+ gpu: 2+
12
+ mem: 24+
13
+ no_plugin: true
14
+ working_dir: "."
15
+ env:
16
+ REGISTRY: "public.ecr.aws/q9t5s3a7"
17
+ REPO: "vllm-ci-test-repo"
18
+ VLLM_TEST_DEVICE: "xpu"
19
+ source_file_dependencies:
20
+ - vllm/model_executor/model_loader/sharded_state_loader.py
21
+ - vllm/model_executor/models/
22
+ - tests/model_executor/model_loader/test_sharded_state_loader.py
23
+ commands:
24
+ - >-
25
+ bash .buildkite/scripts/hardware_ci/run-intel-test.sh
26
+ 'cd tests &&
27
+ pytest -v -s model_executor/model_loader/test_sharded_state_loader.py -m "not slow_test"'
tests/model_executor/model_loader/test_sharded_state_loader.py
+1
-1
@@ -97,7 +97,7 @@ def test_sharded_state_loader(
97
ctx = mp.get_context("spawn")
98
99
platform_args = {}
100
- if current_platform.is_rocm():
100
+ if current_platform.is_rocm() or current_platform.is_xpu():
101
platform_args["max_num_seqs"] = 1
102
103
# Run in separate processes for memory & CUDA isolation