[Hardware][AMD][CI] Mirror Basic Models (Others) and Weight Loading Multiple GPU test groups (#46668)
Signed-off-by: Matthew Wong <Matthew.Wong2@amd.com>
Matt committed
Jun 25, 2026 at 08:25 UTC
2365b7a8e7c0f6c55eeb820fba43416f880d3a51
3 files changed
+18
-6
.buildkite/test-amd.yaml
+6
-6
@@ -1588,10 +1588,10 @@ steps:
1588
- pytest -v -s models/test_initialization.py::test_can_initialize_small_subset
1589
1590
- label: Basic Models Tests (Other) # TBD
1591
- timeout_in_minutes: 180
1591
+ timeout_in_minutes: 45
1592
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1593
agent_pool: mi300_1
1594
- torch_nightly: true
1594
+ optional: true
1595
working_dir: "/vllm-workspace/tests"
1596
source_file_dependencies:
1597
- vllm/
@@ -2393,7 +2393,7 @@ steps:
2393
#------------------------------------------------------ mi300 · weight_loading -------------------------------------------------------#
2394
2395
- label: Weight Loading Multiple GPU # TBD
2396
- timeout_in_minutes: 180
2396
+ timeout_in_minutes: 30
2397
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
2398
agent_pool: mi300_2
2399
num_gpus: 2
@@ -2405,7 +2405,7 @@ steps:
2405
- bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models-amd.txt
2406
2407
- label: Weight Loading Multiple GPU - Large Models # TBD
2408
- timeout_in_minutes: 180
2408
+ timeout_in_minutes: 30
2409
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
2410
agent_pool: mi300_2
2411
num_gpus: 2
@@ -3318,7 +3318,7 @@ steps:
3318
#------------------------------------------------------ mi355 · weight_loading -------------------------------------------------------#
3319
3320
- label: Weight Loading Multiple GPU # TBD
3321
- timeout_in_minutes: 180
3321
+ timeout_in_minutes: 30
3322
mirror_hardwares: [amdexperimental, amdproduction, amdgfx950nightly, amdmi355]
3323
agent_pool: mi355_2
3324
num_gpus: 2
@@ -3330,7 +3330,7 @@ steps:
3330
- bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models-amd.txt
3331
3332
- label: Weight Loading Multiple GPU - Large Models # TBD
3333
- timeout_in_minutes: 180
3333
+ timeout_in_minutes: 30
3334
mirror_hardwares: [amdexperimental, amdproduction, amdgfx950nightly, amdmi355]
3335
agent_pool: mi355_2
3336
working_dir: "/vllm-workspace/tests"
.buildkite/test_areas/models_basic.yaml
+5
@@ -45,6 +45,11 @@ steps:
45
- tests/models/test_registry.py
46
commands:
47
- pytest -v -s models/test_terratorch.py models/test_transformers.py models/test_registry.py
48
+ mirror:
49
+ amd:
50
+ device: mi325_1
51
+ depends_on:
52
+ - image-build-amd
53
54
- label: Basic Models Test (Other CPU) # 5min
55
key: basic-models-test-other-cpu
.buildkite/test_areas/weight_loading.yaml
+7
@@ -13,6 +13,13 @@ steps:
13
- tests/weight_loading
14
commands:
15
- bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models.txt
16
+ mirror:
17
+ amd:
18
+ device: mi300_2
19
+ depends_on:
20
+ - image-build-amd
21
+ commands:
22
+ - bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models-amd.txt
23
24
# - label: Weight Loading Multiple GPU - Large Models # optional
25
# working_dir: "/vllm-workspace/tests"