[ROCm][CI] Move PyTorch Compilation Unit Tests to MI300(gfx942) (#47065)
Signed-off-by: charlifu <charlifu@amd.com>
Charlie Fu committed
Jun 30, 2026 at 11:32 UTC
f41e8ddc97ad055647dbcd5a39daca3c13544ca5
1 file changed
+20
-20
.buildkite/test-amd.yaml
+20
-20
@@ -114,26 +114,6 @@ steps:
114
115
#---------------------------------------------------------- mi250 · compile ----------------------------------------------------------#
116
117
-- label: PyTorch Compilation Unit Tests # TBD
118
- timeout_in_minutes: 180
119
- mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250]
120
- agent_pool: mi250_1
121
- torch_nightly: true
122
- optional: true
123
- working_dir: "/vllm-workspace/tests"
124
- source_file_dependencies:
125
- - vllm/compilation/
126
- - vllm/model_executor/layers/
127
- - vllm/v1/worker/
128
- - vllm/v1/attention/
129
- - vllm/v1/cudagraph_dispatcher.py
130
- - vllm/config/compilation.py
131
- - csrc/
132
- - tests/compile
133
- - vllm/platforms/rocm.py
134
- commands:
135
- - "find compile/ -maxdepth 1 -name 'test_*.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
136
-
117
- label: PyTorch Fullgraph Smoke Test # TBD
118
timeout_in_minutes: 180
119
mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250]
@@ -498,6 +478,26 @@ steps:
478
479
#---------------------------------------------------------- mi300 · compile ----------------------------------------------------------#
480
481
+- label: PyTorch Compilation Unit Tests # TBD
482
+ timeout_in_minutes: 180
483
+ mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
484
+ agent_pool: mi300_1
485
+ torch_nightly: true
486
+ optional: true
487
+ working_dir: "/vllm-workspace/tests"
488
+ source_file_dependencies:
489
+ - vllm/compilation/
490
+ - vllm/model_executor/layers/
491
+ - vllm/v1/worker/
492
+ - vllm/v1/attention/
493
+ - vllm/v1/cudagraph_dispatcher.py
494
+ - vllm/config/compilation.py
495
+ - csrc/
496
+ - tests/compile
497
+ - vllm/platforms/rocm.py
498
+ commands:
499
+ - "find compile/ -maxdepth 1 -name 'test_*.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
500
+
501
- label: Fusion E2E Config Sweep (H100-MI300) # TBD
502
timeout_in_minutes: 180
503
mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]