From 694db075f8af1ccb026720782bc0e2400a3f5c61 Mon Sep 17 00:00:00 2001 From: "Kevin H. Luu" Date: Fri, 14 Aug 2026 15:11:48 -0700 Subject: [PATCH] [CI] Shard multimodal extended generation 2 (#52323) Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> --- .buildkite/test_areas/models_multimodal.yaml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/.buildkite/test_areas/models_multimodal.yaml b/.buildkite/test_areas/models_multimodal.yaml index 4a84abd5ba..2ea3621342 100644 --- a/.buildkite/test_areas/models_multimodal.yaml +++ b/.buildkite/test_areas/models_multimodal.yaml @@ -169,16 +169,17 @@ steps: depends_on: - image-build-amd -- label: Multi-Modal Models (Extended Generation 2) +- label: Multi-Modal Models (Extended Generation 2) %N device: h200_35gb key: multi-modal-models-extended-generation-2 + parallelism: 4 optional: true source_file_dependencies: - vllm/ - "!vllm/distributed/kv_transfer/" - tests/models/multimodal/generation commands: - - pytest -v -s models/multimodal/generation/test_common.py -m 'split(group=0) and not core_model' + - pytest -v -s models/multimodal/generation/test_common.py -m 'split(group=0) and not core_model' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB - label: Multi-Modal Models (Extended Generation 3) device: h200_35gb