Fix latest main ci (#10799)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
2
.github/workflows/pr-test.yml
vendored
2
.github/workflows/pr-test.yml
vendored
@@ -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: [
|
||||
|
||||
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user