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

!3656 modify the wrong word of memory pool 

Merge pull request !3656 from limingqi107/master
This commit is contained in:
mindspore-ci-bot 2020-07-30 11:36:24 +08:00 committed by Gitee
parent 65c6755fab af39ca8252
commit 89dd5a865e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/backend/optimizer/mem_reuse/mem_dynamic_allocator.cc
View File

@ -191,7 +191,8 @@ void DynamicMemPoolBestFit::FreeTensorMem(const DeviceMemPtr device_addr) {
MS_EXCEPTION_IF_NULL(device_addr);
auto mem_block = FindMemBlock(device_addr);
if (mem_block == nullptr) {
MS_LOG(WARNING) << "Can't find the mem_block of the device address[" << device_addr << "].";
// May be destory the memory pool first, then destory the address, so this is normal case.
MS_LOG(DEBUG) << "Can't find the mem_block of the device address[" << device_addr << "].";
return;
}
CombineMemBuf(mem_block, device_addr);
@ -262,7 +263,7 @@ void DynamicMemPoolBestFit::EraseIdleMemBuf(size_t size, const DeviceMemPtr devi
}
void DynamicMemPoolBestFit::ReleaseDeviceRes() {
MS_LOG(INFO) << "The dynamic memmory pool total size is " << total_mem_statistics_ << ", total used size is "
MS_LOG(INFO) << "The dynamic memory pool total size is " << total_mem_statistics_ << ", total used size is "
<< total_used_mem_statistics_ << ", used peak size is " << used_mem_peak_statistics_ << ".";
for (auto iter = global_mem_block_list_.begin(); iter != global_mem_block_list_.end(); ++iter) {
auto device_addr = (*iter)->device_addr();