Home
last modified time | relevance | path

Searched refs:block_all_mem_buf_map_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.cc52 const auto &iter = mem_block->block_all_mem_buf_map_.find(device_addr); in AllocContinuousTensorMem()
53 if (iter == mem_block->block_all_mem_buf_map_.end()) { in AllocContinuousTensorMem()
62 (void)mem_block->block_all_mem_buf_map_.erase(iter); in AllocContinuousTensorMem()
68 (void)mem_block->block_all_mem_buf_map_.emplace(buf_addr, continuous_mem_buf); in AllocContinuousTensorMem()
133 (void)mem_block->block_all_mem_buf_map_.emplace(device_addr, mem_buf); in AddMemBlockAndMemBuf()
181 (void)mem_block->block_all_mem_buf_map_.emplace(newbuf_addr, new_mem_buf); in DivideMemBuf()
217 const auto &iter = mem_block->block_all_mem_buf_map_.find(device_addr); in CombineMemBuf()
218 if (iter == mem_block->block_all_mem_buf_map_.end()) { in CombineMemBuf()
234 if (next_iter != mem_block->block_all_mem_buf_map_.end()) { in CombineMemBuf()
240 (void)mem_block->block_all_mem_buf_map_.erase(next_iter); in CombineMemBuf()
[all …]
Dmem_dynamic_allocator.h65 ~DynamicMemBlock() { block_all_mem_buf_map_.clear(); } in ~DynamicMemBlock()
73 DeviceAddrMapMemBuf block_all_mem_buf_map_; variable