[CI] split tests to reduce CI time (#48219)
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
Jiangyun Zhu committed
Jul 12, 2026 at 04:00 UTC
1ef1c7ebba7f96d9835ade1fbd3658397e0b9be6
2 files changed
+4
-3
.buildkite/test_areas/models_basic.yaml
+1
-1
@@ -27,7 +27,7 @@ steps:
27
# subset of supported models (the complement of the small subset in the above
28
# test.) Also run if model initialization test file is modified
29
- 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
31
32
- label: Basic Models Tests (Other)
33
device: h200_35gb
.buildkite/test_areas/models_multimodal.yaml
+3
-2
@@ -69,7 +69,7 @@ steps:
69
depends_on:
70
- image-build-amd
71
72
-- label: Multi-Modal Processor (CPU)
72
+- label: Multi-Modal Processor (CPU) %N
73
key: multi-modal-processor-cpu
74
depends_on:
75
- image-build-cpu
@@ -80,7 +80,8 @@ steps:
80
- tests/models/registry.py
81
device: cpu-medium
82
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
85
86
- label: Multi-Modal Processor # 44min
87
key: multi-modal-processor