Update layernorm_gated.py (#378)

chenxu214 · web-flow · commit b7e88d6982e1 · 2026-02-15T22:45:39.000+08:00
add activation attr
diff --git a/python/sgl_kernel_npu/sgl_kernel_npu/fla/layernorm_gated.py b/python/sgl_kernel_npu/sgl_kernel_npu/fla/layernorm_gated.py
@@ -248,6 +248,7 @@ def layer_norm_fwd_npu(
     group_size=None,
     norm_before_gate=True,
     is_rms_norm=False,
+    activation=None,
 ):
     M, N = x.shape
     if group_size is None:
