From 23632d350ce6df7b8d5bc6e3816031522c5022c9 Mon Sep 17 00:00:00 2001
From: Shangming Cai <csmthu@gmail.com>
Date: Wed, 24 Sep 2025 03:46:13 +0800
Subject: [PATCH] Fix latest main ci (#10799)

Signed-off-by: Shangming Cai <csmthu@gmail.com>
---
 .github/workflows/pr-test.yml | 2 +-
 test/srt/run_suite.py         | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 34849cf36..bd434c5ed 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -618,7 +618,7 @@ jobs:
         timeout-minutes: 60
         run: |
           cd test/srt
-          python3 run_suite.py --suite per-commit-4-gpu-b200 --auto-partition-id 0 --auto-partition-size 1
+          python3 run_suite.py --suite per-commit-4-gpu-b200 --auto-partition-id 0 --auto-partition-size 1 --timeout-per-file 3600
 
   pr-test-finish:
     needs: [
diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index 684faf7a8..a54330dfc 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -86,7 +86,7 @@ suites = {
         TestFile("test_metrics.py", 32),
         TestFile("test_metrics_utils.py", 1),
         TestFile("test_mla.py", 167),
-        TestFile("test_mla_deepseek_v3.py", 700),
+        TestFile("test_mla_deepseek_v3.py", 1420),
         TestFile("test_mla_int8_deepseek_v3.py", 429),
         TestFile("test_mla_flashinfer.py", 302),
         TestFile("test_mla_fp8.py", 93),
@@ -121,7 +121,7 @@ suites = {
         TestFile("test_utils_update_weights.py", 48),
         TestFile("test_vision_chunked_prefill.py", 175),
         TestFile("test_vlm_input_format.py", 300),
-        TestFile("test_vision_openai_server_a.py", 403),
+        TestFile("test_vision_openai_server_a.py", 724),
         TestFile("test_vision_openai_server_b.py", 446),
     ],
     "per-commit-2-gpu": [
@@ -131,7 +131,7 @@ suites = {
         TestFile("test_dp_attention.py", 277),
         TestFile("test_load_weights_from_remote_instance.py", 72),
         TestFile("test_patch_torch.py", 19),
-        TestFile("test_release_memory_occupation.py", 127),
+        TestFile("test_release_memory_occupation.py", 257),
         TestFile("hicache/test_hicache_storage_file_backend.py", 400),
         TestFile("hicache/test_hicache_storage_3fs_backend.py", 400),
     ],
@@ -154,7 +154,7 @@ suites = {
     ],
     "per-commit-4-gpu-b200": [
         # TestFile("test_gpt_oss_4gpu.py", 600),
-        TestFile("test_deepseek_v3_fp4_4gpu.py", 600),
+        TestFile("test_deepseek_v3_fp4_4gpu.py", 3600),
     ],
     "per-commit-4-gpu-deepep": [
         TestFile("ep/test_deepep_small.py", 531),