diff --git a/.buildkite/test_areas/entrypoints.yaml b/.buildkite/test_areas/entrypoints.yaml index 4860159f90..4d8bfdc55f 100644 --- a/.buildkite/test_areas/entrypoints.yaml +++ b/.buildkite/test_areas/entrypoints.yaml @@ -121,6 +121,7 @@ steps: - label: ":nvidia: (H200 MIG 35GB) Entrypoints Integration (OpenAI API chat_completion)" device: h200_35gb key: entrypoints-integration-api-server-openai-chat_completion + automatic_shard: true timeout_in_minutes: 83 working_dir: "/vllm-workspace/tests" source_file_dependencies: diff --git a/.buildkite/test_areas/lm_eval.yaml b/.buildkite/test_areas/lm_eval.yaml index 3a5d91277c..c7d9c1777e 100644 --- a/.buildkite/test_areas/lm_eval.yaml +++ b/.buildkite/test_areas/lm_eval.yaml @@ -105,6 +105,7 @@ steps: - label: ":nvidia: (H100) LM Eval Large Models" key: lm-eval-large-models-4xh100 + automatic_shard: true device: h100 optional: true num_devices: 4 @@ -139,6 +140,7 @@ steps: - label: ":nvidia: (B200) LM Eval Small Models" key: lm-eval-small-models-1xb200 + automatic_shard: true timeout_in_minutes: 50 device: b200-k8s optional: true diff --git a/.buildkite/test_areas/models_basic.yaml b/.buildkite/test_areas/models_basic.yaml index 848ab85ef7..a500b838c5 100644 --- a/.buildkite/test_areas/models_basic.yaml +++ b/.buildkite/test_areas/models_basic.yaml @@ -58,6 +58,7 @@ steps: - label: ":nvidia: (H200 MIG 35GB) Basic Models (Other)" device: h200_35gb key: basic-models-tests-other + automatic_shard: true timeout_in_minutes: 38 source_file_dependencies: - vllm/ diff --git a/.buildkite/test_areas/models_multimodal.yaml b/.buildkite/test_areas/models_multimodal.yaml index 0750d92eac..d281d16f7f 100644 --- a/.buildkite/test_areas/models_multimodal.yaml +++ b/.buildkite/test_areas/models_multimodal.yaml @@ -24,6 +24,7 @@ steps: - label: ":nvidia: (H200 MIG 18GB) Multimodal Models (Standard) 2: qwen3 + gemma" key: multi-modal-models-standard-2-qwen3-gemma + automatic_shard: true timeout_in_minutes: 75 device: h200_18gb source_file_dependencies: