[ROCm][CI] Shard MI300 Multimodal Processor (#57056)
Signed-off-by: aarushjain29 <Aarushi.Jain2@amd.com> Co-authored-by: Andreas Karatzas <akaratza@amd.com>
Aarushi Jain committed
Sep 16, 2026 at 18:55 UTC
91b96a533cfb06579a360d3aefb2c8341fc89bdb
1 file changed
+3
-2
.buildkite/test-amd.yaml
+3
-2
@@ -2175,19 +2175,20 @@ steps:
2175
- pytest -v -s models/multimodal/generation/test_memory_leak.py -m core_model
2176
- cd .. && VLLM_WORKER_MULTIPROC_METHOD=spawn pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model
2177
2178
-- label: ":amd: (MI300) Multimodal Processor" # 1h 42m
2178
+- label: ":amd: (MI300) Multimodal Processor Shard %N" # TBD
2179
timeout_in_minutes: 110
2180
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
2181
dind: false
2182
agent_pool: mi300_1
2183
optional: true
2184
+ parallelism: 4
2185
working_dir: "/vllm-workspace/tests"
2186
source_file_dependencies:
2187
- vllm/
2188
- tests/models/multimodal
2189
- tests/models/registry.py
2190
commands:
2190
- - pytest -v -s models/multimodal/processing/test_tensor_schema.py
2191
+ - pytest -v -s models/multimodal/processing/test_tensor_schema.py --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
2192
2193
- label: ":amd: (MI300) Multimodal Models (Extended Pooling)" # TBD
2194
timeout_in_minutes: 75