forked from huawei/mindspore2022
!8873 Assign size when malloc memory
From: @wilfchen Reviewed-by: @cristoval,@limingqi107 Signed-off-by: @cristoval
This commit is contained in:
commit
122710303d
|
|
@ -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;
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue