[CI] Shard multimodal Processor 1->4 (#56316)
Signed-off-by: Thang Nguyen <thang.nguyen@inferact.ai> Co-authored-by: Thang Nguyen <thang.nguyen@inferact.ai> Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com> Co-authored-by: Kimi Code <noreply@moonshot.ai>
Thang Nguyen committed
Sep 17, 2026 at 14:11 UTC
668d6c3a778b9ac525f78d39b0c413c8604235d1
1 file changed
+5
-3
.buildkite/test_areas/models_multimodal.yaml
+5
-3
@@ -109,22 +109,24 @@ steps:
109
depends_on:
110
- image-build-amd
111
112
-- label: ":nvidia: (H200 MIG 18GB) Multimodal Processor"
112
+- label: ":nvidia: (H200 MIG 18GB) Multimodal Processor Shard %N"
113
key: multi-modal-processor
114
timeout_in_minutes: 98
115
device: h200_18gb
116
+ parallelism: 4
117
source_file_dependencies:
118
- vllm/
119
- "!vllm/distributed/kv_transfer/"
120
- tests/models/multimodal
121
- tests/models/registry.py
122
commands:
122
- - pytest -v -s models/multimodal/processing/test_tensor_schema.py
123
+ - pytest -v -s models/multimodal/processing/test_tensor_schema.py --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
124
mirror:
125
amd:
125
- label: ":amd: (MI355) Multimodal Processor"
126
+ label: ":amd: (MI355) Multimodal Processor Shard %N"
127
dind: false
128
device: mi355_1
129
+ parallelism: 4
130
timeout_in_minutes: 115
131
depends_on:
132
- image-build-amd