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

[bugfix]LSTM SyncDeviceToHost failed

This commit is contained in:
lizhenyu 2020-08-24 20:13:23 +08:00
parent 891228bcbe
commit 68b37ee1b3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/runtime/device/gpu/gpu_device_address.cc
View File

@ -32,7 +32,7 @@ namespace device {
namespace gpu {
bool GPUDeviceAddress::SyncDeviceToHost(const std::vector<int> &, size_t size, TypeId, void *host_ptr) const {
MS_EXCEPTION_IF_NULL(host_ptr);
bool need_sync = (size != 0) && (size_ != 0);
bool need_sync = (size != 0) && (size_ != 0) && (size <= size_);
if (!need_sync) {
return true;
}
@ -52,7 +52,7 @@ bool GPUDeviceAddress::SyncDeviceToHost(const std::vector<int> &, size_t size, T
bool GPUDeviceAddress::SyncHostToDevice(const std::vector<int> &, size_t size, TypeId, const void *host_ptr) const {
MS_EXCEPTION_IF_NULL(host_ptr);
bool need_sync = (size != 0) && (size_ != 0);
bool need_sync = (size != 0) && (size_ != 0) && (size <= size_);
if (!need_sync) {
return true;
}