[ROCm][CI] move lora%N test to mi300 and gate (#45970)
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Divakar Verma committed
Jun 18, 2026 at 12:50 UTC
21da47dabe27559bf46b80ff6caacafd9dde6035
2 files changed
+25
-16
.buildkite/test-amd.yaml
+14
-16
@@ -415,22 +415,6 @@ steps:
415
commands:
416
- pytest -v -s kernels/mamba
417
418
-#----------------------------------------------------------- mi250 · lora ------------------------------------------------------------#
419
-
420
-- label: LoRA %N # TBD
421
- timeout_in_minutes: 180
422
- mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250]
423
- agent_pool: mi250_1
424
- parallelism: 4
425
- optional: true
426
- working_dir: "/vllm-workspace/tests"
427
- source_file_dependencies:
428
- - vllm/lora
429
- - tests/lora
430
- - vllm/platforms/rocm.py
431
- commands:
432
- - pytest -v -s lora --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --ignore=lora/test_chatglm3_tp.py --ignore=lora/test_llama_tp.py --ignore=lora/test_qwen3_with_multi_loras.py --ignore=lora/test_olmoe_tp.py --ignore=lora/test_deepseekv2_tp.py --ignore=lora/test_gptoss_tp.py --ignore=lora/test_qwen3moe_tp.py --ignore=lora/test_qwen35_densemodel_lora.py
433
-
418
#------------------------------------------------------ mi250 · models / basic -------------------------------------------------------#
419
420
- label: Basic Models Test (Other CPU) # TBD
@@ -1699,6 +1683,20 @@ steps:
1683
1684
#----------------------------------------------------------- mi300 · lora ------------------------------------------------------------#
1685
1686
+- label: LoRA %N # TBD
1687
+ timeout_in_minutes: 180
1688
+ mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
1689
+ agent_pool: mi300_1
1690
+ parallelism: 4
1691
+ optional: true
1692
+ working_dir: "/vllm-workspace/tests"
1693
+ source_file_dependencies:
1694
+ - vllm/lora
1695
+ - tests/lora
1696
+ - vllm/platforms/rocm.py
1697
+ commands:
1698
+ - pytest -v -s lora --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --ignore=lora/test_chatglm3_tp.py --ignore=lora/test_llama_tp.py --ignore=lora/test_qwen3_with_multi_loras.py --ignore=lora/test_olmoe_tp.py --ignore=lora/test_deepseekv2_tp.py --ignore=lora/test_gptoss_tp.py --ignore=lora/test_qwen3moe_tp.py --ignore=lora/test_qwen35_densemodel_lora.py
1699
+
1700
- label: LoRA TP (Distributed) # TBD
1701
timeout_in_minutes: 180
1702
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
.buildkite/test_areas/lora.yaml
+11
@@ -12,6 +12,17 @@ steps:
12
commands:
13
- pytest -v -s lora --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --ignore=lora/test_chatglm3_tp.py --ignore=lora/test_llama_tp.py --ignore=lora/test_qwen3_with_multi_loras.py --ignore=lora/test_olmoe_tp.py --ignore=lora/test_deepseekv2_tp.py --ignore=lora/test_gptoss_tp.py --ignore=lora/test_qwen3moe_tp.py --ignore=lora/test_qwen35_densemodel_lora.py
14
parallelism: 4
15
+ mirror:
16
+ amd:
17
+ device: mi325_1
18
+ working_dir: "/vllm-workspace/tests"
19
+ timeout_in_minutes: 60
20
+ source_file_dependencies:
21
+ - vllm/lora
22
+ - tests/lora
23
+ - vllm/platforms/rocm.py
24
+ depends_on:
25
+ - image-build-amd
26
27
28
- label: LoRA TP (Distributed)