Browse Source

!2005 fix codex warnings

Merge pull request !2005 from wenkai/wk_0610_fix_codex
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
a6a9f88453
1 changed files with 10 additions and 3 deletions
  1. +10
    -3
      mindspore/ccsrc/device/cpu/cpu_device_address.cc

+ 10
- 3
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;


Loading…
Cancel
Save