diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index a652ec3cda1..c67e2581362 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -115,6 +115,12 @@ jobs: cd test/srt python3 test_verl_engine.py + - name: Test Patch Torch + timeout-minutes: 10 + run: | + cd test/srt + python3 test_patch_torch.py + - name: Test expert parallelism (EP=2) timeout-minutes: 10 run: | diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 6095ecf2adf..fa1a7c376f9 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -46,7 +46,6 @@ class TestFile: TestFile("test_openai_server.py", 124), TestFile("test_penalty.py", 41), TestFile("test_page_size.py", 60), - TestFile("test_patch_torch.py", 60), TestFile("test_pytorch_sampling_backend.py", 66), TestFile("test_radix_attention.py", 167), TestFile("test_reasoning_content.py", 89),