From 8f9a4d08e09b18d42d8d0ff3387a37ee77d4b53c Mon Sep 17 00:00:00 2001 From: Kevin Luu <51931015+khluu@users.noreply.github.com> Date: Fri, 14 Aug 2026 11:32:21 +0000 Subject: [PATCH 1/2] [CI] Shard extended pooling model tests Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> --- .buildkite/test_areas/models_language.yaml | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/.buildkite/test_areas/models_language.yaml b/.buildkite/test_areas/models_language.yaml index cc47c895bc91..abe348daa6d2 100644 --- a/.buildkite/test_areas/models_language.yaml +++ b/.buildkite/test_areas/models_language.yaml @@ -124,24 +124,28 @@ steps: commands: - pytest -v -s models/language/generation_ppl_test -- label: Language Models Test (Extended Pooling) +- label: Language Models Test (Extended Pooling) %N device: h200_35gb key: language-models-test-extended-pooling timeout_in_minutes: 120 + parallelism: 4 optional: true source_file_dependencies: - vllm/ - "!vllm/distributed/kv_transfer/" - tests/models/language/pooling commands: - - pytest -v -s models/language/pooling -m 'not core_model' + - pytest -v -s models/language/pooling -m 'not core_model' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB mirror: amd: dind: false device: mi300_1 timeout_in_minutes: 95 + parallelism: 1 depends_on: - image-build-amd + commands: + - pytest -v -s models/language/pooling -m 'not core_model' - label: Language Models Test (MTEB) key: language-models-test-mteb From f391ffb38b4c22925f30dc5043f527420ae19c7a Mon Sep 17 00:00:00 2001 From: khluu Date: Fri, 14 Aug 2026 14:41:04 -0700 Subject: [PATCH 2/2] [CI] Shard AMD extended pooling tests Co-authored-by: Codex Signed-off-by: khluu --- .buildkite/test_areas/models_language.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.buildkite/test_areas/models_language.yaml b/.buildkite/test_areas/models_language.yaml index abe348daa6d2..d0eff968f7c8 100644 --- a/.buildkite/test_areas/models_language.yaml +++ b/.buildkite/test_areas/models_language.yaml @@ -141,11 +141,11 @@ steps: dind: false device: mi300_1 timeout_in_minutes: 95 - parallelism: 1 + parallelism: 2 depends_on: - image-build-amd commands: - - pytest -v -s models/language/pooling -m 'not core_model' + - pytest -v -s models/language/pooling -m 'not core_model' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB - label: Language Models Test (MTEB) key: language-models-test-mteb