Home
last modified time | relevance | path

Searched refs:MallocMemFromMemPool (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_manager.cc27 void *GPUMemoryManager::MallocMemFromMemPool(size_t size) { in MallocMemFromMemPool() function in mindspore::device::gpu::GPUMemoryManager
81 auto device_addr = MallocMemFromMemPool(1); in MallocDeviceMemory()
109 auto device_ptr = MallocMemFromMemPool(size); in MallocStaticMem()
Dgpu_memory_manager.h32 void *MallocMemFromMemPool(size_t size) override;
Dgpu_kernel_runtime.cc344 auto ret = mem_manager_->MallocMemFromMemPool(device_address, output_size[output_index]); in AllocInplaceNodeMemory()
1056 auto ret = mem_manager_->MallocMemFromMemPool(device_address, size); in AttemptMallocMem()
1066 ret = mem_manager_->MallocMemFromMemPool(device_address, size); in AttemptMallocMem()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dmemory_manager.h55 virtual bool MallocMemFromMemPool(const DeviceAddressPtr address, size_t size);
56 virtual void *MallocMemFromMemPool(size_t size);
68 void *MallocDevice(size_t mem_size) override { return MallocMemFromMemPool(mem_size); } in MallocDevice()
Dmemory_manager.cc144 bool MemoryManager::MallocMemFromMemPool(const DeviceAddressPtr address, size_t size) { in MallocMemFromMemPool() function in mindspore::device::MemoryManager
146 auto device_ptr = MallocMemFromMemPool(size); in MallocMemFromMemPool()
157 void *MemoryManager::MallocMemFromMemPool(size_t size) { in MallocMemFromMemPool() function in mindspore::device::MemoryManager
Dkernel_runtime.cc410 if (!mem_manager_->MallocMemFromMemPool(output_address, output_address->size())) { in RunOpAssignInputMemory()
427 auto ret = mem_manager_->MallocMemFromMemPool(device_address, tensor_size); in RunOpAssignInputMemory()
460 if (!mem_manager_->MallocMemFromMemPool(address, address->size())) { in RunOpAssignOutputMemory()
476 auto ret = mem_manager_->MallocMemFromMemPool(device_address, output_sizes[i]); in RunOpAssignOutputMemory()
494 auto ret = mem_manager_->MallocMemFromMemPool(device_address, workspace_lists[i]); in RunOpAssignWorkSpaceMemory()
982 !mem_manager_->MallocMemFromMemPool(address, node_size)) { in AssignValueNodeTensor()
1021 if (!mem_manager_->MallocMemFromMemPool(device_address, device_address->size_)) { in AssignStaticMemoryValueNode()
1044 !mem_manager_->MallocMemFromMemPool(address, tensor_size)) { in AssignStaticMemoryValueNode()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_memory_manager.h35 void *MallocMemFromMemPool(size_t size) override;
Dascend_memory_manager.cc126 void *AscendMemoryManager::MallocMemFromMemPool(size_t size) { in MallocMemFromMemPool() function in mindspore::device::ascend::AscendMemoryManager
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_memory_manager.h49 …void *MallocMemFromMemPool(size_t size) override { return CPUMemoryPool::GetInstance().AllocTensor… in MallocMemFromMemPool() function
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc81 auto device_ptr = mem_manager_->MallocMemFromMemPool(size); in AllocateMemory()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc171 auto device_ptr = mem_manager_->MallocMemFromMemPool(size); in AllocateMemory()