| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_memory_allocator.cc | 71 auto alloc_size = AllocDeviceMem(size, addr); in AllocBufferQueueMem()
77 size_t GPUMemoryAllocator::AllocDeviceMem(size_t size, DeviceMemPtr *addr) { in AllocDeviceMem() function in mindspore::device::gpu::GPUMemoryAllocator
87 auto alloc_size = CudaDriver::AllocDeviceMem(size, addr); in AllocDeviceMem()
|
| D | gpu_launch_mul.cc | 29 uint8_t *GPULaunchMul::AllocDeviceMem(size_t size) { return GPULaunchkernel::AllocDeviceMem(size); } in AllocDeviceMem() function in mindspore::device::gpu::GPULaunchMul
|
| D | gpu_launch_kernel.h | 32 uint8_t *AllocDeviceMem(size_t size) override;
|
| D | gpu_memory_allocator.h | 35 size_t AllocDeviceMem(size_t size, DeviceMemPtr *addr) override;
|
| D | gpu_launch_mul.h | 34 uint8_t *AllocDeviceMem(size_t size) override;
|
| D | gpu_launch_kernel.cc | 42 uint8_t *GPULaunchkernel::AllocDeviceMem(size_t size) { in AllocDeviceMem() function in mindspore::device::gpu::GPULaunchkernel
|
| D | cuda_driver.h | 34 static size_t AllocDeviceMem(size_t size, DeviceMemPtr *addr);
|
| D | gpu_memory_manager.cc | 90 …GPUMemoryAllocator::GetInstance().AllocDeviceMem(mem_size, reinterpret_cast<void **>(&device_mem_b… in MallocDeviceMemory()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_mul.cc | 30 uint8_t *AscendLaunchMul::AllocDeviceMem(size_t size) { return AscendLaunchKernel::AllocDeviceMem(s… in AllocDeviceMem() function in mindspore::device::ascend::AscendLaunchMul
|
| D | ascend_launch_atomic_clean.cc | 30 uint8_t *AscendLaunchAtomicClean::AllocDeviceMem(size_t size) { return AscendLaunchKernel::AllocDev… in AllocDeviceMem() function in mindspore::device::ascend::AscendLaunchAtomicClean
|
| D | ascend_launch_transdata.cc | 31 uint8_t *AscendLaunchTransData::AllocDeviceMem(size_t size) { return AscendLaunchKernel::AllocDevic… in AllocDeviceMem() function in mindspore::device::ascend::AscendLaunchTransData
|
| D | ascend_launch_kernel.h | 32 uint8_t *AllocDeviceMem(size_t size) override;
|
| D | ascend_launch_mul.h | 35 uint8_t *AllocDeviceMem(size_t size) override;
|
| D | ascend_launch_kernel.cc | 28 uint8_t *AscendLaunchKernel::AllocDeviceMem(size_t size) { in AllocDeviceMem() function in mindspore::device::ascend::AscendLaunchKernel
|
| D | ascend_launch_atomic_clean.h | 38 uint8_t *AllocDeviceMem(size_t size) override;
|
| D | ascend_memory_pool.h | 33 size_t AllocDeviceMem(size_t size, DeviceMemPtr *addr) override;
|
| D | ascend_launch_transdata.h | 44 uint8_t *AllocDeviceMem(size_t size) override;
|
| D | ascend_memory_pool.cc | 79 size_t AscendMemoryPool::AllocDeviceMem(size_t size, DeviceMemPtr *addr) { in AllocDeviceMem() function in mindspore::device::ascend::AscendMemoryPool
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_memory_pool.h | 36 size_t AllocDeviceMem(size_t size, DeviceMemPtr *addr) override;
|
| D | cpu_memory_pool.cc | 70 size_t CPUMemoryPool::AllocDeviceMem(size_t alloc_size, DeviceMemPtr *addr) { in AllocDeviceMem() function in mindspore::device::cpu::CPUMemoryPool
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.h | 38 virtual uint8_t *AllocDeviceMem(size_t size) = 0;
|
| D | launch_kernel.h | 36 virtual uint8_t *AllocDeviceMem(size_t size) = 0;
|
| D | launch_mul.cc | 71 input2_addr_ = AllocDeviceMem(input_size); in ObtainMulInputsAddr()
|
| D | launch_kernel.cc | 29 (*addr)[i] = AllocDeviceMem(size); in ObtainKernelAddress()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_dynamic_allocator.h | 114 virtual size_t AllocDeviceMem(size_t size, DeviceMemPtr *addr) = 0;
|