Skip to content

GitLab

Unverified Commit 23632d35 authored by Shangming Cai's avatar Shangming Cai Committed by GitHub
Browse files

Fix latest main ci (#10799) 


Signed-off-by: default avatarShangming Cai <csmthu@gmail.com>
parent 08b8c0c3
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
.github/workflows/pr-test.yml
View file @ 23632d35
......@@ -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: [
......
test/srt/run_suite.py
View file @ 23632d35
......@@ -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),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment