From 936d3c6ef8d172dec46a77db6446c5edc31bfc67 Mon Sep 17 00:00:00 2001 From: wenkai Date: Thu, 11 Jun 2020 15:28:09 +0800 Subject: [PATCH] fix codex --- mindspore/ccsrc/device/cpu/cpu_device_address.cc | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/device/cpu/cpu_device_address.cc b/mindspore/ccsrc/device/cpu/cpu_device_address.cc index 1edb248341..7e808d6d3b 100644 --- a/mindspore/ccsrc/device/cpu/cpu_device_address.cc +++ b/mindspore/ccsrc/device/cpu/cpu_device_address.cc @@ -22,7 +22,10 @@ namespace device { namespace cpu { bool CPUDeviceAddress::SyncDeviceToHost(const std::vector & /*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 & /*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;