From 4e5a322f49d5bf17cd269cbdf1de9a371416cd6e Mon Sep 17 00:00:00 2001 From: zhuyuxiao Date: Sat, 8 May 2021 00:58:32 +0800 Subject: [PATCH] assign add bug fix --- .../ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc index 333c9dbb4d..60482e5c8a 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc @@ -292,7 +292,7 @@ bool ArithmeticCPUKernel::Launch(const std::vector &inputs, const } else if (operate_type_ == POW) { Pow(broadcastedInput1, broadcastedInput2, output); } else if (operate_type_ == ASSIGNADD) { - AssignAdd(broadcastedInput1, broadcastedInput2, output); + AssignAdd(input1, input2, output); } else if (operate_type_ == ATAN2) { Atan2(broadcastedInput1, broadcastedInput2, output); } else if (operate_type_ == SQUAREDDIFFERENCE) {