[VLM] Support Piecewise CUDA Graph for Qwen3-Omni-MOE (#14222)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
This commit is contained in:
Yuan Luo
2025-12-02 10:12:10 +08:00
committed by GitHub
co-authored by luoyuan.luo
parent 3ab8ae6847
commit 26aebf83d3
3 changed files with 52 additions and 1 deletions
+1 -1
View File
@@ -97,7 +97,6 @@ suites = {
TestFile("test_original_logprobs.py", 41),
TestFile("test_page_size.py", 60),
TestFile("test_penalty.py", 82),
TestFile("test_piecewise_cuda_graph.py", 850),
TestFile("test_priority_scheduling.py", 130),
TestFile("test_pytorch_sampling_backend.py", 66),
TestFile("test_radix_attention.py", 105),
@@ -158,6 +157,7 @@ suites = {
TestFile("test_local_attn.py", 411),
TestFile("test_multi_instance_release_memory_occupation.py", 64),
TestFile("test_pp_single_node.py", 481),
TestFile("test_piecewise_cuda_graph.py", 1200),
],
"per-commit-8-gpu-h200": [
TestFile("test_deepseek_v3_basic.py", 275),