Searched refs:AscendMemoryManager (Results 1 – 4 of 4) sorted by relevance
55 void AscendMemoryManager::MallocDeviceMemory() { in MallocDeviceMemory()77 uint64_t AscendMemoryManager::GetDeviceMemSize() { in GetDeviceMemSize()82 uint64_t AscendMemoryManager::GetDeviceMemSizeFromContext() { in GetDeviceMemSizeFromContext()108 void AscendMemoryManager::FreeDeviceMemory() { in FreeDeviceMemory()118 void AscendMemoryManager::ResetDynamicMemory() { in ResetDynamicMemory()124 void AscendMemoryManager::ClearGlobalIdleMem() { AscendMemoryPool::GetInstance().ResetIdleMemBuf();… in ClearGlobalIdleMem()126 void *AscendMemoryManager::MallocMemFromMemPool(size_t size) { in MallocMemFromMemPool()131 void AscendMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()135 uint8_t *AscendMemoryManager::MallocStaticMem(size_t size, bool communication_mem, uint32_t graph_i… in MallocStaticMem()168 uint8_t *AscendMemoryManager::MallocDynamicMem(size_t size, bool communication_mem) { in MallocDynamicMem()[all …]
26 class AscendMemoryManager : public MemoryManager {28 AscendMemoryManager() = default;29 ~AscendMemoryManager() override = default;
351 mem_manager_ = std::make_shared<AscendMemoryManager>(); in Init()1188 auto ascend_mem_manager = std::dynamic_pointer_cast<AscendMemoryManager>(mem_manager_); in GetAvailableMemMaxSize()
40 class AscendMemoryManager; variable126 friend class mindspore::device::ascend::AscendMemoryManager;