diff --git a/.buildkite/test_areas/entrypoints.yaml b/.buildkite/test_areas/entrypoints.yaml index abbf354add..ae59d3cb53 100644 --- a/.buildkite/test_areas/entrypoints.yaml +++ b/.buildkite/test_areas/entrypoints.yaml @@ -254,6 +254,7 @@ steps: amd: label: ":amd: (MI355 DPX) Entrypoints Integration (Pooling)" dind: false + automatic_shard: true device: mi355_dpx timeout_in_minutes: 65 working_dir: "/vllm-workspace/tests" diff --git a/.buildkite/test_areas/kernels.yaml b/.buildkite/test_areas/kernels.yaml index 901d3251c2..1c1633c06e 100644 --- a/.buildkite/test_areas/kernels.yaml +++ b/.buildkite/test_areas/kernels.yaml @@ -532,6 +532,7 @@ steps: amd: label: ":amd: (MI250) Mamba Kernels" dind: false + automatic_shard: true device: mi250_1 timeout_in_minutes: 55 depends_on: diff --git a/.buildkite/test_areas/quantization.yaml b/.buildkite/test_areas/quantization.yaml index 490e20534f..7c9ddf9438 100644 --- a/.buildkite/test_areas/quantization.yaml +++ b/.buildkite/test_areas/quantization.yaml @@ -137,6 +137,7 @@ steps: amd: label: ":amd: (MI355 DPX) Quantized Models" dind: false + automatic_shard: true device: mi355_dpx timeout_in_minutes: 80 working_dir: "/vllm-workspace/tests" diff --git a/.buildkite/test_areas/spec_decode.yaml b/.buildkite/test_areas/spec_decode.yaml index eceb914c86..c2742b4c51 100644 --- a/.buildkite/test_areas/spec_decode.yaml +++ b/.buildkite/test_areas/spec_decode.yaml @@ -28,6 +28,7 @@ steps: amd: label: ":amd: (MI300) V1 Spec Decode" dind: false + automatic_shard: true device: mi300_1 timeout_in_minutes: 50 depends_on: