Browse Source

!13072 fix assign cpu output

From: @fangzehua
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
4fcf7a50fa
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc

+ 4
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc View File

@@ -57,6 +57,10 @@ bool AssignCPUKernel::Launch(const std::vector<AddressPtr> &inputs, const std::v
if (ret != 0) { if (ret != 0) {
MS_LOG(EXCEPTION) << "memcpy_s error, error no " << ret; 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; return true;
} }
} // namespace kernel } // namespace kernel


Loading…
Cancel
Save