Searched refs:CPUMemoryManager (Results 1 – 5 of 5) sorted by relevance
25 uint8_t *CPUMemoryManager::MemMalloc(size_t size) { in MemMalloc()37 uint8_t *CPUMemoryManager::MallocStaticMem(size_t size, bool, uint32_t) { in MallocStaticMem()43 uint8_t *CPUMemoryManager::MallocDynamicMem(size_t size, bool) { in MallocDynamicMem()67 void CPUMemoryManager::ResetDynamicMemory() { in ResetDynamicMemory()75 CPUMemoryManager::~CPUMemoryManager() { MemFree(); } in ~CPUMemoryManager()77 void CPUMemoryManager::MemFree() noexcept { in MemFree()88 void CPUMemoryManager::AssignMemory(const session::KernelGraph *graph) { in AssignMemory()111 void *CPUMemoryManager::StaticMemMalloc(size_t mem_size) { in StaticMemMalloc()121 void CPUMemoryManager::MemFree(void *ptr) { in MemFree()132 void CPUMemoryManager::IncreaseSummaryRefCount(const session::NamedSummaryOutputs &summary_outputs)… in IncreaseSummaryRefCount()[all …]
32 class CPUMemoryManager : public MemoryManager {34 CPUMemoryManager() = default;35 virtual ~CPUMemoryManager();
55 mem_manager_ = std::make_shared<CPUMemoryManager>(); in Init()83 static_cast<CPUMemoryManager *>(mem_manager_.get())->AssignMemory(kernel_graph); in AssignKernelAddress()119 … address->ptr_ = static_cast<CPUMemoryManager *>(mem_manager_.get())->StaticMemMalloc(tensor_size); in AssignValueNodeAddress()230 … address->ptr_ = static_cast<CPUMemoryManager *>(mem_manager_.get())->StaticMemMalloc(tensor_size); in CreatTensorForOutput()339 … address->ptr_ = static_cast<CPUMemoryManager *>(mem_manager_.get())->StaticMemMalloc(tensor_size); in BindInputTensorAddressPtr()397 …address->ptr_ = static_cast<CPUMemoryManager *>(mem_manager_.get())->StaticMemMalloc(address->size… in AddRuntimeAddress()406 static_cast<CPUMemoryManager *>(mem_manager_.get())->IncreaseSummaryRefCount(summary_outputs); in IncreaseSummaryRefCount()410 static_cast<CPUMemoryManager *>(mem_manager_.get())->DecreaseSummaryRefCount(summary_outputs); in DecreaseSummaryRefCount()414 static_cast<CPUMemoryManager *>(mem_manager_.get())->IncreaseAddressRefCount(&kernel_graph); in Run()489 static_cast<CPUMemoryManager *>(mem_manager_.get())->DecreaseAddressRefCount(kernel); in Run()
34 class CPUMemoryManager; variable119 friend class mindspore::device::cpu::CPUMemoryManager;
55 mem_manager_ = std::make_shared<CPUMemoryManager>(); in Initialize()