[ROCm][CI] Increasing parallelism in Basic Models Tests (Extra Initialization) (#47591)
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Andreas Karatzas committed
Jul 6, 2026 at 02:06 UTC
fb265fc8fb4e35b5160938bb2ad909be0e0ab4d3
1 file changed
+6
-6
.buildkite/test-amd.yaml
+6
-6
@@ -1301,7 +1301,7 @@ steps:
1301
#---------------------------------------------------------- mi300 · kernels ----------------------------------------------------------#
1302
1303
- label: vLLM IR Tests # TBD
1304
- timeout_in_minutes: 30
1304
+ timeout_in_minutes: 180
1305
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1306
agent_pool: mi300_1
1307
optional: true
@@ -1350,7 +1350,7 @@ steps:
1350
- pytest -v -s kernels/core --ignore=kernels/core/test_minimax_reduce_rms.py kernels/test_concat_mla_q.py kernels/test_top_k_per_row.py
1351
1352
- label: Kernels KDA Test # TBD
1353
- timeout_in_minutes: 30
1353
+ timeout_in_minutes: 180
1354
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1355
agent_pool: mi300_1
1356
optional: true
@@ -1558,7 +1558,7 @@ steps:
1558
- TP_SIZE=1 DP_SIZE=2 pytest -v -s v1/distributed/test_eagle_dp.py
1559
1560
- label: Model Runner V2 Pipeline Parallelism (4 GPUs) # TBD
1561
- timeout_in_minutes: 60
1561
+ timeout_in_minutes: 180
1562
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1563
agent_pool: mi300_4
1564
num_gpus: 4
@@ -1577,7 +1577,7 @@ steps:
1577
- pytest -v -s distributed/test_pp_cudagraph.py -k "not ray"
1578
1579
- label: Model Runner V2 Spec Decode # TBD
1580
- timeout_in_minutes: 45
1580
+ timeout_in_minutes: 180
1581
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1582
agent_pool: mi300_1
1583
optional: true
@@ -1604,7 +1604,7 @@ steps:
1604
timeout_in_minutes: 180
1605
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1606
agent_pool: mi300_1
1607
- parallelism: 2
1607
+ parallelism: 6
1608
working_dir: "/vllm-workspace/tests"
1609
source_file_dependencies:
1610
- vllm/model_executor/models/
@@ -1915,7 +1915,7 @@ steps:
1915
- pytest -v -s plugins_tests/lora_resolvers # unit tests for in-tree lora resolver plugins
1916
1917
- label: GGUF Plugin # TBD
1918
- timeout_in_minutes: 30
1918
+ timeout_in_minutes: 180
1919
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1920
agent_pool: mi300_1
1921
optional: true