Skip to content

Commit 60bc25e

Browse files
authored
[CI] Add Blackwell LM Eval Small Models test to nightly (vllm-project#26052)
Signed-off-by: mgoin <[email protected]>
1 parent b893d66 commit 60bc25e

File tree

2 files changed

+16
-2
lines changed

2 files changed

+16
-2
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -835,11 +835,11 @@ steps:
835835
- pytest -v -s tests/kernels/moe/test_flashinfer.py
836836
- pytest -v -s tests/compile/test_silu_mul_quant_fusion.py
837837

838-
- label: GPT-OSS Eval (Blackwell)
838+
- label: Blackwell GPT-OSS Eval
839839
timeout_in_minutes: 60
840840
working_dir: "/vllm-workspace/"
841841
gpu: b200
842-
optional: true # disable while debugging
842+
optional: true # run on nightlies
843843
source_file_dependencies:
844844
- tests/evals/gpt_oss
845845
- vllm/model_executor/models/gpt_oss.py
@@ -866,6 +866,16 @@ steps:
866866
commands:
867867
- pytest -s -v tests/quantization/test_blackwell_moe.py
868868

869+
- label: Blackwell LM Eval Small Models
870+
timeout_in_minutes: 75
871+
gpu: b200
872+
optional: true # run on nightlies
873+
source_file_dependencies:
874+
- csrc/
875+
- vllm/model_executor/layers/quantization
876+
commands:
877+
- pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-blackwell.txt --tp-size=1
878+
869879
##### 1 GPU test #####
870880
##### multi gpus test #####
871881

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
Qwen3-0.6B-FP8.yaml
2+
Qwen2.5-VL-3B-Instruct-FP8-dynamic.yaml
3+
Qwen1.5-MoE-W4A16-CT.yaml
4+
DeepSeek-V2-Lite-Instruct-FP8.yaml

0 commit comments

Comments
 (0)