Home
last modified time | relevance | path

Searched refs:FreeDeviceMem (Results 1 – 25 of 33) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_allocator.cc62 if (!CudaDriver::FreeDeviceMem(buffer_q_addr_)) { in Finalize()
98 bool GPUMemoryAllocator::FreeDeviceMem(const DeviceMemPtr &addr) { return CudaDriver::FreeDeviceMem in FreeDeviceMem() function in mindspore::device::gpu::GPUMemoryAllocator
Dgpu_launch_mul.cc25 void GPULaunchMul::FreeDeviceMem(void *addr) { GPULaunchkernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::gpu::GPULaunchMul
Dgpu_bucket.cc82 void GPUBucket::FreeDeviceMem(void *dev_ptr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(dev_… in FreeDeviceMem() function in mindspore::device::gpu::GPUBucket
87 FreeDeviceMem(ar_input_addr_); in FreeAllDeviceMem()
91 FreeDeviceMem(ar_output_addr_); in FreeAllDeviceMem()
Dgpu_bucket.h35 void FreeDeviceMem(void *dev_ptr) override;
Dgpu_launch_kernel.h30 void FreeDeviceMem(void *addr) override;
Dgpu_memory_allocator.h36 bool FreeDeviceMem(const DeviceMemPtr &addr) override;
Dgpu_launch_mul.h32 void FreeDeviceMem(void *addr) override;
Dgpu_launch_kernel.cc38 void GPULaunchkernel::FreeDeviceMem(void *addr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(a… in FreeDeviceMem() function in mindspore::device::gpu::GPULaunchkernel
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_mul.cc24 void AscendLaunchMul::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchMul
Dascend_bucket.cc75 void AscendBucket::FreeDeviceMem(void *dev_ptr) { AscendMemoryPool::GetInstance().FreeTensorMem(dev… in FreeDeviceMem() function in mindspore::device::ascend::AscendBucket
80 FreeDeviceMem(origin_dev_addr); in FreeAllDeviceMem()
85 FreeDeviceMem(origin_dev_addr); in FreeAllDeviceMem()
Dascend_launch_atomic_clean.cc24 void AscendLaunchAtomicClean::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchAtomicClean
Dascend_launch_transdata.cc25 void AscendLaunchTransData::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchTransData
Dascend_bucket.h35 void FreeDeviceMem(void *dev_ptr) override;
Dascend_launch_kernel.h30 void FreeDeviceMem(void *addr) override;
Dascend_launch_mul.h33 void FreeDeviceMem(void *addr) override;
Dascend_launch_kernel.cc24 void AscendLaunchKernel::FreeDeviceMem(void *addr) { AscendMemoryPool::GetInstance().FreeTensorMem(… in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchKernel
Dascend_launch_atomic_clean.h36 void FreeDeviceMem(void *addr) override;
Dascend_memory_pool.h34 bool FreeDeviceMem(const DeviceMemPtr &addr) override;
Dascend_launch_transdata.h42 void FreeDeviceMem(void *addr) override;
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.cc96 FreeDeviceMem(outputs_addr_[i]); in FreeOutputAndWorkspaceDeviceMem()
102 FreeDeviceMem(workspaces_addr_[i]); in FreeOutputAndWorkspaceDeviceMem()
Dlaunch_mul.h36 virtual void FreeDeviceMem(void *addr) = 0;
Dlaunch_kernel.h34 virtual void FreeDeviceMem(void *addr) = 0;
Dlaunch_mul.cc79 FreeDeviceMem(input2_addr_); in FreeInputDeviceMemory()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_memory_pool.h37 bool FreeDeviceMem(const DeviceMemPtr &addr) override;
Dcpu_memory_pool.cc87 bool CPUMemoryPool::FreeDeviceMem(const DeviceMemPtr &addr) { in FreeDeviceMem() function in mindspore::device::cpu::CPUMemoryPool

12