[CP] Support breakable CUDA graphs for zigzag strategy (#33136)

2026-08-02 23:11:57 -07:00
parent 85484c457d
commit 0bf0640b9d
5 changed files with 311 additions and 4 deletions
@@ -22,6 +22,8 @@ class TestGptOss4GpuMxfp4CP(BaseTestGptOss):
                "4",
                "--cp-strategy",
                "zigzag",
+                "--cuda-graph-backend-prefill",
+                "breakable",
                "--cuda-graph-max-bs-decode",
                "200",
            ],