[CI] Shard multimodal extended generation 2 (#52323)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com>
Kevin H. Luu committed
Aug 14, 2026 at 15:11 UTC
694db075f8af1ccb026720782bc0e2400a3f5c61
1 file changed
+3
-2
.buildkite/test_areas/models_multimodal.yaml
+3
-2
@@ -169,16 +169,17 @@ steps:
169
depends_on:
170
- image-build-amd
171
172
-- label: Multi-Modal Models (Extended Generation 2)
172
+- label: Multi-Modal Models (Extended Generation 2) %N
173
device: h200_35gb
174
key: multi-modal-models-extended-generation-2
175
+ parallelism: 4
176
optional: true
177
source_file_dependencies:
178
- vllm/
179
- "!vllm/distributed/kv_transfer/"
180
- tests/models/multimodal/generation
181
commands:
181
- - pytest -v -s models/multimodal/generation/test_common.py -m 'split(group=0) and not core_model'
182
+ - 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
183
184
- label: Multi-Modal Models (Extended Generation 3)
185
device: h200_35gb