Support piecewise cuda graph for dsv3 fp4 (#15531)

2025-12-21 14:50:32 +08:00
parent 6014365564
commit 8fe3e37468
7 changed files with 148 additions and 16 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -163,7 +163,7 @@ suites = {
        TestFile("test_disaggregation_dp_attention.py", 155),
    ],
    "per-commit-4-gpu-b200-stage-b": [
-        TestFile("test_deepseek_v3_fp4_4gpu.py", 1800),  # Stage B test
+        TestFile("test_deepseek_v3_fp4_4gpu.py", 2000),  # Stage B test
    ],
    "per-commit-4-gpu-b200": [
        TestFile("test_flash_attention_4.py", 90),