Skip to content

Commit 1dce3d3

Browse files
ZJY0516NickLucche
authored andcommitted
[CI] split tests to reduce CI time (vllm-project#48219)
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
1 parent 32e18a5 commit 1dce3d3

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

.buildkite/test_areas/models_basic.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ steps:
2727
# subset of supported models (the complement of the small subset in the above
2828
# test.) Also run if model initialization test file is modified
2929
- 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
30-
parallelism: 2
30+
parallelism: 4
3131

3232
- label: Basic Models Tests (Other)
3333
device: h200_35gb

.buildkite/test_areas/models_multimodal.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ steps:
6969
depends_on:
7070
- image-build-amd
7171

72-
- label: Multi-Modal Processor (CPU)
72+
- label: Multi-Modal Processor (CPU) %N
7373
key: multi-modal-processor-cpu
7474
depends_on:
7575
- image-build-cpu
@@ -80,7 +80,8 @@ steps:
8080
- tests/models/registry.py
8181
device: cpu-medium
8282
commands:
83-
- pytest -v -s models/multimodal/processing --ignore models/multimodal/processing/test_tensor_schema.py
83+
- pytest -v -s models/multimodal/processing --ignore models/multimodal/processing/test_tensor_schema.py --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB
84+
parallelism: 4
8485

8586
- label: Multi-Modal Processor # 44min
8687
key: multi-modal-processor

0 commit comments

Comments
 (0)