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
|
timeout-minutes: 60
|
||||||
run: |
|
run: |
|
||||||
cd test/srt
|
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:
|
pr-test-finish:
|
||||||
needs: [
|
needs: [
|
||||||
|
|||||||
@@ -86,7 +86,7 @@ suites = {
|
|||||||
TestFile("test_metrics.py", 32),
|
TestFile("test_metrics.py", 32),
|
||||||
TestFile("test_metrics_utils.py", 1),
|
TestFile("test_metrics_utils.py", 1),
|
||||||
TestFile("test_mla.py", 167),
|
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_int8_deepseek_v3.py", 429),
|
||||||
TestFile("test_mla_flashinfer.py", 302),
|
TestFile("test_mla_flashinfer.py", 302),
|
||||||
TestFile("test_mla_fp8.py", 93),
|
TestFile("test_mla_fp8.py", 93),
|
||||||
@@ -121,7 +121,7 @@ suites = {
|
|||||||
TestFile("test_utils_update_weights.py", 48),
|
TestFile("test_utils_update_weights.py", 48),
|
||||||
TestFile("test_vision_chunked_prefill.py", 175),
|
TestFile("test_vision_chunked_prefill.py", 175),
|
||||||
TestFile("test_vlm_input_format.py", 300),
|
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),
|
TestFile("test_vision_openai_server_b.py", 446),
|
||||||
],
|
],
|
||||||
"per-commit-2-gpu": [
|
"per-commit-2-gpu": [
|
||||||
@@ -131,7 +131,7 @@ suites = {
|
|||||||
TestFile("test_dp_attention.py", 277),
|
TestFile("test_dp_attention.py", 277),
|
||||||
TestFile("test_load_weights_from_remote_instance.py", 72),
|
TestFile("test_load_weights_from_remote_instance.py", 72),
|
||||||
TestFile("test_patch_torch.py", 19),
|
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_file_backend.py", 400),
|
||||||
TestFile("hicache/test_hicache_storage_3fs_backend.py", 400),
|
TestFile("hicache/test_hicache_storage_3fs_backend.py", 400),
|
||||||
],
|
],
|
||||||
@@ -154,7 +154,7 @@ suites = {
|
|||||||
],
|
],
|
||||||
"per-commit-4-gpu-b200": [
|
"per-commit-4-gpu-b200": [
|
||||||
# TestFile("test_gpt_oss_4gpu.py", 600),
|
# 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": [
|
"per-commit-4-gpu-deepep": [
|
||||||
TestFile("ep/test_deepep_small.py", 531),
|
TestFile("ep/test_deepep_small.py", 531),
|
||||||
|
|||||||
Reference in New Issue
Block a user