Log if cuda graph is used & extend cuda graph capture to cuda-graph-max-bs (#6201)

Co-authored-by: SangBin Cho <rkooo567@gmail.com>
2025-05-12 00:17:33 -07:00
parent 7d3a3d4510
commit fba8eccd7e
27 changed files with 293 additions and 121 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -101,8 +101,8 @@ suites = {
        # TestFile("test_deepep_intranode.py", 50),
        # TestFile("test_deepep_low_latency.py", 50),
        # TestFile("test_moe_deepep_eval_accuracy_large.py", 250),
+        # TestFile("test_disaggregation.py", 90),
        TestFile("test_local_attn.py", 250),
-        TestFile("test_disaggregation.py", 90),
        TestFile("test_full_deepseek_v3.py", 250),
        TestFile("test_pp_single_node.py", 150),
    ],