diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 3aa892a26..d6a985147 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -1282,11 +1282,11 @@ jobs: run: | IS_BLACKWELL=1 bash scripts/ci/ci_install_dependency.sh - - name: Run test - timeout-minutes: 45 - run: | - cd test/srt - python3 run_suite.py --suite per-commit-8-gpu-b200 --timeout-per-file 1800 + # - name: Run test + # timeout-minutes: 45 + # run: | + # cd test/srt + # python3 run_suite.py --suite per-commit-8-gpu-b200 --timeout-per-file 1800 pr-test-finish: needs: @@ -1321,7 +1321,7 @@ jobs: unit-test-deepep-8-gpu, unit-test-backend-4-gpu-b200, unit-test-backend-4-gpu-gb200, - unit-test-backend-8-gpu-b200, + # unit-test-backend-8-gpu-b200, # Moved to nightly - large models only ] if: always() runs-on: ubuntu-latest diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 57c0e1716..24404e3e1 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -174,9 +174,9 @@ suites = { TestFile("test_llama31_fp4.py", 300), TestFile("test_eagle_infer_beta_dp_attention.py", 300), ], - "per-commit-8-gpu-b200": [ - TestFile("test_mistral_large3_basic.py", 275), - ], + # "per-commit-8-gpu-b200": [ + # TestFile("test_mistral_large3_basic.py", 275), # Moved to nightly - large model + # ], "per-commit-4-gpu-gb200": [ TestFile("test_cutedsl_moe.py", 300), TestFile("test_deepseek_v3_cutedsl_4gpu.py", 1800), @@ -211,6 +211,7 @@ suites = { TestFile("test_vision_openai_server_common.py"), TestFile("test_profile_v2.py"), TestFile("models/test_ministral3_models.py"), + TestFile("test_mistral_large3_basic.py"), ], }