flashinfer.norm.supports_gated_residual_next_rmsnorm¶
- flashinfer.norm.supports_gated_residual_next_rmsnorm(branch: Tensor, residual: Tensor, next_weight: Tensor, alpha: Tensor | float, beta: Tensor, *, post_weight: Tensor | None = None) bool¶
Return whether inputs satisfy the SM100/SM103/SM107 two-norm kernel contract.