[CI] increase ut buckets & adjust estimation time. (#12919)

This commit is contained in:
Liangsheng Yin
2025-11-09 18:17:50 +08:00
committed by GitHub
parent 3633f8b0cf
commit 877cb52840
2 changed files with 5 additions and 5 deletions

View File

@@ -322,7 +322,7 @@ jobs:
timeout-minutes: 30
run: |
cd test/srt
python3 run_suite.py --suite per-commit-1-gpu --auto-partition-id ${{ matrix.part }} --auto-partition-size 13
python3 run_suite.py --suite per-commit-1-gpu --auto-partition-id ${{ matrix.part }} --auto-partition-size 15
unit-test-backend-2-gpu:
needs: [check-changes, sgl-kernel-build-wheels]

View File

@@ -35,7 +35,7 @@ suites = {
TestFile("models/test_encoder_embedding_models.py", 460),
TestFile("models/test_generation_models.py", 103),
TestFile("models/test_nvidia_nemotron_nano_v2.py", 160),
TestFile("models/test_qwen_models.py", 82),
TestFile("models/test_qwen_models.py", 150),
TestFile("models/test_reward_models.py", 132),
TestFile("models/test_transformers_models.py", 320),
TestFile("models/test_vlm_models.py", 741),
@@ -69,11 +69,11 @@ suites = {
TestFile("test_build_eagle_tree.py", 8),
TestFile("test_chunked_prefill.py", 410),
TestFile("test_create_kvindices.py", 2),
TestFile("test_deterministic.py", 320),
TestFile("test_deterministic.py", 400),
TestFile("test_eagle_infer_a.py", 750),
TestFile("test_eagle_infer_b.py", 750),
TestFile("test_eagle_infer_beta.py", 90),
TestFile("test_constrained_decoding.py", 120),
TestFile("test_constrained_decoding.py", 150),
TestFile("test_eval_fp8_accuracy.py", 303),
TestFile("test_fa3.py", 420),
TestFile("test_flashmla.py", 230),
@@ -116,7 +116,7 @@ suites = {
TestFile("test_speculative_registry.py", 1),
TestFile("test_skip_tokenizer_init.py", 117),
TestFile("test_srt_endpoint.py", 130),
TestFile("test_srt_engine.py", 261),
TestFile("test_srt_engine.py", 450),
TestFile("test_standalone_speculative_decoding.py", 150),
TestFile("test_start_profile.py", 60),
TestFile("test_profile_merger.py", 60),