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

!11176 Opt get_func by uniforming format 

From: @liangzelang
Reviewed-by: @limingqi107,@kisnwang
Signed-off-by: @kisnwang
This commit is contained in:
mindspore-ci-bot 2021-01-12 16:36:10 +08:00 committed by Gitee
parent e805d06499 a97ac180ba
commit bc276774e3
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/optimizer/mem_reuse/mem_dynamic_allocator.h
View File

@ -87,7 +87,8 @@ class DynamicMemPoolBestFit {
void ReleaseDeviceRes();
// Display the information of memory block and memory buf.
void DumpDynamicMemPoolInfo();
SizeMapMemBuf GetIdleMemBufMap() { return global_idle_mem_buf_map_; }
// Get the map of global idle mem buf and size.
SizeMapMemBuf global_idle_mem_buf_map() const { return global_idle_mem_buf_map_; }
// Get the related memory statistics information.
size_t total_mem_statistics() const { return total_mem_statistics_; }

2
mindspore/ccsrc/runtime/device/ascend/ascend_memory_pool.cc
View File

@ -63,7 +63,7 @@ bool AscendMemoryPool::FreeDeviceMem(const DeviceMemPtr &addr) {
}
void AscendMemoryPool::ResetIdleMemBuf() {
auto idle_mem_buf_map = DynamicMemPoolBestFit::GetIdleMemBufMap();
auto idle_mem_buf_map = DynamicMemPoolBestFit::global_idle_mem_buf_map();
for (auto &it : idle_mem_buf_map) {
rtMemset(it.second->device_addr_, it.first, 0, it.first);
}