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

fix bug of CPU workspace address must not be nullptr

This commit is contained in:
limingqi107 2021-07-08 23:09:09 +08:00
parent 04956a776c
commit 863e25913e
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/framework/actor/memory_manager_actor.cc
View File

@ -30,7 +30,7 @@ void MemoryManagerActor::AllocateMemory(std::vector<DeviceTensor *> *alloc_list,
for (auto &device_tensor : *alloc_list) {
MS_EXCEPTION_IF_NULL(device_tensor);
if ((device_tensor->GetPtr() != nullptr) || (device_tensor->GetSize() == 0)) {
if (device_tensor->GetPtr() != nullptr) {
continue;
}
// Allocate memory through the device context.
@ -97,7 +97,7 @@ void MemoryManagerActor::AllocateBatchMemory(std::vector<DeviceTensor *> *alloc_
auto &device_context = (*device_contexts)[i];
MS_EXCEPTION_IF_NULL(device_tensor);
MS_EXCEPTION_IF_NULL(device_context);
if ((device_tensor->GetPtr() != nullptr) || (device_tensor->GetSize() == 0)) {
if (device_tensor->GetPtr() != nullptr) {
continue;
}