[CI] Shard basic model initialization tests (#54753)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com>
Kevin H. Luu committed
Sep 1, 2026 at 21:31 UTC
05201d8ecf340cee4318ae7b962ce7dabe736647
1 file changed
+2
-2
.buildkite/test_areas/models_basic.yaml
+2
-2
@@ -26,7 +26,7 @@ steps:
26
- label: ":nvidia: (H200 MIG 35GB) Basic Models (Extra Initialization) Shard %N"
27
device: h200_35gb
28
key: basic-models-tests-extra-initialization
29
- timeout_in_minutes: 100
29
+ timeout_in_minutes: 30
30
source_file_dependencies:
31
- vllm/model_executor/models/
32
- tests/models/test_initialization.py
@@ -36,7 +36,7 @@ steps:
36
# subset of supported models (the complement of the small subset in the above
37
# test.) Also run if model initialization test file is modified
38
- pytest -v -s models/test_initialization.py -k 'not test_can_initialize_small_subset' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
39
- parallelism: 4
39
+ parallelism: 14
40
41
- label: ":nvidia: (H200 MIG 35GB) Basic Models (Other)"
42
device: h200_35gb