Fix latest main ci (#10799)

Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
Shangming Cai
2025-09-24 03:46:13 +08:00
committed by GitHub
parent 08b8c0c3cd
commit 23632d350c
2 changed files with 5 additions and 5 deletions

View File

@@ -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: [

View File

@@ -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),