[CP V1 Deprecation 1/5] Migrate tests to strategy-based prefill CP (#36222)

2026-08-24 20:15:34 -07:00
parent a0f52eca98
commit 833be86c15
20 changed files with 37 additions and 558 deletions
@@ -39,7 +39,9 @@ class TestUnifiedQwen3HiCacheCP(UnifiedRadixTreeTestMixin, CustomTestCase):
                "4",
                "--attn-cp-size",
                "2",
-                "--enable-prefill-context-parallel",
+                "--enable-prefill-cp",
+                "--cp-strategy",
+                "zigzag",
                "--mem-fraction-static",
                "0.8",
                "--cuda-graph-max-bs-decode",