From 14a01d327fd51d0f4d26e3440976c5b5aa6a6fee Mon Sep 17 00:00:00 2001
From: fangzehua <fangzehua1@huawei.com>
Date: Wed, 10 Mar 2021 09:21:17 +0800
Subject: [PATCH] fix assign cpu output

---
 .../ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc    | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc
index 4064190e19..4d791e19e3 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc
@@ -57,6 +57,10 @@ bool AssignCPUKernel::Launch(const std::vector<AddressPtr> &inputs, const std::v
   if (ret != 0) {
     MS_LOG(EXCEPTION) << "memcpy_s error, error no " << ret;
   }
+  ret = memcpy_s(outputs[0]->addr, max_size, inputs[1]->addr, total_size);
+  if (ret != 0) {
+    MS_LOG(EXCEPTION) << "memcpy_s error, error no " << ret;
+  }
   return true;
 }
 }  // namespace kernel