main
yaml 28 lines 697 Bytes
Raw
1 group: Expert Parallelism
2 depends_on:
3 - image-build-xpu
4 steps:
5 - label: EPLB Algorithm
6 key: eplb-algorithm
7 timeout_in_minutes: 45
8 device: intel_gpu
9 agent_tags:
10 label: production
11 gpu: 1+
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/distributed/eplb
21 - tests/distributed/test_eplb_algo.py
22 - tests/distributed/test_eplb_utils.py
23 commands:
24 - >-
25 bash .buildkite/scripts/hardware_ci/run-intel-test.sh
26 'cd tests &&
27 pytest -v -s distributed/test_eplb_algo.py &&
28 pytest -v -s distributed/test_eplb_utils.py'