flashinfer.norm.supports_rmsnorm_sigmoid_gate¶
- flashinfer.norm.supports_rmsnorm_sigmoid_gate(dtype: dtype, head_dim: int, device: device | str | None = None) bool¶
Whether the SM100/SM103/SM107 fused kernel supports this dtype and head size.
Back to top