!8873 Assign size when malloc memory

From: @wilfchen
Reviewed-by: @cristoval,@limingqi107
Signed-off-by: @cristoval
This commit is contained in:
mindspore-ci-bot 2020-11-23 11:24:06 +08:00 committed by Gitee
commit 122710303d
1 changed files with 1 additions and 0 deletions

View File

@ -220,6 +220,7 @@ bool MemoryManager::MallocMemFromMemPool(const DeviceAddressPtr address, size_t
return false;
}
address->ptr_ = device_ptr;
address->size_ = size;
address->from_mem_pool_ = true;
return true;
}