fix: fix regression and unclear pattern (#16561)

This commit is contained in:
Nan Jiang
2026-01-16 23:21:42 -08:00
committed by GitHub
parent 8ce64aa155
commit dd99f818e0
4 changed files with 46 additions and 54 deletions
@@ -43,6 +43,7 @@ def npu_wrapper_rmsnorm_forward(func):
self,
x: torch.Tensor,
residual: Optional[torch.Tensor] = None,
post_residual_addition: Optional[torch.Tensor] = None,
) -> Union[torch.Tensor, Tuple[torch.Tensor, torch.Tensor]]:
from sgl_kernel_npu.norm.add_rmsnorm_bias import add_rmsnorm_bias