Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.h61 class DynamicMemBlock {
63 DynamicMemBlock() = default;
64DynamicMemBlock(DeviceMemPtr addr_base, size_t size) : device_addr_base_(addr_base), mem_block_siz… in DynamicMemBlock() function
65 ~DynamicMemBlock() { block_all_mem_buf_map_.clear(); } in ~DynamicMemBlock()
78 using DynamicMemBlockPtr = std::shared_ptr<DynamicMemBlock>;
Dmem_dynamic_allocator.cc124 auto mem_block = std::make_shared<DynamicMemBlock>(device_addr, real_alloc_size); in AddMemBlockAndMemBuf()