From 1b78322b9f4375995060283f788e73e6f97555ad Mon Sep 17 00:00:00 2001 From: cathychan0027 Date: Mon, 28 Jul 2025 11:33:28 +0800 Subject: [PATCH] finish --- torch_npu/onnx/wrapper_onnx_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_npu/onnx/wrapper_onnx_ops.py b/torch_npu/onnx/wrapper_onnx_ops.py index c05d368565..268d9ae9e2 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): -- Gitee