diff --git a/.buildkite/test_areas/entrypoints.yaml b/.buildkite/test_areas/entrypoints.yaml index 4d8bfdc55f..abbf354add 100644 --- a/.buildkite/test_areas/entrypoints.yaml +++ b/.buildkite/test_areas/entrypoints.yaml @@ -41,6 +41,7 @@ steps: amd: label: ":amd: (MI355 DPX) Entrypoints Unit" dind: false + automatic_shard: true device: mi355_dpx timeout_in_minutes: 50 depends_on: @@ -207,6 +208,7 @@ steps: amd: label: ":amd: (MI355 DPX) Entrypoints Integration (Speech to Text)" dind: false + automatic_shard: true device: mi355_dpx timeout_in_minutes: 60 working_dir: "/vllm-workspace/tests" diff --git a/.buildkite/test_areas/misc.yaml b/.buildkite/test_areas/misc.yaml index de5b235e20..8b9deb1a82 100644 --- a/.buildkite/test_areas/misc.yaml +++ b/.buildkite/test_areas/misc.yaml @@ -32,6 +32,7 @@ steps: amd: label: ":amd: (MI355 DPX) V1 Sample" dind: false + automatic_shard: true device: mi355_dpx timeout_in_minutes: 70 depends_on: diff --git a/.buildkite/test_areas/pytorch.yaml b/.buildkite/test_areas/pytorch.yaml index 60061af414..1dc6659e8e 100644 --- a/.buildkite/test_areas/pytorch.yaml +++ b/.buildkite/test_areas/pytorch.yaml @@ -160,6 +160,7 @@ steps: amd: label: ':amd: (MI300) PyTorch Compilation Passes' dind: false + automatic_shard: true device: mi300_1 timeout_in_minutes: 80 working_dir: /vllm-workspace/tests