Searched refs:GPUMemoryManager (Results 1 – 5 of 5) sorted by relevance
27 void *GPUMemoryManager::MallocMemFromMemPool(size_t size) { in MallocMemFromMemPool()31 void GPUMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()35 std::vector<void *> GPUMemoryManager::MallocContinuousMemFromMemPool(size_t total_size, std::vector… in MallocContinuousMemFromMemPool()39 bool GPUMemoryManager::MallocContinuousMemFromMemPool(const DeviceAddressPtrList &addr_list, size_t… in MallocContinuousMemFromMemPool()73 void GPUMemoryManager::MallocDeviceMemory() { in MallocDeviceMemory()96 void GPUMemoryManager::FreeDeviceMemory() { in FreeDeviceMemory()105 uint8_t *GPUMemoryManager::MallocStaticMem(size_t size, bool, uint32_t) { in MallocStaticMem()
24 class GPUMemoryManager : public MemoryManager {26 GPUMemoryManager() = default;27 virtual ~GPUMemoryManager() = default;
95 mem_manager_ = std::make_shared<GPUMemoryManager>(); in Init()
50 class GPUMemoryManager; variable123 friend class mindspore::device::gpu::GPUMemoryManager;
88 mem_manager_ = std::make_shared<GPUMemoryManager>(); in Initialize()