[CI] Shard V1 KV Connectors 1→4 (#56324)
Signed-off-by: Thang Nguyen <thang.nguyen@inferact.ai> Co-authored-by: Thang Nguyen <thang.nguyen@inferact.ai> Co-authored-by: Kimi Code <noreply@moonshot.cn>
Thang Nguyen committed
Sep 15, 2026 at 05:58 UTC
abb4cb9fc5f1b1039876ed8a657b025f65643d11
1 file changed
+5
-4
.buildkite/test_areas/misc.yaml
+5
-4
@@ -142,19 +142,20 @@ steps:
142
depends_on:
143
- image-build-amd
144
145
-- label: ":nvidia: (H200 MIG 35GB) V1 KV Connectors"
145
+- label: ":nvidia: (H200 MIG 35GB) V1 KV Connectors Shard %N"
146
device: h200_35gb
147
key: v1-kv-connectors
148
timeout_in_minutes: 45
149
+ parallelism: 4
150
source_file_dependencies: *v1-core-kv-metrics-deps
151
commands:
152
- bash /vllm-workspace/.buildkite/scripts/install-kv-connectors.sh
153
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
153
- - pytest -v -s -m 'not cpu_test' v1/kv_connector/unit
154
- - pytest -v -s -m 'not cpu_test' v1/ec_connector/unit
154
+ - pytest -v -s -m 'not cpu_test' v1/kv_connector/unit --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
155
+ - pytest -v -s -m 'not cpu_test' v1/ec_connector/unit --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
156
mirror:
157
amd:
157
- label: ":amd: (MI300) V1 KV Connectors"
158
+ label: ":amd: (MI300) V1 KV Connectors Shard %N"
159
dind: false
160
device: mi300_1
161
timeout_in_minutes: 45