[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
@@ -80,7 +80,8 @@ class TestQwen3235BFP8(unittest.TestCase):
            "--moe-dp-size=2",
            "--attn-cp-size=2",
            "--ep-size=4",
-            "--enable-prefill-context-parallel",
+            "--enable-prefill-cp",
+            "--cp-strategy=zigzag",
        ]

        MTP_ARGS = [