diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 727f2e2e9..51dee9f90 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -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] diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 40739b4bb..194e4ab17 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -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),