Browse Source

fix assign cpu output

tags/v1.2.0-rc1
fangzehua 4 years ago
parent
commit
14a01d327f
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) {
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


Loading…
Cancel
Save