main
yaml 29 lines 714 Bytes
Raw
1 group: Samplers Intel
2 depends_on:
3 - image-build-xpu
4 steps:
5 - label: Samplers Test (XPU Kernel)
6 key: samplers-test-xpu-kernel-intel
7 timeout_in_minutes: 40
8 device: intel_gpu
9 agent_tags:
10 label: production
11 gpu: 1+
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/layers
21 - vllm/sampling_metadata.py
22 - tests/samplers
23 - tests/conftest.py
24 - vllm/entrypoints/generate/beam_search
25 commands:
26 - >-
27 bash .buildkite/scripts/hardware_ci/run-intel-test.sh
28 'cd tests &&
29 VLLM_XPU_USE_SAMPLER_KERNEL=1 pytest -v -s samplers'