[CI] Shard extended pooling model tests (#52322)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Signed-off-by: khluu <khluu000@gmail.com>
Kevin H. Luu committed
Aug 14, 2026 at 15:11 UTC
549cef0b0b2c8924d302686c613cb4c39fc24dcd
1 file changed
+6
-2
.buildkite/test_areas/models_language.yaml
+6
-2
@@ -124,24 +124,28 @@ steps:
124
commands:
125
- pytest -v -s models/language/generation_ppl_test
126
127
-- label: Language Models Test (Extended Pooling)
127
+- label: Language Models Test (Extended Pooling) %N
128
device: h200_35gb
129
key: language-models-test-extended-pooling
130
timeout_in_minutes: 120
131
+ parallelism: 4
132
optional: true
133
source_file_dependencies:
134
- vllm/
135
- "!vllm/distributed/kv_transfer/"
136
- tests/models/language/pooling
137
commands:
137
- - pytest -v -s models/language/pooling -m 'not core_model'
138
+ - pytest -v -s models/language/pooling -m 'not core_model' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
139
mirror:
140
amd:
141
dind: false
142
device: mi300_1
143
timeout_in_minutes: 95
144
+ parallelism: 2
145
depends_on:
146
- image-build-amd
147
+ commands:
148
+ - pytest -v -s models/language/pooling -m 'not core_model' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
149
150
- label: Language Models Test (MTEB)
151
key: language-models-test-mteb