flashinfer.comm.supports_tp_moe_finalize_rmsnorm

flashinfer.comm.supports_tp_moe_finalize_rmsnorm(gemm2_permuted: Tensor, expert_weights: Tensor, expanded_idx_to_permuted_idx: Tensor, norm_weight: Tensor, output: Tensor) → bool

Return whether tensors satisfy the TP SM100/SM103/SM107 kernel contract.