flashinfer.comm.decode_cp_a2a_lse_reduce_workspace_size

flashinfer.comm.decode_cp_a2a_lse_reduce_workspace_size(max_tokens: int, local_heads: int, cp_size: int, head_dim: int, dtype: dtype) → int

Return the required NCCL symmetric workspace size in bytes.