Skip to content

Commit 772ce5a

Browse files
[Misc] Add dummy maverick test to CI (#21324)
Signed-off-by: Ming Yang <[email protected]> Co-authored-by: Cyrus Leung <[email protected]>
1 parent 63d92ab commit 772ce5a

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -718,6 +718,7 @@ steps:
718718
- VLLM_USE_V1=0 CUDA_VISIBLE_DEVICES=0,1 pytest -v -s test_sharded_state_loader.py
719719
- VLLM_USE_V1=0 CUDA_VISIBLE_DEVICES=0,1 pytest -v -s kv_transfer/test_disagg.py
720720
- CUDA_VISIBLE_DEVICES=0,1 pytest -v -s v1/shutdown
721+
- pytest -v -s models/multimodal/generation/test_maverick.py
721722

722723
- label: Plugin Tests (2 GPUs) # 40min
723724
mirror_hardwares: [amdexperimental]

tests/models/multimodal/generation/test_maverick.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323

2424
from vllm import LLM, SamplingParams
2525

26+
from ....utils import multi_gpu_test
27+
2628
# Sample prompts for testing
2729
PROMPTS: list[str] = [
2830
"Hello, my name is",
@@ -541,6 +543,7 @@ def run_reduced_model(model_path: str,
541543
print("-" * 40)
542544

543545

546+
@multi_gpu_test(num_gpus=2)
544547
@pytest.mark.parametrize(
545548
"original_model_name,text_layers,num_experts,vision_layers,",
546549
[("meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8", 4, 4, 2)])

0 commit comments

Comments
 (0)