diff --git a/torch_npu/onnx/wrapper_onnx_ops.py b/torch_npu/onnx/wrapper_onnx_ops.py index 16ae07087e5bd17121ab4519a929e7c564270ab7..7d647f50d4ed8d00a35f0a5d3f52947025e93277 100644 --- a/torch_npu/onnx/wrapper_onnx_ops.py +++ b/torch_npu/onnx/wrapper_onnx_ops.py @@ -204,7 +204,7 @@ class _NPUAddRmsNormOP(torch.autograd.Function): @staticmethod def symbolic(g, x1: Tensor, x2: Tensor, gamma: Tensor, epsilon: float = 1e-6): - return g.op("npu::NPURmsNorm", x1, x2, gamma, epsilon_f=epsilon, outputs=3) + return g.op("npu::NPUAddRmsNorm", x1, x2, gamma, epsilon_f=epsilon, outputs=3) class _NPUDiouOP(torch.autograd.Function):