[NPU] Fix error due to missing parameter quant_linear passing (#37384)

This commit is contained in:
ZeyuanChen2000
2026-09-14 14:37:44 +08:00
committed by GitHub
parent ce66ba2844
commit a4781c9fe5
@@ -63,6 +63,7 @@ def npu_wrapper_rmsnorm_forward(func):
x: torch.Tensor,
residual: Optional[torch.Tensor] = None,
post_residual_addition: Optional[torch.Tensor] = None,
**kwargs,
) -> Union[torch.Tensor, Tuple[torch.Tensor, torch.Tensor]]:
if not x.is_contiguous():
x = x.contiguous()