flashinfer.comm.mnnvl_cutedsl_ag.supports_all_gather_gated_residual_next_rmsnorm¶
- flashinfer.comm.mnnvl_cutedsl_ag.supports_all_gather_gated_residual_next_rmsnorm(workspace: MNNVLCuteDSLAllGatherDualNormWorkspace, branch_shard: Tensor, residual: Tensor, alpha: Tensor, beta: Tensor, next_weight: Tensor, post_weight: Tensor | None = None) bool¶
Check only this rank. Use
workspace.check_supportto agree across ranks.