Searched refs:AllocTensorMem (Results 1 – 18 of 18) sorted by relevance
128 return AscendMemoryPool::GetInstance().AllocTensorMem(align_size); in MallocMemFromMemPool()161 …_address = reinterpret_cast<uint8_t *>(AscendMemoryPool::GetInstance().AllocTensorMem(align_size)); in MallocStaticMem()164 return reinterpret_cast<uint8_t *>(AscendMemoryPool::GetInstance().AllocTensorMem(align_size)); in MallocStaticMem()214 …uint8_t *base_ptr = reinterpret_cast<uint8_t *>(AscendMemoryPool::GetInstance().AllocTensorMem(ali… in MallocCommunicationMemFromMemPool()
29 auto device_memory = AscendMemoryPool::GetInstance().AllocTensorMem(size); in AllocDeviceMem()
71 …ar_input_addr_ = static_cast<uint8_t *>(GPUMemoryAllocator::GetInstance().AllocTensorMem(total_siz… in AllocateAllReduceAddr()72 …ar_output_addr_ = static_cast<uint8_t *>(GPUMemoryAllocator::GetInstance().AllocTensorMem(total_si… in AllocateAllReduceAddr()
43 auto device_memory = GPUMemoryAllocator::GetInstance().AllocTensorMem(size); in AllocDeviceMem()
28 return GPUMemoryAllocator::GetInstance().AllocTensorMem(size); in MallocMemFromMemPool()
107 …void *dev_batch_strides_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(strid… in Init()114 …void *dev_batch_indices_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(indic… in Init()
117 …void *indices_stride_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(indices_… in Init()124 …void *work_shape_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(vec_work_len… in Init()
118 …void *indices_stride_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(indices_… in Init()125 …void *work_shape_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(vec_work_len… in Init()
127 …void *indices_stride_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(indices_… in Init()134 …void *work_shape_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(vec_work_len… in Init()
123 …void *indices_stride_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(indices_… in Init()130 …void *work_shape_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(vec_work_len… in Init()
128 …void *indices_stride_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(indices_… in Init()135 …void *work_shape_work = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(vec_work_len… in Init()
132 return device::ascend::AscendMemoryPool::GetInstance().AllocTensorMem(size); in MallocMemory()136 …interpret_cast<int *>(device::ascend::AscendMemoryPool::GetInstance().AllocTensorMem(sizeof(int))); in MallocConstantMemory()140 …reinterpret_cast<int *>(device::ascend::AscendMemoryPool::GetInstance().AllocTensorMem(sizeof(int)… in MallocConstantMemory()
49 …cMemFromMemPool(size_t size) override { return CPUMemoryPool::GetInstance().AllocTensorMem(size); } in MallocMemFromMemPool()
41 return device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(size); in MallocMemory()
29 DeviceMemPtr DynamicMemPoolBestFit::AllocTensorMem(size_t size) { in AllocTensorMem() function in mindspore::device::DynamicMemPoolBestFit44 auto device_addr = AllocTensorMem(total_size); in AllocContinuousTensorMem()
87 DeviceMemPtr AllocTensorMem(size_t size);
67 void *dev_state = device::gpu::GPUMemoryAllocator::GetInstance().AllocTensorMem(cap_state_size); in Init()