[ROCm][CI] Shard MI300 Entrypoints Integration (Pooling) (#57583)
Signed-off-by: aarushjain29 <Aarushi.Jain2@amd.com>
Aarushi Jain committed
Sep 18, 2026 at 16:16 UTC
0390299309c5bc11b8aa78ea0f1d10993273dbb4
1 file changed
+3
-17
.buildkite/test-amd.yaml
+3
-17
@@ -1341,21 +1341,6 @@ steps:
1341
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
1342
- pytest -v -s entrypoints/multimodal
1343
1344
-- label: ":amd: (MI300) Entrypoints Integration (Pooling)" # TBD
1345
- timeout_in_minutes: 65
1346
- mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1347
- dind: false
1348
- agent_pool: mi300_1
1349
- optional: true
1350
- fast_check: true
1351
- working_dir: "/vllm-workspace/tests"
1352
- source_file_dependencies:
1353
- - vllm/
1354
- - tests/entrypoints/pooling
1355
- commands:
1356
- - export VLLM_WORKER_MULTIPROC_METHOD=spawn
1357
- - pytest -v -s entrypoints/pooling
1358
-
1344
- label: ":amd: (MI300) OpenAI API Correctness" # TBD
1345
timeout_in_minutes: 30
1346
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
@@ -3615,12 +3600,13 @@ steps:
3600
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
3601
- pytest -v -s entrypoints/multimodal
3602
3618
-- label: ":amd: (MI355) Entrypoints Integration (Pooling)" # TBD
3603
+- label: ":amd: (MI355) Entrypoints Integration (Pooling) Shard %N" # TBD
3604
timeout_in_minutes: 65
3605
mirror_hardwares: [amdexperimental, amdproduction, amdgfx950nightly, amdmi355]
3606
dind: false
3607
agent_pool: mi355_1
3608
fast_check: true
3609
+ parallelism: 4
3610
working_dir: "/vllm-workspace/tests"
3611
source_file_dependencies:
3612
- vllm/
@@ -3628,7 +3614,7 @@ steps:
3614
optional: true
3615
commands:
3616
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
3631
- - pytest -v -s entrypoints/pooling
3617
+ - pytest -v -s entrypoints/pooling --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
3618
3619
#----------------------------------------------------------- mi355 · evals -----------------------------------------------------------#
3620