From 0d3e3072ee2775ee3fa2b0403e1629525790f1c9 Mon Sep 17 00:00:00 2001
From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
Date: Fri, 28 Mar 2025 12:22:45 +0800
Subject: [PATCH] Fix CI of test_patch_torch (#4844)

---
 .github/workflows/pr-test.yml | 6 ++++++
 test/srt/run_suite.py         | 1 -
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index a652ec3cd..c67e25813 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 6095ecf2a..fa1a7c376 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -46,7 +46,6 @@ suites = {
         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),