@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / b6101d384d
Fork 0

A high-throughput and memory-efficient inference and serving engine for LLMs

Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git

Synced Nov 8, 02:30 UTC

Deprecate test-pipeline.yaml (#34864)

Signed-off-by: Kevin H. Luu <khluu000@gmail.com>

Kevin H. Luu committed Feb 18, 2026 at 18:15 UTC b6101d384db5709b4422ebd05fe84f0891ff63ce
1 file changed +6 -1522
.buildkite/test-pipeline.yaml
+6 -1522
@@ -1,1524 +1,8 @@
1 -# In this file, you can add more tests to run either by adding a new step or
2 -# adding a new command to an existing step. See different options here for examples.
1 +# This file has been deprecated as of Feb 18, 2026. The content has already been migrated to:
2
4 -# This script will be feed into Jinja template in `test-template-aws.j2` at
5 -# https://github.com/vllm-project/buildkite-ci/blob/main/scripts/test-template-aws.j2
6 -# to generate the final pipeline yaml file.
3 +# .buildkite/test_areas for test jobs
4 +# .buildkite/image_build for image building jobs
5 +# .buildkite/hardware_tests for jobs running on other hardwares (Intel, Ascend NPU, Arm, etc..)
6 +# .buildkite/ci_config.yaml for configuration of CI pipeline
7
8 -# Documentation
9 -# label(str): the name of the test. emojis allowed.
10 -# fast_check(bool): whether to run this on each commit on the fastcheck pipeline.
11 -# torch_nightly(bool): whether to run this on vllm against the torch nightly pipeline.
12 -# fast_check_only(bool): run this test on the fastcheck pipeline only
13 -# optional(bool): never run this test by default (i.e. need to unblock manually) unless it's a scheduled nightly run.
14 -# soft_fail(bool): allow this step to fail without failing the entire pipeline (useful for flaky or experimental tests).
15 -# command(str): the single command to run for tests. incompatible with commands.
16 -# commands(list): the list of commands to run for the test. incompatible with command.
17 -# mirror_hardwares(list): the list of hardware to run the test on as well. currently only supports [amdexperimental]
18 -# gpu(str): override the GPU selection for the test. default is L4 GPUs. supports a100, b200, h200
19 -# num_gpus(int): override the number of GPUs for the test. defaults to 1 GPU. currently supports 2,4.
20 -# num_nodes(int): whether to simulate multi-node setup by launching multiple containers on one host,
21 -# in this case, commands must be specified. the first command runs on the first host, the second
22 -# command runs on the second host.
23 -# timeout_in_minutes(int): sets a timeout for the step in minutes. if not specified, uses the default timeout.
24 -# parallelism(int): number of parallel jobs to run for this step. enables test sharding using $$BUILDKITE_PARALLEL_JOB
25 -# and $$BUILDKITE_PARALLEL_JOB_COUNT environment variables.
26 -# working_dir(str): specify the place where the command should execute, default to /vllm-workspace/tests
27 -# source_file_dependencies(list): the list of prefixes to opt-in the test for, if empty, the test will always run.
28 -# autorun_on_main (bool): default to false, if true, the test will run automatically when commit is pushed to main branch.
29 -
30 -# When adding a test
31 -# - If the test belongs to an existing group, add it there
32 -# - If the test is short, add to any existing step
33 -# - If the test takes more than 10min, then it is okay to create a new step.
34 -# Note that all steps execute in parallel.
35 -
36 -steps:
37 -##### fast check tests #####
38 -
39 -- label: Pytorch Nightly Dependency Override Check # 2min
40 - # if this test fails, it means the nightly torch version is not compatible with some
41 - # of the dependencies. Please check the error message and add the package to whitelist
42 - # in /vllm/tools/pre_commit/generate_nightly_torch_test.py
43 - soft_fail: true
44 - source_file_dependencies:
45 - - requirements/nightly_torch_test.txt
46 - commands:
47 - - bash standalone_tests/pytorch_nightly_dependency.sh
48 -
49 -- label: Async Engine, Inputs, Utils, Worker Test # 36min
50 - timeout_in_minutes: 50
51 - mirror_hardwares: [amdexperimental]
52 - source_file_dependencies:
53 - - vllm/
54 - - tests/detokenizer
55 - - tests/multimodal
56 - - tests/utils_
57 - commands:
58 - - pytest -v -s detokenizer
59 - - pytest -v -s -m 'not cpu_test' multimodal
60 - - pytest -v -s utils_
61 -
62 -- label: Async Engine, Inputs, Utils, Worker, Config Test (CPU) # 20min
63 - timeout_in_minutes: 30
64 - source_file_dependencies:
65 - - vllm/
66 - - tests/test_inputs.py
67 - - tests/test_outputs.py
68 - - tests/test_pooling_params.py
69 - - tests/multimodal
70 - - tests/renderers
71 - - tests/standalone_tests/lazy_imports.py
72 - - tests/tokenizers_
73 - - tests/tool_parsers
74 - - tests/transformers_utils
75 - - tests/config
76 - no_gpu: true
77 - commands:
78 - - python3 standalone_tests/lazy_imports.py
79 - - pytest -v -s test_inputs.py
80 - - pytest -v -s test_outputs.py
81 - - pytest -v -s test_pooling_params.py
82 - - pytest -v -s -m 'cpu_test' multimodal
83 - - pytest -v -s renderers
84 - - pytest -v -s tokenizers_
85 - - pytest -v -s tool_parsers
86 - - pytest -v -s transformers_utils
87 - - pytest -v -s config
88 -
89 -- label: Python-only Installation Test # 10min
90 - timeout_in_minutes: 20
91 - mirror_hardwares: [amdexperimental]
92 - source_file_dependencies:
93 - - tests/standalone_tests/python_only_compile.sh
94 - - setup.py
95 - commands:
96 - - bash standalone_tests/python_only_compile.sh
97 -
98 -- label: Basic Correctness Test # 20min
99 - timeout_in_minutes: 30
100 - mirror_hardwares: [amdexperimental]
101 - fast_check: true
102 - torch_nightly: true
103 - source_file_dependencies:
104 - - vllm/
105 - - tests/basic_correctness/test_basic_correctness
106 - - tests/basic_correctness/test_cpu_offload
107 - - tests/basic_correctness/test_cumem.py
108 - commands:
109 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
110 - - pytest -v -s basic_correctness/test_cumem.py
111 - - pytest -v -s basic_correctness/test_basic_correctness.py
112 - - pytest -v -s basic_correctness/test_cpu_offload.py
113 -
114 -- label: Entrypoints Unit Tests # 5min
115 - timeout_in_minutes: 10
116 - working_dir: "/vllm-workspace/tests"
117 - fast_check: true
118 - source_file_dependencies:
119 - - vllm/entrypoints
120 - - tests/entrypoints/
121 - commands:
122 - - pytest -v -s entrypoints/openai/tool_parsers
123 - - pytest -v -s entrypoints/ --ignore=entrypoints/llm --ignore=entrypoints/rpc --ignore=entrypoints/instrumentator --ignore=entrypoints/openai --ignore=entrypoints/offline_mode --ignore=entrypoints/test_chat_utils.py --ignore=entrypoints/pooling
124 -
125 -- label: Entrypoints Integration Test (LLM) # 30min
126 - timeout_in_minutes: 40
127 - mirror_hardwares: [amdexperimental]
128 - working_dir: "/vllm-workspace/tests"
129 - fast_check: true
130 - torch_nightly: true
131 - source_file_dependencies:
132 - - vllm/
133 - - tests/entrypoints/llm
134 - - tests/entrypoints/offline_mode
135 - commands:
136 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
137 - - pytest -v -s entrypoints/llm --ignore=entrypoints/llm/test_generate.py --ignore=entrypoints/llm/test_collective_rpc.py
138 - - pytest -v -s entrypoints/llm/test_generate.py # it needs a clean process
139 - - pytest -v -s entrypoints/offline_mode # Needs to avoid interference with other tests
140 -
141 -- label: Entrypoints Integration Test (API Server 1) # 100min
142 - timeout_in_minutes: 130
143 - mirror_hardwares: [amdexperimental]
144 - working_dir: "/vllm-workspace/tests"
145 - fast_check: true
146 - torch_nightly: true
147 - source_file_dependencies:
148 - - vllm/
149 - - tests/entrypoints/openai
150 - - tests/entrypoints/test_chat_utils
151 - commands:
152 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
153 - - pytest -v -s entrypoints/openai --ignore=entrypoints/openai/test_chat_with_tool_reasoning.py --ignore=entrypoints/instrumentator --ignore=entrypoints/openai/test_oot_registration.py --ignore=entrypoints/openai/test_tensorizer_entrypoint.py --ignore=entrypoints/openai/correctness/ --ignore=entrypoints/openai/tool_parsers/ --ignore=entrypoints/openai/responses
154 - - pytest -v -s entrypoints/test_chat_utils.py
155 -
156 -- label: Entrypoints Integration Test (API Server 2)
157 - timeout_in_minutes: 50
158 - mirror_hardwares: [amdexperimental]
159 - working_dir: "/vllm-workspace/tests"
160 - fast_check: true
161 - torch_nightly: true
162 - source_file_dependencies:
163 - - vllm/
164 - - tests/entrypoints/rpc
165 - - tests/entrypoints/instrumentator
166 - - tests/tool_use
167 - commands:
168 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
169 - - pytest -v -s entrypoints/instrumentator
170 - - PYTHONPATH=/vllm-workspace pytest -v -s entrypoints/rpc
171 - - pytest -v -s tool_use
172 -
173 -- label: Entrypoints Integration Test (Pooling)
174 - timeout_in_minutes: 50
175 - mirror_hardwares: [amdexperimental]
176 - working_dir: "/vllm-workspace/tests"
177 - fast_check: true
178 - torch_nightly: true
179 - source_file_dependencies:
180 - - vllm/
181 - - tests/entrypoints/pooling
182 - commands:
183 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
184 - - pytest -v -s entrypoints/pooling
185 -
186 -- label: Entrypoints Integration Test (Responses API)
187 - timeout_in_minutes: 50
188 - mirror_hardwares: [amdexperimental]
189 - working_dir: "/vllm-workspace/tests"
190 - fast_check: true
191 - torch_nightly: true
192 - source_file_dependencies:
193 - - vllm/
194 - - tests/entrypoints/openai/responses
195 - commands:
196 - - pytest -v -s entrypoints/openai/responses
197 -
198 -- label: Distributed Tests (4 GPUs) # 35min
199 - timeout_in_minutes: 50
200 - mirror_hardwares: [amdexperimental]
201 - working_dir: "/vllm-workspace/tests"
202 - num_gpus: 4
203 - source_file_dependencies:
204 - - vllm/distributed/
205 - - tests/distributed/test_utils
206 - - tests/distributed/test_pynccl
207 - - tests/distributed/test_events
208 - - tests/compile/fullgraph/test_basic_correctness.py
209 - - examples/offline_inference/rlhf.py
210 - - examples/offline_inference/rlhf_colocate.py
211 - - examples/offline_inference/new_weight_syncing/
212 - - tests/examples/offline_inference/data_parallel.py
213 - - tests/v1/distributed
214 - - tests/v1/engine/test_engine_core_client.py
215 - - tests/distributed/test_symm_mem_allreduce.py
216 - commands:
217 - # https://github.com/NVIDIA/nccl/issues/1838
218 - - export NCCL_CUMEM_HOST_ENABLE=0
219 - # test with torchrun tp=2 and external_dp=2
220 - - torchrun --nproc-per-node=4 distributed/test_torchrun_example.py
221 - # test with torchrun tp=2 and pp=2
222 - - PP_SIZE=2 torchrun --nproc-per-node=4 distributed/test_torchrun_example.py
223 - # test with torchrun tp=4 and dp=1
224 - - TP_SIZE=4 torchrun --nproc-per-node=4 distributed/test_torchrun_example_moe.py
225 - # test with torchrun tp=2, pp=2 and dp=1
226 - - PP_SIZE=2 TP_SIZE=2 torchrun --nproc-per-node=4 distributed/test_torchrun_example_moe.py
227 - # test with torchrun tp=1 and dp=4 with ep
228 - - DP_SIZE=4 ENABLE_EP=1 torchrun --nproc-per-node=4 distributed/test_torchrun_example_moe.py
229 - # test with torchrun tp=2 and dp=2 with ep
230 - - TP_SIZE=2 DP_SIZE=2 ENABLE_EP=1 torchrun --nproc-per-node=4 distributed/test_torchrun_example_moe.py
231 - # test with internal dp
232 - - python3 ../examples/offline_inference/data_parallel.py --enforce-eager
233 - - TP_SIZE=2 DP_SIZE=2 pytest -v -s v1/distributed/test_async_llm_dp.py
234 - - TP_SIZE=2 DP_SIZE=2 pytest -v -s v1/distributed/test_eagle_dp.py
235 - - TP_SIZE=2 DP_SIZE=2 pytest -v -s v1/distributed/test_external_lb_dp.py
236 - - TP_SIZE=1 DP_SIZE=4 pytest -v -s v1/distributed/test_internal_lb_dp.py
237 - - TP_SIZE=1 DP_SIZE=4 pytest -v -s v1/distributed/test_hybrid_lb_dp.py
238 - - pytest -v -s v1/engine/test_engine_core_client.py::test_kv_cache_events_dp
239 - - pytest -v -s distributed/test_utils.py
240 - - pytest -v -s compile/fullgraph/test_basic_correctness.py
241 - - pytest -v -s distributed/test_pynccl.py
242 - - pytest -v -s distributed/test_events.py
243 - - pytest -v -s distributed/test_symm_mem_allreduce.py
244 - # TODO: create a dedicated test section for multi-GPU example tests
245 - # when we have multiple distributed example tests
246 - # OLD rlhf examples
247 - - pushd ../examples/offline_inference
248 - - VLLM_ALLOW_INSECURE_SERIALIZATION=1 python3 rlhf.py
249 - - VLLM_ALLOW_INSECURE_SERIALIZATION=1 RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py
250 - - popd
251 - # NEW rlhf examples
252 - - pushd ../examples/offline_inference/new_weight_syncing
253 - - VLLM_ALLOW_INSECURE_SERIALIZATION=1 python3 rlhf.py
254 - - VLLM_ALLOW_INSECURE_SERIALIZATION=1 python3 rlhf_async_new_apis.py
255 - - popd
256 -
257 -- label: Distributed Tests (8 GPUs) # 4min
258 - timeout_in_minutes: 10
259 - gpu: h100
260 - num_gpus: 8
261 - working_dir: "/vllm-workspace/tests"
262 - source_file_dependencies:
263 - - examples/offline_inference/torchrun_dp_example.py
264 - - vllm/config/parallel.py
265 - - vllm/distributed/
266 - - vllm/v1/engine/llm_engine.py
267 - - vllm/v1/executor/uniproc_executor.py
268 - - vllm/v1/worker/gpu_worker.py
269 - commands:
270 - # https://github.com/NVIDIA/nccl/issues/1838
271 - - export NCCL_CUMEM_HOST_ENABLE=0
272 - # test with torchrun tp=2 and dp=4 with ep
273 - - torchrun --nproc-per-node=8 ../examples/offline_inference/torchrun_dp_example.py --tp-size=2 --pp-size=1 --dp-size=4 --enable-ep
274 -
275 -- label: EPLB Algorithm Test # 5min
276 - timeout_in_minutes: 15
277 - working_dir: "/vllm-workspace/tests"
278 - source_file_dependencies:
279 - - vllm/distributed/eplb
280 - - tests/distributed/test_eplb_algo.py
281 - commands:
282 - - pytest -v -s distributed/test_eplb_algo.py
283 -
284 -- label: EPLB Execution Test # 10min
285 - timeout_in_minutes: 20
286 - working_dir: "/vllm-workspace/tests"
287 - num_gpus: 4
288 - source_file_dependencies:
289 - - vllm/distributed/eplb
290 - - tests/distributed/test_eplb_execute.py
291 - commands:
292 - - pytest -v -s distributed/test_eplb_execute.py
293 - - pytest -v -s distributed/test_eplb_spec_decode.py
294 -
295 -- label: Metrics, Tracing Test # 12min
296 - timeout_in_minutes: 20
297 - mirror_hardwares: [amdexperimental]
298 - num_gpus: 2
299 - source_file_dependencies:
300 - - vllm/
301 - - tests/v1/tracing
302 - commands:
303 - - "pip install \
304 - 'opentelemetry-sdk>=1.26.0' \
305 - 'opentelemetry-api>=1.26.0' \
306 - 'opentelemetry-exporter-otlp>=1.26.0' \
307 - 'opentelemetry-semantic-conventions-ai>=0.4.1'"
308 - - pytest -v -s v1/tracing
309 -
310 -##### fast check tests #####
311 -##### 1 GPU test #####
312 -
313 -- label: Regression Test # 7min
314 - timeout_in_minutes: 20
315 - mirror_hardwares: [amdexperimental]
316 - source_file_dependencies:
317 - - vllm/
318 - - tests/test_regression
319 - commands:
320 - - pip install modelscope
321 - - pytest -v -s test_regression.py
322 - working_dir: "/vllm-workspace/tests" # optional
323 -
324 -- label: Engine Test # 9min
325 - timeout_in_minutes: 15
326 - mirror_hardwares: [amdexperimental]
327 - source_file_dependencies:
328 - - vllm/
329 - - tests/engine
330 - - tests/test_sequence
331 - - tests/test_config
332 - - tests/test_logger
333 - - tests/test_vllm_port
334 - commands:
335 - - pytest -v -s engine test_sequence.py test_config.py test_logger.py test_vllm_port.py
336 -
337 -- label: V1 Test e2e + engine # 30min
338 - timeout_in_minutes: 45
339 - mirror_hardwares: [amdexperimental]
340 - source_file_dependencies:
341 - - vllm/
342 - - tests/v1
343 - commands:
344 - # TODO: accuracy does not match, whether setting
345 - # VLLM_USE_FLASHINFER_SAMPLER or not on H100.
346 - - pytest -v -s v1/e2e
347 - # Run this test standalone for now;
348 - # need to untangle use (implicit) use of spawn/fork across the tests.
349 - - pytest -v -s v1/engine/test_preprocess_error_handling.py
350 - - pytest -v -s v1/engine --ignore v1/engine/test_preprocess_error_handling.py
351 -
352 -- label: V1 Test entrypoints # 35min
353 - timeout_in_minutes: 50
354 - mirror_hardwares: [amdexperimental]
355 - source_file_dependencies:
356 - - vllm/
357 - - tests/v1
358 - commands:
359 - - pytest -v -s v1/entrypoints
360 -
361 -- label: V1 Test others # 42min
362 - timeout_in_minutes: 60
363 - mirror_hardwares: [amdexperimental]
364 - source_file_dependencies:
365 - - vllm/
366 - - tests/v1
367 - commands:
368 - - uv pip install --system -r /vllm-workspace/requirements/kv_connectors.txt
369 - # split the test to avoid interference
370 - - pytest -v -s -m 'not cpu_test' v1/core
371 - - pytest -v -s v1/executor
372 - - pytest -v -s v1/kv_offload
373 - - pytest -v -s v1/sample
374 - - pytest -v -s v1/logits_processors
375 - - pytest -v -s v1/worker
376 - - pytest -v -s -m 'not slow_test' v1/spec_decode
377 - - pytest -v -s -m 'not cpu_test' v1/kv_connector/unit
378 - - pytest -v -s -m 'not cpu_test' v1/metrics
379 - - pytest -v -s v1/test_oracle.py
380 - - pytest -v -s v1/test_request.py
381 - - pytest -v -s v1/test_outputs.py
382 - # Integration test for streaming correctness (requires special branch).
383 - - pip install -U git+https://github.com/robertgshaw2-redhat/lm-evaluation-harness.git@streaming-api
384 - - pytest -v -s entrypoints/openai/correctness/test_lmeval.py::test_lm_eval_accuracy_v1_engine
385 -
386 -- label: V1 Test attention (H100) # 10min
387 - timeout_in_minutes: 30
388 - gpu: h100
389 - source_file_dependencies:
390 - - vllm/config/attention.py
391 - - vllm/model_executor/layers/attention
392 - - vllm/v1/attention
393 - - tests/v1/attention
394 - commands:
395 - - pytest -v -s v1/attention
396 -
397 -- label: Batch Invariance Tests (H100) # 10min
398 - timeout_in_minutes: 25
399 - gpu: h100
400 - source_file_dependencies:
401 - - vllm/v1/attention
402 - - vllm/model_executor/layers
403 - - tests/v1/determinism/
404 - commands:
405 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
406 - - pip install pytest-timeout pytest-forked
407 - - pytest -v -s v1/determinism/test_batch_invariance.py
408 - - pytest -v -s v1/determinism/test_rms_norm_batch_invariant.py
409 -
410 -- label: V1 Test attention (B200) # 10min
411 - timeout_in_minutes: 30
412 - gpu: b200
413 - source_file_dependencies:
414 - - vllm/config/attention.py
415 - - vllm/model_executor/layers/attention
416 - - vllm/v1/attention
417 - - tests/v1/attention
418 - commands:
419 - - pytest -v -s v1/attention
420 -
421 -- label: V1 Test others (CPU) # 5 mins
422 - source_file_dependencies:
423 - - vllm/
424 - - tests/v1
425 - no_gpu: true
426 - commands:
427 - # split the test to avoid interference
428 - - pytest -v -s -m 'cpu_test' v1/core
429 - - pytest -v -s v1/structured_output
430 - - pytest -v -s v1/test_serial_utils.py
431 - - pytest -v -s -m 'cpu_test' v1/kv_connector/unit
432 - - pytest -v -s -m 'cpu_test' v1/metrics
433 -
434 -
435 -- label: Examples Test # 30min
436 - timeout_in_minutes: 45
437 - mirror_hardwares: [amdexperimental]
438 - working_dir: "/vllm-workspace/examples"
439 - source_file_dependencies:
440 - - vllm/entrypoints
441 - - vllm/multimodal
442 - - examples/
443 - commands:
444 - - pip install tensorizer # for tensorizer test
445 - # for basic
446 - - python3 offline_inference/basic/chat.py
447 - - python3 offline_inference/basic/generate.py --model facebook/opt-125m
448 - - python3 offline_inference/basic/generate.py --model meta-llama/Llama-2-13b-chat-hf --cpu-offload-gb 10
449 - - python3 offline_inference/basic/classify.py
450 - - python3 offline_inference/basic/embed.py
451 - - python3 offline_inference/basic/score.py
452 - # for multi-modal models
453 - - python3 offline_inference/audio_language.py --seed 0
454 - - python3 offline_inference/vision_language.py --seed 0
455 - - python3 offline_inference/vision_language_multi_image.py --seed 0
456 - - python3 offline_inference/encoder_decoder_multimodal.py --model-type whisper --seed 0
457 - # for pooling models
458 - - python3 pooling/embed/vision_embedding_offline.py --seed 0
459 - # for features demo
460 - - python3 offline_inference/prefix_caching.py
461 - - python3 offline_inference/llm_engine_example.py
462 - - python3 others/tensorize_vllm_model.py --model facebook/opt-125m serialize --serialized-directory /tmp/ --suffix v1 && python3 others/tensorize_vllm_model.py --model facebook/opt-125m deserialize --path-to-tensors /tmp/vllm/facebook/opt-125m/v1/model.tensors
463 - - python3 offline_inference/spec_decode.py --test --method eagle --num_spec_tokens 3 --dataset-name hf --dataset-path philschmid/mt-bench --num-prompts 80 --temp 0 --top-p 1.0 --top-k -1 --tp 1 --enable-chunked-prefill --max-model-len 2048
464 - # https://github.com/vllm-project/vllm/pull/26682 uses slightly more memory in PyTorch 2.9+ causing this test to OOM in 1xL4 GPU
465 - - python3 offline_inference/spec_decode.py --test --method eagle3 --num_spec_tokens 3 --dataset-name hf --dataset-path philschmid/mt-bench --num-prompts 80 --temp 0 --top-p 1.0 --top-k -1 --tp 1 --enable-chunked-prefill --max-model-len 1536
466 -
467 -- label: Platform Tests (CUDA) # 4min
468 - timeout_in_minutes: 15
469 - mirror_hardwares: [amdexperimental]
470 - source_file_dependencies:
471 - - vllm/
472 - - tests/cuda
473 - commands:
474 - - pytest -v -s cuda/test_cuda_context.py
475 -
476 -- label: Samplers Test # 56min
477 - timeout_in_minutes: 75
478 - mirror_hardwares: [amdexperimental]
479 - source_file_dependencies:
480 - - vllm/model_executor/layers
481 - - vllm/sampling_metadata.py
482 - - tests/samplers
483 - - tests/conftest.py
484 - commands:
485 - - pytest -v -s samplers
486 - - VLLM_USE_FLASHINFER_SAMPLER=1 pytest -v -s samplers
487 -
488 -- label: LoRA Test %N # 20min each
489 - timeout_in_minutes: 30
490 - mirror_hardwares: [amdexperimental]
491 - source_file_dependencies:
492 - - vllm/lora
493 - - tests/lora
494 - commands:
495 - - pytest -v -s lora \
496 - --shard-id=$$BUILDKITE_PARALLEL_JOB \
497 - --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT \
498 - --ignore=lora/test_chatglm3_tp.py \
499 - --ignore=lora/test_llama_tp.py \
500 - --ignore=lora/test_llm_with_multi_loras.py \
501 - --ignore=lora/test_olmoe_tp.py \
502 - --ignore=lora/test_deepseekv2_tp.py \
503 - --ignore=lora/test_gptoss_tp.py \
504 - --ignore=lora/test_qwen3moe_tp.py
505 -
506 - parallelism: 4
507 -
508 -- label: PyTorch Compilation Unit Tests # 15min
509 - timeout_in_minutes: 30
510 - mirror_hardwares: [amdexperimental]
511 - torch_nightly: true
512 - source_file_dependencies:
513 - - vllm/
514 - - tests/compile
515 - commands:
516 - # Run unit tests defined directly under compile/,
517 - # not including subdirectories, which are usually heavier
518 - # tests covered elsewhere.
519 - # Use `find` to launch multiple instances of pytest so that
520 - # they do not suffer from https://github.com/vllm-project/vllm/issues/28965
521 - # However, find does not normally propagate error codes, so we combine it with xargs
522 - # (using -0 for proper path handling)
523 - - "find compile/ -maxdepth 1 -name 'test_*.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
524 - - pytest -s -v compile/passes --ignore compile/passes/distributed
525 -
526 -- label: PyTorch Fullgraph Smoke Test # 15min
527 - timeout_in_minutes: 30
528 - mirror_hardwares: [amdexperimental]
529 - torch_nightly: true
530 - source_file_dependencies:
531 - - vllm/
532 - - tests/compile
533 - commands:
534 - # Run smoke tests under fullgraph directory, except test_full_graph.py
535 - # as it is a heavy test that is covered in other steps.
536 - # Use `find` to launch multiple instances of pytest so that
537 - # they do not suffer from https://github.com/vllm-project/vllm/issues/28965
538 - # However, find does not normally propagate error codes, so we combine it with xargs
539 - # (using -0 for proper path handling)
540 - - "find compile/fullgraph -maxdepth 1 -name 'test_*.py' -not -name 'test_full_graph.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
541 -
542 -- label: PyTorch Fullgraph Test # 27min
543 - timeout_in_minutes: 40
544 - mirror_hardwares: [amdexperimental]
545 - torch_nightly: true
546 - source_file_dependencies:
547 - - vllm/
548 - - tests/compile
549 - commands:
550 - # fp8 kv scales not supported on sm89, tested on Blackwell instead
551 - - pytest -v -s compile/fullgraph/test_full_graph.py -k 'not test_fp8_kv_scale_compile'
552 - # # Limit to no custom ops to reduce running time
553 - # # Wrap with quotes to escape yaml and avoid starting -k string with a -
554 - # - "pytest -v -s compile/distributed/test_fusions_e2e.py -k 'TRITON and not +quant_fp8 and not Llama-4'"
555 - # Old E2E tests were removed in https://github.com/vllm-project/vllm/pull/33293
556 - # in favor of new tests in fusions_e2e. We avoid replicating the new jobs in this file as it's deprecated.
557 -
558 -- label: Cudagraph test
559 - timeout_in_minutes: 20
560 - mirror_hardwares: [amdexperimental]
561 - source_file_dependencies:
562 - - tests/v1/cudagraph
563 - - vllm/v1/cudagraph_dispatcher.py
564 - - vllm/config/compilation.py
565 - - vllm/compilation
566 - commands:
567 - - pytest -v -s v1/cudagraph/test_cudagraph_dispatch.py
568 - - pytest -v -s v1/cudagraph/test_cudagraph_mode.py
569 -
570 -- label: Kernels Core Operation Test # 48min
571 - timeout_in_minutes: 75
572 - mirror_hardwares: [amdexperimental]
573 - source_file_dependencies:
574 - - csrc/
575 - - tests/kernels/core
576 - - tests/kernels/test_top_k_per_row.py
577 - commands:
578 - - pytest -v -s kernels/core kernels/test_top_k_per_row.py
579 -
580 -- label: Kernels Attention Test %N # 23min
581 - timeout_in_minutes: 35
582 - mirror_hardwares: [amdexperimental]
583 - source_file_dependencies:
584 - - csrc/attention/
585 - - vllm/v1/attention
586 - # TODO: remove this dependency (https://github.com/vllm-project/vllm/issues/32267)
587 - - vllm/model_executor/layers/attention
588 - - tests/kernels/attention
589 - commands:
590 - - pytest -v -s kernels/attention --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
591 - parallelism: 2
592 -
593 -- label: Kernels Quantization Test %N # 64min
594 - timeout_in_minutes: 90
595 - mirror_hardwares: [amdexperimental]
596 - source_file_dependencies:
597 - - csrc/quantization/
598 - - vllm/model_executor/layers/quantization
599 - - tests/kernels/quantization
600 - commands:
601 - - pytest -v -s kernels/quantization --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
602 - parallelism: 2
603 -
604 -- label: Kernels MoE Test %N # 40min
605 - timeout_in_minutes: 60
606 - mirror_hardwares: [amdexperimental]
607 - source_file_dependencies:
608 - - csrc/quantization/cutlass_w8a8/moe/
609 - - csrc/moe/
610 - - tests/kernels/moe
611 - - vllm/model_executor/layers/fused_moe/
612 - - vllm/distributed/device_communicators/
613 - - vllm/envs.py
614 - - vllm/config
615 - commands:
616 - - pytest -v -s kernels/moe --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
617 - parallelism: 2
618 -
619 -- label: Kernels Mamba Test # 31min
620 - timeout_in_minutes: 45
621 - mirror_hardwares: [amdexperimental]
622 - source_file_dependencies:
623 - - csrc/mamba/
624 - - tests/kernels/mamba
625 - - vllm/model_executor/layers/mamba/ops
626 - commands:
627 - - pytest -v -s kernels/mamba
628 -
629 -- label: Kernels DeepGEMM Test (H100)
630 - timeout_in_minutes: 45
631 - gpu: h100
632 - num_gpus: 1
633 - source_file_dependencies:
634 - - tools/install_deepgemm.sh
635 - - vllm/utils/deep_gemm.py
636 - - vllm/model_executor/layers/fused_moe
637 - - vllm/model_executor/layers/quantization
638 - - tests/kernels/quantization/test_block_fp8.py
639 - - tests/kernels/moe/test_deepgemm.py
640 - - tests/kernels/moe/test_batched_deepgemm.py
641 - - tests/kernels/attention/test_deepgemm_attention.py
642 - commands:
643 - - pytest -v -s kernels/quantization/test_block_fp8.py -k deep_gemm
644 - - pytest -v -s kernels/moe/test_deepgemm.py
645 - - pytest -v -s kernels/moe/test_batched_deepgemm.py
646 - - pytest -v -s kernels/attention/test_deepgemm_attention.py
647 -
648 -- label: Kernels Helion Test
649 - timeout_in_minutes: 30
650 - gpu: h100
651 - source_file_dependencies:
652 - - vllm/utils/import_utils.py
653 - - tests/kernels/helion/
654 - commands:
655 - - pip install helion
656 - - pytest -v -s kernels/helion/
657 -
658 -
659 -- label: Kernels FP8 MoE Test (1 H100)
660 - timeout_in_minutes: 90
661 - gpu: h100
662 - num_gpus: 1
663 - optional: true
664 - commands:
665 - - pytest -v -s kernels/moe/test_cutlass_moe.py
666 - - pytest -v -s kernels/moe/test_flashinfer.py
667 - - pytest -v -s kernels/moe/test_gpt_oss_triton_kernels.py
668 - - pytest -v -s kernels/moe/test_modular_oai_triton_moe.py
669 - - pytest -v -s kernels/moe/test_moe.py
670 - # - pytest -v -s kernels/moe/test_block_fp8.py - failing on main
671 - - pytest -v -s kernels/moe/test_block_int8.py
672 - - pytest -v -s kernels/moe/test_triton_moe_no_act_mul.py
673 - - pytest -v -s kernels/moe/test_triton_moe_ptpc_fp8.py
674 -
675 -- label: Kernels FP8 MoE Test (2 H100s)
676 - timeout_in_minutes: 90
677 - gpu: h100
678 - num_gpus: 2
679 - optional: true
680 - commands:
681 - - pytest -v -s kernels/moe/test_deepep_deepgemm_moe.py
682 - - pytest -v -s kernels/moe/test_deepep_moe.py
683 - - pytest -v -s kernels/moe/test_pplx_cutlass_moe.py
684 - # - pytest -v -s kernels/moe/test_pplx_moe.py - failing on main
685 -
686 -- label: Kernels Fp4 MoE Test (B200)
687 - timeout_in_minutes: 60
688 - gpu: b200
689 - num_gpus: 1
690 - optional: true
691 - commands:
692 - - pytest -v -s kernels/moe/test_cutedsl_moe.py
693 - - pytest -v -s kernels/moe/test_flashinfer_moe.py
694 - - pytest -v -s kernels/moe/test_nvfp4_moe.py
695 - - pytest -v -s kernels/moe/test_ocp_mx_moe.py
696 -
697 -
698 -- label: Model Executor Test # 23min
699 - timeout_in_minutes: 35
700 - torch_nightly: true
701 - mirror_hardwares: [amdexperimental]
702 - source_file_dependencies:
703 - - vllm/engine/arg_utils.py
704 - - vllm/config/model.py
705 - - vllm/model_executor
706 - - tests/model_executor
707 - - tests/entrypoints/openai/test_tensorizer_entrypoint.py
708 - commands:
709 - - apt-get update && apt-get install -y curl libsodium23
710 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
711 - - pytest -v -s model_executor
712 - - pytest -v -s entrypoints/openai/test_tensorizer_entrypoint.py
713 -
714 -- label: Benchmarks # 11min
715 - timeout_in_minutes: 20
716 - mirror_hardwares: [amdexperimental]
717 - working_dir: "/vllm-workspace/.buildkite"
718 - source_file_dependencies:
719 - - benchmarks/
720 - commands:
721 - - bash scripts/run-benchmarks.sh
722 -
723 -- label: Benchmarks CLI Test # 7min
724 - timeout_in_minutes: 20
725 - mirror_hardwares: [amdexperimental]
726 - source_file_dependencies:
727 - - vllm/
728 - - tests/benchmarks/
729 - commands:
730 - - pytest -v -s benchmarks/
731 -
732 -- label: Quantization Test # 70min
733 - timeout_in_minutes: 90
734 - mirror_hardwares: [amdexperimental]
735 - source_file_dependencies:
736 - - csrc/
737 - - vllm/model_executor/layers/quantization
738 - - tests/quantization
739 - commands:
740 - # temporary install here since we need nightly, will move to requirements/test.in
741 - # after torchao 0.12 release, and pin a working version of torchao nightly here
742 -
743 - # since torchao nightly is only compatible with torch nightly currently
744 - # https://github.com/pytorch/ao/issues/2919, we'll have to skip new torchao tests for now
745 - # we can only upgrade after this is resolved
746 - # TODO(jerryzh168): resolve the above comment
747 - - uv pip install --system torchao==0.14.1 --index-url https://download.pytorch.org/whl/cu129
748 - - uv pip install --system conch-triton-kernels
749 - - VLLM_TEST_FORCE_LOAD_FORMAT=auto pytest -v -s quantization/ --ignore quantization/test_blackwell_moe.py
750 -
751 -- label: LM Eval Small Models # 53min
752 - timeout_in_minutes: 75
753 - mirror_hardwares: [amdexperimental]
754 - source_file_dependencies:
755 - - csrc/
756 - - vllm/model_executor/layers/quantization
757 - autorun_on_main: true
758 - commands:
759 - - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-small.txt
760 -
761 -- label: OpenAI API correctness # 22min
762 - timeout_in_minutes: 30
763 - mirror_hardwares: [amdexperimental]
764 - source_file_dependencies:
765 - - csrc/
766 - - vllm/entrypoints/openai/
767 - - vllm/model_executor/models/whisper.py
768 - commands: # LMEval+Transcription WER check
769 - - pytest -s entrypoints/openai/correctness/
770 -
771 -##### models test #####
772 -
773 -- label: Basic Models Tests (Initialization)
774 - timeout_in_minutes: 45
775 - mirror_hardwares: [amdexperimental]
776 - torch_nightly: true
777 - source_file_dependencies:
778 - - vllm/
779 - - tests/models/test_initialization.py
780 - - tests/models/registry.py
781 - commands:
782 - # Run a subset of model initialization tests
783 - - pytest -v -s models/test_initialization.py::test_can_initialize_small_subset
784 -
785 -- label: Basic Models Tests (Extra Initialization) %N
786 - timeout_in_minutes: 45
787 - mirror_hardwares: [amdexperimental]
788 - torch_nightly: true
789 - source_file_dependencies:
790 - - vllm/model_executor/models/
791 - - vllm/transformers_utils/
792 - - tests/models/test_initialization.py
793 - - tests/models/registry.py
794 - commands:
795 - # Only when vLLM model source is modified - test initialization of a large
796 - # subset of supported models (the complement of the small subset in the above
797 - # test.) Also run if model initialization test file is modified
798 - - pytest -v -s models/test_initialization.py \
799 - -k 'not test_can_initialize_small_subset' \
800 - --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT \
801 - --shard-id=$$BUILDKITE_PARALLEL_JOB
802 - parallelism: 2
803 -
804 -- label: Basic Models Tests (Other)
805 - timeout_in_minutes: 45
806 - mirror_hardwares: [amdexperimental]
807 - torch_nightly: true
808 - source_file_dependencies:
809 - - vllm/
810 - - tests/models/test_terratorch.py
811 - - tests/models/test_transformers.py
812 - - tests/models/test_registry.py
813 - commands:
814 - - pytest -v -s models/test_terratorch.py models/test_transformers.py models/test_registry.py
815 -
816 -- label: Basic Models Test (Other CPU) # 5min
817 - timeout_in_minutes: 10
818 - torch_nightly: true
819 - source_file_dependencies:
820 - - vllm/
821 - - tests/models/test_utils.py
822 - - tests/models/test_vision.py
823 - no_gpu: true
824 - commands:
825 - - pytest -v -s models/test_utils.py models/test_vision.py
826 -
827 -- label: Language Models Tests (Standard)
828 - timeout_in_minutes: 25
829 - mirror_hardwares: [amdexperimental]
830 - torch_nightly: true
831 - source_file_dependencies:
832 - - vllm/
833 - - tests/models/language
834 - commands:
835 - # Test standard language models, excluding a subset of slow tests
836 - - pip freeze | grep -E 'torch'
837 - - pytest -v -s models/language -m 'core_model and (not slow_test)'
838 -
839 -- label: Language Models Tests (Extra Standard) %N
840 - timeout_in_minutes: 45
841 - mirror_hardwares: [amdexperimental]
842 - torch_nightly: true
843 - source_file_dependencies:
844 - - vllm/model_executor/models/
845 - - tests/models/language/pooling/test_embedding.py
846 - - tests/models/language/generation/test_common.py
847 - - tests/models/language/pooling/test_classification.py
848 - commands:
849 - # Shard slow subset of standard language models tests. Only run when model
850 - # source is modified, or when specified test files are modified
851 - - pip freeze | grep -E 'torch'
852 - - pytest -v -s models/language -m 'core_model and slow_test' \
853 - --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT \
854 - --shard-id=$$BUILDKITE_PARALLEL_JOB
855 - parallelism: 2
856 -
857 -- label: Language Models Tests (Hybrid) %N
858 - timeout_in_minutes: 75
859 - mirror_hardwares: [amdexperimental]
860 - torch_nightly: true
861 - source_file_dependencies:
862 - - vllm/
863 - - tests/models/language/generation
864 - commands:
865 - # Install fast path packages for testing against transformers
866 - # Note: also needed to run plamo2 model in vLLM
867 - - uv pip install --system --no-build-isolation 'git+https://github.com/state-spaces/mamba@v2.3.0'
868 - - uv pip install --system --no-build-isolation 'git+https://github.com/Dao-AILab/causal-conv1d@v1.5.2'
869 - # Shard hybrid language model tests
870 - - pytest -v -s models/language/generation \
871 - -m hybrid_model \
872 - --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT \
873 - --shard-id=$$BUILDKITE_PARALLEL_JOB
874 - parallelism: 2
875 -
876 -- label: Language Models Test (Extended Generation) # 80min
877 - timeout_in_minutes: 110
878 - mirror_hardwares: [amdexperimental]
879 - optional: true
880 - source_file_dependencies:
881 - - vllm/
882 - - tests/models/language/generation
883 - commands:
884 - # Install fast path packages for testing against transformers
885 - # Note: also needed to run plamo2 model in vLLM
886 - - uv pip install --system --no-build-isolation 'git+https://github.com/state-spaces/mamba@v2.3.0'
887 - - uv pip install --system --no-build-isolation 'git+https://github.com/Dao-AILab/causal-conv1d@v1.5.2'
888 - - pytest -v -s models/language/generation -m '(not core_model) and (not hybrid_model)'
889 -
890 -- label: Language Models Test (PPL)
891 - timeout_in_minutes: 110
892 - mirror_hardwares: [amdexperimental]
893 - optional: true
894 - source_file_dependencies:
895 - - vllm/
896 - - tests/models/language/generation_ppl_test
897 - commands:
898 - - pytest -v -s models/language/generation_ppl_test
899 -
900 -- label: Language Models Test (Extended Pooling) # 36min
901 - timeout_in_minutes: 50
902 - mirror_hardwares: [amdexperimental]
903 - optional: true
904 - source_file_dependencies:
905 - - vllm/
906 - - tests/models/language/pooling
907 - commands:
908 - - pytest -v -s models/language/pooling -m 'not core_model'
909 -
910 -- label: Language Models Test (MTEB)
911 - timeout_in_minutes: 110
912 - mirror_hardwares: [amdexperimental]
913 - optional: true
914 - source_file_dependencies:
915 - - vllm/
916 - - tests/models/language/pooling_mteb_test
917 - commands:
918 - - pytest -v -s models/language/pooling_mteb_test
919 -
920 -- label: Multi-Modal Processor Test (CPU)
921 - timeout_in_minutes: 60
922 - source_file_dependencies:
923 - - vllm/
924 - - tests/models/multimodal
925 - no_gpu: true
926 - commands:
927 - - "pip install git+https://github.com/TIGER-AI-Lab/Mantis.git || echo 'Mantis installation skipped (decord not available on CPU-only environment)'"
928 - - pytest -v -s models/multimodal/processing --ignore models/multimodal/processing/test_tensor_schema.py
929 -
930 -- label: Multi-Modal Processor Test
931 - timeout_in_minutes: 60
932 - source_file_dependencies:
933 - - vllm/
934 - - tests/models/multimodal
935 - commands:
936 - - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
937 - - pytest -v -s models/multimodal/processing/test_tensor_schema.py
938 -
939 -- label: Multi-Modal Models Test (Standard) # 60min
940 - timeout_in_minutes: 80
941 - mirror_hardwares: [amdexperimental]
942 - torch_nightly: true
943 - source_file_dependencies:
944 - - vllm/
945 - - tests/models/multimodal
946 - commands:
947 - - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
948 - - pip freeze | grep -E 'torch'
949 - - pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing
950 - - cd .. && VLLM_WORKER_MULTIPROC_METHOD=spawn pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model # Otherwise, mp_method="spawn" doesn't work
951 -
952 -- label: Multi-Modal Accuracy Eval (Small Models) # 50min
953 - timeout_in_minutes: 70
954 - working_dir: "/vllm-workspace/.buildkite/lm-eval-harness"
955 - source_file_dependencies:
956 - - vllm/multimodal/
957 - - vllm/inputs/
958 - - vllm/v1/core/
959 - commands:
960 - - pytest -s -v test_lm_eval_correctness.py --config-list-file=configs/models-mm-small.txt --tp-size=1
961 -
962 -- label: Multi-Modal Models Test (Extended) 1
963 - mirror_hardwares: [amdexperimental]
964 - optional: true
965 - source_file_dependencies:
966 - - vllm/
967 - - tests/models/multimodal
968 - commands:
969 - - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
970 - - pytest -v -s models/multimodal -m 'not core_model' --ignore models/multimodal/generation/test_common.py --ignore models/multimodal/processing
971 -
972 -- label: Multi-Modal Models Test (Extended) 2
973 - mirror_hardwares: [amdexperimental]
974 - optional: true
975 - source_file_dependencies:
976 - - vllm/
977 - - tests/models/multimodal
978 - commands:
979 - - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
980 - - pytest -v -s models/multimodal/generation/test_common.py -m 'split(group=0) and not core_model'
981 -
982 -- label: Multi-Modal Models Test (Extended) 3
983 - mirror_hardwares: [amdexperimental]
984 - optional: true
985 - source_file_dependencies:
986 - - vllm/
987 - - tests/models/multimodal
988 - commands:
989 - - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
990 - - pytest -v -s models/multimodal/generation/test_common.py -m 'split(group=1) and not core_model'
991 -
992 -- label: Quantized Models Test # 45 min
993 - timeout_in_minutes: 60
994 - mirror_hardwares: [amdexperimental]
995 - source_file_dependencies:
996 - - vllm/model_executor/layers/quantization
997 - - tests/models/quantization
998 - commands:
999 - - pytest -v -s models/quantization
1000 -
1001 -# This test is used only in PR development phase to test individual models and should never run on main
1002 -- label: Custom Models Test
1003 - mirror_hardwares: [amdexperimental]
1004 - optional: true
1005 - commands:
1006 - - echo 'Testing custom models...'
1007 - # PR authors can temporarily add commands below to test individual models
1008 - # e.g. pytest -v -s models/encoder_decoder/vision_language/test_mllama.py
1009 - # *To avoid merge conflicts, remember to REMOVE (not just comment out) them before merging the PR*
1010 -
1011 -- label: Transformers Nightly Models Test
1012 - working_dir: "/vllm-workspace/"
1013 - optional: true
1014 - soft_fail: true
1015 - commands:
1016 - - pip install --upgrade git+https://github.com/huggingface/transformers
1017 - - pytest -v -s tests/models/test_initialization.py
1018 - - pytest -v -s tests/models/test_transformers.py
1019 - - pytest -v -s tests/models/multimodal/processing/
1020 - - pytest -v -s tests/models/multimodal/test_mapping.py
1021 - - python3 examples/offline_inference/basic/chat.py
1022 - - python3 examples/offline_inference/vision_language.py --model-type qwen2_5_vl
1023 - # Whisper needs spawn method to avoid deadlock
1024 - - VLLM_WORKER_MULTIPROC_METHOD=spawn python3 examples/offline_inference/audio_language.py --model-type whisper
1025 -
1026 -- label: Blackwell Test # 23 min
1027 - timeout_in_minutes: 30
1028 - working_dir: "/vllm-workspace/"
1029 - gpu: b200
1030 - source_file_dependencies:
1031 - - csrc/quantization/fp4/
1032 - - csrc/attention/mla/
1033 - - csrc/quantization/cutlass_w8a8/moe/
1034 - - vllm/model_executor/layers/fused_moe/cutlass_moe.py
1035 - - vllm/model_executor/layers/fused_moe/flashinfer_cutlass_moe.py
1036 - - vllm/model_executor/layers/fused_moe/flashinfer_a2a_prepare_finalize.py
1037 - - vllm/model_executor/layers/quantization/utils/flashinfer_utils.py
1038 - - vllm/v1/attention/backends/flashinfer.py
1039 - - vllm/v1/attention/backends/mla/cutlass_mla.py
1040 - - vllm/v1/attention/backends/mla/flashinfer_mla.py
1041 - - vllm/v1/attention/selector.py
1042 - - vllm/platforms/cuda.py
1043 - commands:
1044 - - nvidia-smi
1045 - - python3 examples/offline_inference/basic/chat.py
1046 - # Attention
1047 - # num_heads2 broken by https://github.com/flashinfer-ai/flashinfer/issues/1353
1048 - - pytest -v -s tests/kernels/attention/test_attention_selector.py
1049 - - pytest -v -s tests/kernels/attention/test_flashinfer.py -k 'not num_heads2'
1050 - - pytest -v -s tests/kernels/attention/test_flashinfer_trtllm_attention.py
1051 - - pytest -v -s tests/kernels/attention/test_cutlass_mla_decode.py
1052 - - pytest -v -s tests/kernels/attention/test_flashinfer_mla_decode.py
1053 - # Quantization
1054 - - pytest -v -s tests/kernels/quantization/test_cutlass_scaled_mm.py -k 'fp8'
1055 - - pytest -v -s tests/kernels/quantization/test_nvfp4_quant.py
1056 - - pytest -v -s tests/kernels/quantization/test_silu_mul_nvfp4_quant.py
1057 - - pytest -v -s tests/kernels/quantization/test_nvfp4_scaled_mm.py
1058 - - pytest -v -s tests/kernels/quantization/test_flashinfer_scaled_mm.py
1059 - - pytest -v -s tests/kernels/quantization/test_flashinfer_nvfp4_scaled_mm.py
1060 - - pytest -v -s tests/kernels/quantization/test_nvfp4_qutlass.py
1061 - - pytest -v -s tests/kernels/quantization/test_mxfp4_qutlass.py
1062 - - pytest -v -s tests/kernels/moe/test_nvfp4_moe.py
1063 - - pytest -v -s tests/kernels/moe/test_ocp_mx_moe.py
1064 - - pytest -v -s tests/kernels/moe/test_flashinfer.py
1065 - - pytest -v -s tests/kernels/moe/test_cutedsl_moe.py
1066 - # e2e
1067 - - pytest -v -s tests/models/quantization/test_nvfp4.py
1068 -
1069 -- label: Blackwell Fusion and Compile Tests # 30 min
1070 - timeout_in_minutes: 40
1071 - working_dir: "/vllm-workspace/"
1072 - gpu: b200
1073 - source_file_dependencies:
1074 - - csrc/quantization/fp4/
1075 - - vllm/model_executor/layers/quantization/utils/flashinfer_utils.py
1076 - - vllm/v1/attention/backends/flashinfer.py
1077 - - vllm/v1/worker/
1078 - - vllm/v1/cudagraph_dispatcher.py
1079 - - vllm/compilation/
1080 - # can affect pattern matching
1081 - - vllm/model_executor/layers/layernorm.py
1082 - - vllm/model_executor/layers/activation.py
1083 - - vllm/model_executor/layers/quantization/input_quant_fp8.py
1084 - - tests/compile/test_fusion_attn.py
1085 - - tests/compile/test_silu_mul_quant_fusion.py
1086 - - tests/compile/passes/distributed/test_fusion_all_reduce.py
1087 - - tests/compile/fullgraph/test_full_graph.py
1088 - commands:
1089 - - nvidia-smi
1090 - - pytest -v -s tests/compile/test_fusion_attn.py
1091 - - pytest -v -s tests/compile/test_silu_mul_quant_fusion.py
1092 - # this runner has 2 GPUs available even though num_gpus=2 is not set
1093 - - pytest -v -s tests/compile/passes/distributed/test_fusion_all_reduce.py
1094 - # # Limit to Inductor partition, no custom ops, and allreduce & attn fusion to reduce running time
1095 - # # Wrap with quotes to escape yaml
1096 - # - "pytest -v -s tests/compile/distributed/test_fusions_e2e.py::test_tp2_attn_quant_allreduce_rmsnorm -k 'True and not +quant_fp8 and not +rms_norm'"
1097 - # Old E2E tests were removed in https://github.com/vllm-project/vllm/pull/33293
1098 - # in favor of new tests in fusions_e2e. We avoid replicating the new jobs in this file as it's deprecated.
1099 -
1100 - # test_fp8_kv_scale_compile requires FlashAttention (not supported on default L4/L40)
1101 - - pytest -v -s tests/compile/fullgraph/test_full_graph.py::test_fp8_kv_scale_compile
1102 -
1103 -- label: Blackwell GPT-OSS Eval
1104 - timeout_in_minutes: 60
1105 - working_dir: "/vllm-workspace/"
1106 - gpu: b200
1107 - optional: true # run on nightlies
1108 - source_file_dependencies:
1109 - - tests/evals/gpt_oss
1110 - - vllm/model_executor/models/gpt_oss.py
1111 - - vllm/model_executor/layers/quantization/mxfp4.py
1112 - - vllm/v1/attention/backends/flashinfer.py
1113 - commands:
1114 - - uv pip install --system 'gpt-oss[eval]==0.0.5'
1115 - - pytest -s -v tests/evals/gpt_oss/test_gpqa_correctness.py --model openai/gpt-oss-20b --metric 0.58
1116 -
1117 -- label: Blackwell Quantized MoE Test
1118 - timeout_in_minutes: 60
1119 - working_dir: "/vllm-workspace/"
1120 - gpu: b200
1121 - source_file_dependencies:
1122 - - tests/quantization/test_blackwell_moe.py
1123 - - vllm/model_executor/models/deepseek_v2.py
1124 - - vllm/model_executor/models/gpt_oss.py
1125 - - vllm/model_executor/models/llama4.py
1126 - - vllm/model_executor/layers/fused_moe
1127 - - vllm/model_executor/layers/quantization/compressed_tensors
1128 - - vllm/model_executor/layers/quantization/modelopt.py
1129 - - vllm/model_executor/layers/quantization/mxfp4.py
1130 - - vllm/v1/attention/backends/flashinfer.py
1131 - commands:
1132 - - pytest -s -v tests/quantization/test_blackwell_moe.py
1133 -
1134 -- label: Blackwell LM Eval Small Models
1135 - timeout_in_minutes: 120
1136 - gpu: b200
1137 - optional: true # run on nightlies
1138 - source_file_dependencies:
1139 - - csrc/
1140 - - vllm/model_executor/layers/quantization
1141 - commands:
1142 - - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-blackwell.txt
1143 -
1144 -##### 1 GPU test #####
1145 -##### multi gpus test #####
1146 -
1147 -- label: Distributed Comm Ops Test # 7min
1148 - timeout_in_minutes: 20
1149 - mirror_hardwares: [amdexperimental]
1150 - working_dir: "/vllm-workspace/tests"
1151 - num_gpus: 2
1152 - source_file_dependencies:
1153 - - vllm/distributed
1154 - - tests/distributed
1155 - commands:
1156 - - pytest -v -s distributed/test_comm_ops.py
1157 - - pytest -v -s distributed/test_shm_broadcast.py
1158 - - pytest -v -s distributed/test_shm_buffer.py
1159 - - pytest -v -s distributed/test_shm_storage.py
1160 - - pytest -v -s distributed/test_packed_tensor.py
1161 - - pytest -v -s distributed/test_weight_transfer.py
1162 -
1163 -- label: 2 Node Tests (4 GPUs in total) # 16min
1164 - timeout_in_minutes: 30
1165 - mirror_hardwares: [amdexperimental]
1166 - working_dir: "/vllm-workspace/tests"
1167 - num_gpus: 2
1168 - num_nodes: 2
1169 - source_file_dependencies:
1170 - - vllm/distributed/
1171 - - vllm/engine/
1172 - - vllm/executor/
1173 - - vllm/model_executor/models/
1174 - - tests/distributed/
1175 - - tests/examples/offline_inference/data_parallel.py
1176 - - .buildkite/scripts/run-multi-node-test.sh
1177 - commands:
1178 - - # the following commands are for the first node, with ip 192.168.10.10 (ray environment already set up)
1179 - - VLLM_TEST_SAME_HOST=0 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_same_node.py | grep 'Same node test passed'
1180 - - NUM_NODES=2 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_node_count.py | grep 'Node count test passed'
1181 - - python3 ../examples/offline_inference/data_parallel.py -dp=2 -tp=1 --dp-num-nodes=2 --dp-node-rank=0 --dp-master-addr=192.168.10.10 --dp-master-port=12345 --enforce-eager --trust-remote-code
1182 - - VLLM_MULTI_NODE=1 pytest -v -s distributed/test_multi_node_assignment.py
1183 - - VLLM_MULTI_NODE=1 pytest -v -s distributed/test_pipeline_parallel.py
1184 - - # the following commands are for the second node, with ip 192.168.10.11 (ray environment already set up)
1185 - - VLLM_TEST_SAME_HOST=0 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_same_node.py | grep 'Same node test passed'
1186 - - NUM_NODES=2 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_node_count.py | grep 'Node count test passed'
1187 - - python3 ../examples/offline_inference/data_parallel.py -dp=2 -tp=1 --dp-num-nodes=2 --dp-node-rank=1 --dp-master-addr=192.168.10.10 --dp-master-port=12345 --enforce-eager --trust-remote-code
1188 -
1189 -- label: Distributed Tests (2 GPUs) # 68min
1190 - timeout_in_minutes: 90
1191 - mirror_hardwares: [amdexperimental]
1192 - working_dir: "/vllm-workspace/tests"
1193 - num_gpus: 2
1194 - source_file_dependencies:
1195 - - vllm/compilation/
1196 - - vllm/distributed/
1197 - - vllm/engine/
1198 - - vllm/executor/
1199 - - vllm/worker/worker_base.py
1200 - - vllm/v1/engine/
1201 - - vllm/v1/worker/
1202 - - tests/compile/fullgraph/test_basic_correctness.py
1203 - - tests/compile/test_wrapper.py
1204 - - tests/distributed/
1205 - - tests/entrypoints/llm/test_collective_rpc.py
1206 - - tests/v1/distributed
1207 - - tests/v1/entrypoints/openai/test_multi_api_servers.py
1208 - - tests/v1/shutdown
1209 - - tests/v1/worker/test_worker_memory_snapshot.py
1210 - commands:
1211 - # https://github.com/NVIDIA/nccl/issues/1838
1212 - - export NCCL_CUMEM_HOST_ENABLE=0
1213 - - TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_async_llm_dp.py
1214 - - TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_eagle_dp.py
1215 - - TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_external_lb_dp.py
1216 - - DP_SIZE=2 pytest -v -s v1/entrypoints/openai/test_multi_api_servers.py
1217 - - pytest -v -s entrypoints/llm/test_collective_rpc.py
1218 - - pytest -v -s ./compile/fullgraph/test_basic_correctness.py
1219 - - pytest -v -s ./compile/test_wrapper.py
1220 - - VLLM_TEST_SAME_HOST=1 torchrun --nproc-per-node=4 distributed/test_same_node.py | grep 'Same node test passed'
1221 - - VLLM_TEST_SAME_HOST=1 VLLM_TEST_WITH_DEFAULT_DEVICE_SET=1 torchrun --nproc-per-node=4 distributed/test_same_node.py | grep 'Same node test passed'
1222 - - pytest -v -s distributed/test_sequence_parallel.py
1223 - - CUDA_VISIBLE_DEVICES=0,1 pytest -v -s v1/shutdown
1224 - - pytest -v -s v1/worker/test_worker_memory_snapshot.py
1225 -
1226 -- label: Distributed Model Tests (2 GPUs) # 37min
1227 - timeout_in_minutes: 50
1228 - mirror_hardwares: [amdexperimental]
1229 - working_dir: "/vllm-workspace/tests"
1230 - num_gpus: 2
1231 - source_file_dependencies:
1232 - - vllm/model_executor/model_loader/sharded_state_loader.py
1233 - - vllm/model_executor/models/
1234 - - tests/basic_correctness/
1235 - - tests/model_executor/model_loader/test_sharded_state_loader.py
1236 - - tests/models/
1237 - commands:
1238 - - TARGET_TEST_SUITE=L4 pytest basic_correctness/ -v -s -m 'distributed(num_gpus=2)'
1239 - - CUDA_VISIBLE_DEVICES=0,1 pytest -v -s model_executor/model_loader/test_sharded_state_loader.py
1240 - # Avoid importing model tests that cause CUDA reinitialization error
1241 - - pytest models/test_transformers.py -v -s -m 'distributed(num_gpus=2)'
1242 - - pytest models/language -v -s -m 'distributed(num_gpus=2)'
1243 - - pytest models/multimodal -v -s -m 'distributed(num_gpus=2)' --ignore models/multimodal/generation/test_whisper.py
1244 - - VLLM_WORKER_MULTIPROC_METHOD=spawn pytest models/multimodal/generation/test_whisper.py -v -s -m 'distributed(num_gpus=2)'
1245 -
1246 -- label: Plugin Tests (2 GPUs) # 40min
1247 - timeout_in_minutes: 60
1248 - mirror_hardwares: [amdexperimental]
1249 - working_dir: "/vllm-workspace/tests"
1250 - num_gpus: 2
1251 - source_file_dependencies:
1252 - - vllm/plugins/
1253 - - tests/plugins/
1254 - commands:
1255 - # begin platform plugin and general plugin tests, all the code in-between runs on dummy platform
1256 - - pip install -e ./plugins/vllm_add_dummy_platform
1257 - - pytest -v -s plugins_tests/test_platform_plugins.py
1258 - - pip uninstall vllm_add_dummy_platform -y
1259 - # end platform plugin tests
1260 - # begin io_processor plugins test, all the code in between uses the prithvi_io_processor plugin
1261 - - pip install -e ./plugins/prithvi_io_processor_plugin
1262 - - pytest -v -s plugins_tests/test_io_processor_plugins.py
1263 - - pip uninstall prithvi_io_processor_plugin -y
1264 - # end io_processor plugins test
1265 - # begin stat_logger plugins test
1266 - - pip install -e ./plugins/vllm_add_dummy_stat_logger
1267 - - pytest -v -s plugins_tests/test_stats_logger_plugins.py
1268 - - pip uninstall dummy_stat_logger -y
1269 - # end stat_logger plugins test
1270 - # other tests continue here:
1271 - - pytest -v -s plugins_tests/test_scheduler_plugins.py
1272 - - pip install -e ./plugins/vllm_add_dummy_model
1273 - - pytest -v -s distributed/test_distributed_oot.py
1274 - - pytest -v -s entrypoints/openai/test_oot_registration.py # it needs a clean process
1275 - - pytest -v -s models/test_oot_registration.py # it needs a clean process
1276 - - pytest -v -s plugins/lora_resolvers # unit tests for lora resolver plugins
1277 -
1278 -- label: Pipeline + Context Parallelism Test # 45min
1279 - timeout_in_minutes: 60
1280 - mirror_hardwares: [amdexperimental]
1281 - working_dir: "/vllm-workspace/tests"
1282 - num_gpus: 4
1283 - source_file_dependencies:
1284 - - vllm/distributed/
1285 - - vllm/engine/
1286 - - vllm/executor/
1287 - - vllm/model_executor/models/
1288 - - tests/distributed/
1289 - commands:
1290 - - pytest -v -s distributed/test_pp_cudagraph.py
1291 - - pytest -v -s distributed/test_pipeline_parallel.py
1292 -
1293 -- label: LoRA TP Test (Distributed) # 17 min
1294 - timeout_in_minutes: 30
1295 - mirror_hardwares: [amdexperimental]
1296 - num_gpus: 4
1297 - source_file_dependencies:
1298 - - vllm/lora
1299 - - tests/lora
1300 - commands:
1301 - # FIXIT: find out which code initialize cuda before running the test
1302 - # before the fix, we need to use spawn to test it
1303 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
1304 - # Alot of these tests are on the edge of OOMing
1305 - - export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
1306 - # There is some Tensor Parallelism related processing logic in LoRA that
1307 - # requires multi-GPU testing for validation.
1308 - - pytest -v -s -x lora/test_chatglm3_tp.py
1309 - - pytest -v -s -x lora/test_llama_tp.py
1310 - - pytest -v -s -x lora/test_llm_with_multi_loras.py
1311 - - pytest -v -s -x lora/test_olmoe_tp.py
1312 - - pytest -v -s -x lora/test_gptoss_tp.py
1313 -
1314 -
1315 -- label: Weight Loading Multiple GPU Test # 33min
1316 - timeout_in_minutes: 45
1317 - mirror_hardwares: [amdexperimental]
1318 - working_dir: "/vllm-workspace/tests"
1319 - num_gpus: 2
1320 - optional: true
1321 - source_file_dependencies:
1322 - - vllm/
1323 - - tests/weight_loading
1324 - commands:
1325 - - bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models.txt
1326 -
1327 -- label: Weight Loading Multiple GPU Test - Large Models # optional
1328 - mirror_hardwares: [amdexperimental]
1329 - working_dir: "/vllm-workspace/tests"
1330 - num_gpus: 2
1331 - gpu: a100
1332 - optional: true
1333 - source_file_dependencies:
1334 - - vllm/
1335 - - tests/weight_loading
1336 - commands:
1337 - - bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models-large.txt
1338 -
1339 -- label: NixlConnector PD accuracy tests (Distributed) # 40min
1340 - timeout_in_minutes: 40
1341 - working_dir: "/vllm-workspace/tests"
1342 - num_gpus: 4
1343 - source_file_dependencies:
1344 - - vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py
1345 - - tests/v1/kv_connector/nixl_integration/
1346 - commands:
1347 - - uv pip install --system -r /vllm-workspace/requirements/kv_connectors.txt
1348 - - bash v1/kv_connector/nixl_integration/config_sweep_accuracy_test.sh
1349 -
1350 -- label: DP EP NixlConnector PD accuracy tests (Distributed) # 15min
1351 - timeout_in_minutes: 15
1352 - working_dir: "/vllm-workspace/tests"
1353 - num_gpus: 4
1354 - source_file_dependencies:
1355 - - vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py
1356 - - tests/v1/kv_connector/nixl_integration/
1357 - commands:
1358 - - uv pip install --system -r /vllm-workspace/requirements/kv_connectors.txt
1359 - - DP_EP=1 bash v1/kv_connector/nixl_integration/config_sweep_accuracy_test.sh
1360 -
1361 -
1362 -##### multi gpus test #####
1363 -##### A100 test #####
1364 -
1365 -- label: Distributed Tests (A100) # optional
1366 - gpu: a100
1367 - optional: true
1368 - num_gpus: 4
1369 - source_file_dependencies:
1370 - - vllm/
1371 - commands:
1372 - # NOTE: don't test llama model here, it seems hf implementation is buggy
1373 - # see https://github.com/vllm-project/vllm/pull/5689 for details
1374 - - pytest -v -s distributed/test_custom_all_reduce.py
1375 - - torchrun --nproc_per_node=2 distributed/test_ca_buffer_sharing.py
1376 - - TARGET_TEST_SUITE=A100 pytest basic_correctness/ -v -s -m 'distributed(num_gpus=2)'
1377 - - pytest -v -s -x lora/test_mixtral.py
1378 -
1379 -- label: Acceptance Length Test (Large Models) # optional
1380 - timeout_in_minutes: 120
1381 - gpu: h100
1382 - optional: true
1383 - num_gpus: 1
1384 - working_dir: "/vllm-workspace/tests"
1385 - source_file_dependencies:
1386 - - vllm/v1/spec_decode/
1387 - - vllm/model_executor/models/mlp_speculator.py
1388 - - tests/v1/spec_decode/test_acceptance_length.py
1389 - commands:
1390 - - export VLLM_ALLOW_INSECURE_SERIALIZATION=1
1391 - - pytest -v -s v1/spec_decode/test_acceptance_length.py -m slow_test
1392 -
1393 -- label: LM Eval Large Models # optional
1394 - gpu: a100
1395 - optional: true
1396 - num_gpus: 4
1397 - working_dir: "/vllm-workspace/.buildkite/lm-eval-harness"
1398 - source_file_dependencies:
1399 - - csrc/
1400 - - vllm/model_executor/layers/quantization
1401 - commands:
1402 - - export VLLM_WORKER_MULTIPROC_METHOD=spawn
1403 - - pytest -s -v test_lm_eval_correctness.py --config-list-file=configs/models-large.txt --tp-size=4
1404 -
1405 -##### H100 test #####
1406 -- label: LM Eval Large Models (H100) # optional
1407 - gpu: h100
1408 - optional: true
1409 - num_gpus: 4
1410 - working_dir: "/vllm-workspace/.buildkite/lm-eval-harness"
1411 - source_file_dependencies:
1412 - - csrc/
1413 - - vllm/model_executor/layers/quantization
1414 - commands:
1415 - - export VLLM_USE_DEEP_GEMM=0 # We found Triton is faster than DeepGEMM for H100
1416 - - pytest -s -v test_lm_eval_correctness.py --config-list-file=configs/models-large-hopper.txt --tp-size=4
1417 -
1418 -- label: Sequence Parallel Tests (H100) # 60 min
1419 - timeout_in_minutes: 60
1420 - working_dir: "/vllm-workspace/"
1421 - gpu: h100
1422 - optional: true
1423 - num_gpus: 2
1424 - commands:
1425 - - export VLLM_TEST_CLEAN_GPU_MEMORY=1
1426 - # Run sequence parallel tests
1427 - - pytest -v -s tests/compile/correctness_e2e/test_sequence_parallel.py
1428 - - pytest -v -s tests/compile/passes/distributed/test_sequence_parallelism.py
1429 -
1430 -- label: Distributed Tests (H100) # optional
1431 - gpu: h100
1432 - optional: true
1433 - working_dir: "/vllm-workspace/"
1434 - num_gpus: 2
1435 - commands:
1436 - - VLLM_TEST_CLEAN_GPU_MEMORY=1 pytest -v -s tests/compile/passes/distributed/test_async_tp.py
1437 - - pytest -v -s tests/distributed/test_context_parallel.py
1438 - - VLLM_USE_DEEP_GEMM=1 VLLM_LOGGING_LEVEL=DEBUG python3 examples/offline_inference/data_parallel.py --model=Qwen/Qwen1.5-MoE-A2.7B -tp=1 -dp=2 --max-model-len=2048 --all2all-backend=deepep_high_throughput
1439 - - pytest -v -s tests/v1/distributed/test_dbo.py
1440 -
1441 -##### H200 test #####
1442 -
1443 -- label: LM Eval Large Models (H200) # optional
1444 - timeout_in_minutes: 60
1445 - gpu: h200
1446 - optional: true
1447 - num_gpus: 8
1448 - commands:
1449 - - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-h200.txt
1450 -
1451 -##### B200 test #####
1452 -- label: Distributed Tests (B200) # optional
1453 - gpu: b200
1454 - optional: true
1455 - working_dir: "/vllm-workspace/"
1456 - num_gpus: 2
1457 - commands:
1458 - - pytest -v -s tests/distributed/test_context_parallel.py
1459 - - pytest -v -s tests/distributed/test_nccl_symm_mem_allreduce.py
1460 - - pytest -v -s tests/v1/distributed/test_dbo.py
1461 -
1462 -##### RL Integration Tests #####
1463 -- label: Prime-RL Integration Test # 15min
1464 - timeout_in_minutes: 30
1465 - optional: true
1466 - soft_fail: true
1467 - num_gpus: 2
1468 - working_dir: "/vllm-workspace"
1469 - source_file_dependencies:
1470 - - vllm/
1471 - - .buildkite/scripts/run-prime-rl-test.sh
1472 - commands:
1473 - - nvidia-smi
1474 - - bash .buildkite/scripts/run-prime-rl-test.sh
1475 -
1476 -- label: DeepSeek V2-Lite Accuracy
1477 - timeout_in_minutes: 60
1478 - gpu: h100
1479 - optional: true
1480 - num_gpus: 4
1481 - working_dir: "/vllm-workspace"
1482 - commands:
1483 - - bash .buildkite/scripts/scheduled_integration_test/deepseek_v2_lite_ep_eplb.sh 0.25 200 8010
1484 -
1485 -- label: Qwen3-30B-A3B-FP8-block Accuracy (H100)
1486 - timeout_in_minutes: 60
1487 - gpu: h100
1488 - optional: true
1489 - num_gpus: 4
1490 - working_dir: "/vllm-workspace"
1491 - commands:
1492 - - bash .buildkite/scripts/scheduled_integration_test/qwen30b_a3b_fp8_block_ep_eplb.sh 0.8 200 8020
1493 -
1494 -- label: Qwen3-30B-A3B-FP8-block Accuracy (B200)
1495 - timeout_in_minutes: 60
1496 - gpu: b200
1497 - optional: true
1498 - num_gpus: 2
1499 - working_dir: "/vllm-workspace"
1500 - commands:
1501 - - bash .buildkite/scripts/scheduled_integration_test/qwen30b_a3b_fp8_block_ep_eplb.sh 0.8 200 8020 2 1
1502 -
1503 -##### MoE Refactor (Temporary) Tests #####
1504 -
1505 -- label: MoE Refactor Integration Test (H100 - TEMPORARY) # optional
1506 - gpu: h100
1507 - optional: true
1508 - num_gpus: 2
1509 - commands:
1510 - - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=evals/gsm8k/configs/moe-refactor/config-h100.txt
1511 -
1512 -- label: MoE Refactor Integration Test (B200 - TEMPORARY) # optional
1513 - gpu: b200
1514 - optional: true
1515 - num_gpus: 2
1516 - commands:
1517 - - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=evals/gsm8k/configs/moe-refactor/config-b200.txt
1518 -
1519 -- label: MoE Refactor Integration Test (B200 DP - TEMPORARY) # optional
1520 - gpu: b200
1521 - optional: true
1522 - num_gpus: 2
1523 - commands:
1524 - - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=evals/gsm8k/configs/moe-refactor-dp-ep/config-b200.txt
8 +# If you need to make changes to CI, please find the relevant file in these directories and make changes there.