[CP V1 Deprecation 4/5] Canonicalize prefill CP API names (#36229)

2026-09-08 16:03:36 -07:00
parent 559c7fa75b
commit ed183d45ac
33 changed files with 143 additions and 173 deletions
@@ -19,7 +19,7 @@ def test_mhc_fused_post_pre_matches_unfused(
    if not torch.cuda.is_available():
        pytest.skip("CUDA is required for TileLang mHC kernels")

-    monkeypatch.setattr(mhc, "is_dsa_prefill_cp_round_robin_split", lambda: False)
+    monkeypatch.setattr(mhc, "is_dsa_prefill_cp_interleave", lambda: False)
    # This is a single-process kernel unit test with no TP group initialized.
    # mhc_pre / mhc_fused_post_pre allocate the MoE input in the symmetric-memory
    # pool via use_symmetric_memory(get_tp_group(), ...); bypass that path so the