Piecewise Cuda Graph set default (#16331)

2026-03-02 23:18:07 +08:00
parent 468e3dc56b
commit c64274c746
34 changed files with 505 additions and 143 deletions
@@ -38,7 +38,7 @@ class TestDisaggregationPiecewiseCudaGraph(PDDisaggregationServerBase):
            "prefill",
            "--tp",
            "1",
-            "--enable-piecewise-cuda-graph",
+            "--enforce-piecewise-cuda-graph",
        ]
        prefill_args += cls.transfer_backend + cls.rdma_devices
        cls.process_prefill = popen_launch_pd_server(