[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:
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:
|
||||
|
||||
Reference in New Issue
Block a user