[CI] Split Dynamic Shapes out of (H200 MIG 35GB) PyTorch Compilation + (MI300) mirror (#58451)
Signed-off-by: Thang Nguyen <thangnguyenvn647@gmail.com> Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
Thang Nguyen committed
Sep 28, 2026 at 03:32 UTC
66d2d4b2e7a38518110aeba31627d99c41b1735d
4 files changed
+23
-3
.buildkite/test-amd.yaml
+1
@@ -614,6 +614,7 @@ steps:
614
- vllm/v1/
615
commands:
616
- "find compile/ -maxdepth 1 -name 'test_*.py' -print0 | xargs -0 -n1 -I{} pytest -s -v '{}'"
617
+ - pytest -s -v compile/dynamic_shapes/
618
619
- label: ":amd: (MI300) Fusion E2E Config Sweep" # TBD
620
timeout_in_minutes: 40
.buildkite/test_areas/pytorch.yaml
+22
-3
@@ -5,8 +5,8 @@ steps:
5
- label: ":nvidia: (H200 MIG 35GB) PyTorch Compilation"
6
device: h200_35gb
7
key: pytorch-compilation-unit-tests
8
- timeout_in_minutes: 60
9
- source_file_dependencies:
8
+ timeout_in_minutes: 30
9
+ source_file_dependencies: &pytorch-compilation-deps
10
- vllm/__init__.py
11
- vllm/_aiter_ops.py
12
- vllm/_custom_ops.py
@@ -47,7 +47,26 @@ steps:
47
label: ":amd: (MI300) PyTorch Compilation"
48
dind: false
49
device: mi300_1
50
- timeout_in_minutes: 90
50
+ timeout_in_minutes: 45
51
+ depends_on:
52
+ - image-build-amd
53
+ source_file_dependencies:
54
+ - csrc/
55
+ - vllm/platforms/rocm.py
56
+
57
+- label: ":nvidia: (H200 MIG 35GB) PyTorch Compilation / Dynamic Shapes"
58
+ device: h200_35gb
59
+ key: pytorch-compilation-dynamic-shapes
60
+ timeout_in_minutes: 30
61
+ source_file_dependencies: *pytorch-compilation-deps
62
+ commands:
63
+ - pytest -s -v compile/dynamic_shapes/
64
+ mirror:
65
+ amd:
66
+ label: ":amd: (MI300) PyTorch Compilation / Dynamic Shapes"
67
+ dind: false
68
+ device: mi300_1
69
+ timeout_in_minutes: 45
70
depends_on:
71
- image-build-amd
72
source_file_dependencies: