diff --git a/torch_npu/onnx/wrapper_onnx_ops.py b/torch_npu/onnx/wrapper_onnx_ops.py index c05d368565cbd75042d0e092ec99675486bd6aed..268d9ae9e2d818703a9becffbb08b191c2878f6e 100644 --- a/torch_npu/onnx/wrapper_onnx_ops.py +++ b/torch_npu/onnx/wrapper_onnx_ops.py @@ -191,7 +191,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):