[CI][2/N] reduce CI time (#48394)
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
Jiangyun Zhu committed
Jul 12, 2026 at 19:16 UTC
370b678a028872e7c7def9aa18cfef3ab1fa750e
2 files changed
+8
-4
.buildkite/test_areas/attention.yaml
+4
-2
@@ -12,7 +12,8 @@ steps:
12
- vllm/v1/attention
13
- tests/v1/attention
14
commands:
15
- - pytest -v -s v1/attention
15
+ - pytest -v -s v1/attention --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
16
+ parallelism: 2
17
mirror:
18
amd:
19
device: mi325_1
@@ -38,4 +39,5 @@ steps:
39
- vllm/v1/attention
40
- tests/v1/attention
41
commands:
41
- - pytest -v -s v1/attention
42
+ - pytest -v -s v1/attention --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
43
+ parallelism: 2
.buildkite/test_areas/kernels.yaml
+4
-2
@@ -23,7 +23,8 @@ steps:
23
- tests/kernels/test_concat_mla_q.py
24
- tests/kernels/test_fused_qk_norm_rope_gate.py
25
commands:
26
- - pytest -v -s kernels/core --ignore=kernels/core/test_minimax_reduce_rms.py kernels/test_concat_mla_q.py kernels/test_fused_qk_norm_rope_gate.py
26
+ - pytest -v -s kernels/core --ignore=kernels/core/test_minimax_reduce_rms.py kernels/test_concat_mla_q.py kernels/test_fused_qk_norm_rope_gate.py --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
27
+ parallelism: 3
28
29
- label: Kernels MiniMax Reduce RMS Test (2 GPUs)
30
key: kernels-minimax-reduce-rms-test-2-gpus
@@ -271,7 +272,8 @@ steps:
272
- tests/kernels/helion/
273
commands:
274
- pip install helion==1.1.0
274
- - pytest -v -s kernels/helion/
275
+ - pytest -v -s kernels/helion/ --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
276
+ parallelism: 2
277
278
279
- label: Kernels FP8 MoE Test (1xH100)