[VLM] Support Piecewise CUDA Graph for Qwen2.5-VL (#13055)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
Co-authored-by: Yuhao Yang <yhyang201@gmail.com>
This commit is contained in:
Yuan Luo
2025-11-20 10:23:44 +08:00
committed by GitHub
co-authored by luoyuan.luo Yuhao Yang
parent 67fca6b297
commit af6bcadcf7
10 changed files with 710 additions and 29 deletions
+2 -1
View File
@@ -94,7 +94,7 @@ suites = {
TestFile("test_original_logprobs.py", 41),
TestFile("test_page_size.py", 60),
TestFile("test_penalty.py", 82),
TestFile("test_piecewise_cuda_graph.py", 600),
TestFile("test_piecewise_cuda_graph.py", 750),
TestFile("test_priority_scheduling.py", 130),
TestFile("test_pytorch_sampling_backend.py", 66),
TestFile("test_radix_attention.py", 105),
@@ -287,6 +287,7 @@ suites = {
TestFile("nightly/test_text_models_perf.py"),
TestFile("nightly/test_vlms_mmmu_eval.py"),
TestFile("nightly/test_vlms_perf.py"),
TestFile("nightly/test_vlms_piecewise_cuda_graph.py"),
TestFile("test_quick_allreduce.py"),
TestFile("test_sagemaker_server.py"),
TestFile("test_schedule_policy.py"),