Home
last modified time | relevance | path

Searched refs:FreeTensorMem (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_device_address.cc35 CPUMemoryPool::GetInstance().FreeTensorMem(ptr_); in ClearDeviceMemory()
124 CPUMemoryPool::GetInstance().FreeTensorMem(ptr_); in SyncHostToDevice()
Dcpu_memory_manager.h50 …void FreeMemFromMemPool(void *device_ptr) override { CPUMemoryPool::GetInstance().FreeTensorMem(de… in FreeMemFromMemPool()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dgathernd_gpu_kernel.h34 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(dev_batch_strides… in ~GatherNdGpuFwdKernel()
37 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(dev_batch_indices… in ~GatherNdGpuFwdKernel()
Dtensor_scatter_min_gpu_kernel.h44 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(indices_stride_)); in ~TensorScatterMinGpuKernel()
47 … device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(work_shape_)); in ~TensorScatterMinGpuKernel()
Dtensor_scatter_sub_gpu_kernel.h45 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(indices_stride_)); in ~TensorScatterSubGpuKernel()
48 … device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(work_shape_)); in ~TensorScatterSubGpuKernel()
Dtensor_scatter_update_gpu_kernel.h45 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(indices_stride_)); in ~TensorScatterUpdateGpuFwdKernel()
48 … device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(work_shape_)); in ~TensorScatterUpdateGpuFwdKernel()
Dtensor_scatter_max_gpu_kernel.h45 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(indices_stride_)); in ~TensorScatterMaxGpuKernel()
48 … device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(work_shape_)); in ~TensorScatterMaxGpuKernel()
Dscatter_nd_gpu_kernel.h44 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(indices_stride_)); in ~ScatterNdGpuFwdKernel()
47 … device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(work_shape_)); in ~ScatterNdGpuFwdKernel()
Dtensor_scatter_add_gpu_kernel.h45 …device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(indices_stride_)); in ~TensorScatterAddGpuFwdKernel()
48 … device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(work_shape_)); in ~TensorScatterAddGpuFwdKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_launch_kernel.cc38 void GPULaunchkernel::FreeDeviceMem(void *addr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(a… in FreeDeviceMem()
Dgpu_memory_manager.cc32 GPUMemoryAllocator::GetInstance().FreeTensorMem(device_ptr); in FreeMemFromMemPool()
Dgpu_device_address.cc123 GPUMemoryAllocator::GetInstance().FreeTensorMem(ptr_); in ClearDeviceMemory()
Dgpu_bucket.cc82 void GPUBucket::FreeDeviceMem(void *dev_ptr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(dev_… in FreeDeviceMem()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_kernel.cc24 void AscendLaunchKernel::FreeDeviceMem(void *addr) { AscendMemoryPool::GetInstance().FreeTensorMem(… in FreeDeviceMem()
Dascend_bucket.cc75 void AscendBucket::FreeDeviceMem(void *dev_ptr) { AscendMemoryPool::GetInstance().FreeTensorMem(dev… in FreeDeviceMem()
Dascend_memory_manager.cc132 AscendMemoryPool::GetInstance().FreeTensorMem(device_ptr); in FreeMemFromMemPool()
Dascend_device_address.cc485 AscendMemoryPool::GetInstance().FreeTensorMem(communication_ptr_); in ClearDeviceMemory()
488 AscendMemoryPool::GetInstance().FreeTensorMem(ptr_); in ClearDeviceMemory()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/
Dbuffer_sample_gpu_kernel.cc37 device::gpu::GPUMemoryAllocator::GetInstance().FreeTensorMem(static_cast<void *>(devStates_)); in ~BufferSampleKernel()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.h91 void FreeTensorMem(const DeviceMemPtr &device_addr);
Dmem_dynamic_allocator.cc202 void DynamicMemPoolBestFit::FreeTensorMem(const DeviceMemPtr &device_addr) { in FreeTensorMem() function in mindspore::device::DynamicMemPoolBestFit