flashinfer.comm.mnnvl_cutedsl_ag.supports_all_gather_rmsnorm_fp4_quantize

flashinfer.comm.mnnvl_cutedsl_ag.supports_all_gather_rmsnorm_fp4_quantize(workspace: MNNVLCuteDSLAllGatherNormQuantWorkspace, shard: Tensor, weight: Tensor, global_scale: Tensor) → bool

Check only this rank. Use workspace.check_support to agree across ranks.