EP Support for Piecewise Cuda Graph (#14164)

Signed-off-by: Oasis-Git <ayw.sirius19@gmail.com>
2025-12-20 01:59:27 +08:00
parent 05eb0bcc61
commit 9d0347b33a
11 changed files with 224 additions and 56 deletions
@@ -140,7 +140,7 @@ suites = {
        TestFile("test_dp_attention.py", 350),
        TestFile("test_load_weights_from_remote_instance.py", 72),
        TestFile("test_patch_torch.py", 19),
-        TestFile("test_piecewise_cuda_graph_2_gpu.py", 200),
+        TestFile("test_piecewise_cuda_graph_2_gpu.py", 400),
        TestFile("test_eagle_dp_attention.py", 200),
    ],
    "per-commit-4-gpu": [