[PCP][DCP] Declare FlashMLASparse MTP support at CP interleave > 1 (#56722)

Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
Lucas Wilkinson
2026-09-14 10:20:46 -04:00
committed by GitHub
co-authored by Claude Opus 5
parent ff5f6d41b1
commit 4be3dcf0fc
@@ -634,6 +634,7 @@ class FlashMLASparseMetadataBuilder(
class FlashMLASparseImpl(SparseMLACommonImpl[FlashMLASparseMetadata]):
can_return_lse_for_decode: bool = True
supports_dcp: bool = True
supports_mtp_with_cp_non_trivial_interleave_size: bool = True
@staticmethod
def _compute_fp8_decode_padded_heads(num_heads: int) -> int: