[CI] increase ut buckets & adjust estimation time. (#12919)
This commit is contained in:
2
.github/workflows/pr-test.yml
vendored
2
.github/workflows/pr-test.yml
vendored
@@ -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]
|
||||
|
||||
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user