| 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: 65 |
| 8 | device: intel_gpu |
| 9 | agent_tags: |
| 10 | label: production |
| 11 | gpu: 2+ |
| 12 | mem: 16+ |
| 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"' |