| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/ |
| D | tensorrt_allocator.cc | 34 void *device_ptr = nullptr; in MallocDeviceMem() local
97 auto device_ptr = current_cuda_tensor.data; in SyncMemDeviceToHost() local
139 auto device_ptr = current_cuda_tensor.data; in SyncMemDeviceToHost() local
166 auto device_ptr = current_cuda_tensor.data; in SyncMemInHostAndDevice() local
201 auto device_ptr = current_cuda_tensor.data; in SyncMemInHostAndDevice() local
|
| D | tensorrt_subgraph.cc | 537 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, volumn * type_size); in Prepare() local
575 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, elem_num * type_size); in Prepare() local
782 void *device_ptr = nullptr; in PreExecute() local
811 void *device_ptr = nullptr; in PreExecute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_device_synchronizer.cc | 23 bool CPUDeviceSynchronizer::SyncDeviceToHost(void *host_ptr, const void *device_ptr, size_t size, in SyncDeviceToHost()
40 bool CPUDeviceSynchronizer::SyncHostToDevice(void *device_ptr, const void *host_ptr, size_t size, in SyncHostToDevice()
|
| D | cpu_memory_manager.h | 53 …void FreeMemFromMemPool(void *device_ptr) override { CPUMemoryPool::GetInstance().FreeTensorMem(de… in FreeMemFromMemPool()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_device_synchronizer.cc | 25 bool GPUDeviceSynchronizer::SyncDeviceToHost(void *host_ptr, const void *device_ptr, size_t size, in SyncDeviceToHost()
51 bool GPUDeviceSynchronizer::SyncHostToDevice(void *device_ptr, const void *host_ptr, size_t size, in SyncHostToDevice()
|
| D | gpu_memory_copy_manager.cc | 38 DeviceMemPtr device_ptr = const_cast<DeviceMemPtr>(device_address->GetPtr()); in AddMemSwapOutTask() local
65 DeviceMemPtr device_ptr = const_cast<DeviceMemPtr>(device_address->GetPtr()); in AddMemSwapInTask() local
|
| D | gpu_memory_manager.cc | 30 void GPUMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()
132 auto device_ptr = MallocMemFromMemPool(size, false); in MallocStaticMem() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | ascend_device_synchronizer.cc | 27 bool AscendDeviceSynchronizer::SyncDeviceToHost(void *host_ptr, const void *device_ptr, size_t size, in SyncDeviceToHost()
60 bool AscendDeviceSynchronizer::SyncHostToDevice(void *device_ptr, const void *host_ptr, size_t size, in SyncHostToDevice()
|
| D | ascend_memory_manager.cc | 57 void AscendMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()
170 void AscendMemoryManager::SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *str… in SwapIn()
189 void AscendMemoryManager::SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *st… in SwapOut()
|
| D | ascend_kernel_runtime.cc | 381 DeviceAddressPtr AscendKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, con… in CreateDeviceAddress()
393 DeviceAddressPtr AscendKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, con… in CreateDeviceAddress()
531 … auto device_ptr = runtime_instance->MallocMem(MemType::kDynamicMem, size, device_address_ptr); in RunDynamicKernelAsync() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | auto_mem_offload.cc | 86 auto device_ptr = Malloc(key, mem_size, stream, pinned_memory); in Get() local
111 auto device_ptr = mem_handler->MallocContinuousMemFromMemPool(size_list); in MallocContinuous() local
145 auto device_ptr = mem_handler->MallocDevice(size); in Malloc() local
214 const auto device_ptr = iter->second; in SwapOut() local
310 const auto &device_ptr = Malloc(align_size, pinned_mem); in Malloc() local
322 const auto malloc_func = [](size_t size, DynamicMemPoolBestFit *mem_pool, void **device_ptr) { in Malloc()
326 void *device_ptr = nullptr; in Malloc() local
|
| D | auto_mem_offload.h | 56 void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) { in SwapIn()
59 void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) { in SwapOut()
|
| D | memory_manager.h | 73 virtual void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) { in SwapIn()
76 virtual void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) { in SwapOut()
|
| D | memory_manager.cc | 123 …auto device_ptr = MallocMemFromMemPool(size, address->from_persistent_mem_, false, address->stream… in MallocMemFromMemPool() local
168 void MemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()
|
| D | memory_scheduler.cc | 130 void *device_ptr = nullptr; in PreComputeMock() local
142 auto device_ptr = auto_mem_offload_->Get(event->key); in PreComputeInit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/acl/ |
| D | acl_lite_kernel.cc | 107 auto device_ptr = input_device_address; in Run() local
140 auto device_ptr = output_device_address; in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/hardware/ |
| D | device_context.cc | 36 auto device_ptr = AllocateMemory(address->GetSize(), stream_id); in AllocateMemory() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | ge_device_res_manager.cc | 128 void *device_ptr = nullptr; in AllocateMemory() local
242 void GeDeviceResManager::SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stre… in SwapIn()
246 void GeDeviceResManager::SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *str… in SwapOut()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | kernel.cc | 92 KernelTensor::KernelTensor(void *device_ptr, size_t size, Format format, TypeId dtype_id, const Sha… in KernelTensor()
106 … void *device_ptr, size_t size, const std::string &format, TypeId dtype_id, in KernelTensor()
536 void *device_ptr = this->device_ptr(); in SyncDataFromDeviceToHost() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_embedding_operation.cc | 180 void *device_ptr = dst; in MemcpyHostToDeviceAsync() local
202 void *device_ptr = const_cast<void *>(src); in MemcpyDeviceToHostAsync() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/ |
| D | runtime_utils.cc | 66 device::DeviceAddressPtr RuntimeUtils::CreateDeviceAddress(void *device_ptr, size_t device_size, co… in CreateDeviceAddress()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | memory_manager_actor.cc | 220 …auto device_ptr = device_context->device_res_manager_->AllocateMemory(somas_info->whole_block_size… in AllocateSomasMemory() local
249 auto device_ptr = device_context->device_res_manager_->AllocateMemory(block_size); in AllocateSomasMemory() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/device/ |
| D | mem_scheduler_test.cc | 67 void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) override {} in SwapIn()
69 void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) override {} in SwapOut()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/gpu/ |
| D | gpu_cache_mem.cc | 50 void *device_ptr = nullptr; in MallocMemory() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/gpu/ |
| D | gpu_cache_mem.cc | 51 void *device_ptr = nullptr; in MallocMemory() local
|