p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix codex

This commit is contained in:
wenkai 2020-06-11 15:28:09 +08:00
parent cc0add562b
commit 936d3c6ef8
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/device/cpu/cpu_device_address.cc
View File

@ -22,7 +22,10 @@ namespace device {
namespace cpu {
bool CPUDeviceAddress::SyncDeviceToHost(const std::vector<int> & /*shape*/, size_t size, TypeId type,
void *host_ptr) const {
MS_EXCEPTION_IF_NULL(ptr_);
if (ptr_ == nullptr) {
MS_LOG(ERROR) << "The pointer ptr_ is null!";
return false;
}
if (host_ptr == ptr_) {
MS_LOG(DEBUG) << "host_ptr is equal to ptr_, request ignored.";
@ -30,14 +33,18 @@ bool CPUDeviceAddress::SyncDeviceToHost(const std::vector<int> & /*shape*/, size
}
if (type == type_id_) {
(void)memcpy_s(host_ptr, size, ptr_, size);
auto ret_code = memcpy_s(host_ptr, size, ptr_, size);
if (ret_code != EOK) {
MS_LOG(ERROR) << "Failed to copy tensor!";
return false;
}
} else if (type == kNumberTypeFloat16) {
FloatToHalf(host_ptr, ptr_, size / 2);
} else if (type == kNumberTypeFloat64) {
FloatToDouble(host_ptr, ptr_, size / sizeof(double));
} else {
MS_LOG(ERROR) << "Types not match. Device type: " << TypeIdLabel(type_id_) << ", host type: " << TypeIdLabel(type)
<< ".";
<< "!";
return false;
}
return true;