Home
last modified time | relevance | path

Searched refs:device_ptr (Results 1 – 25 of 43) sorted by relevance

12

/third_party/boost/libs/compute/include/boost/compute/detail/
Ddevice_ptr.hpp61 class device_ptr class
71 device_ptr() in device_ptr() function in boost::compute::detail::device_ptr
76 device_ptr(const buffer &buffer, size_t index = 0) in device_ptr() function in boost::compute::detail::device_ptr
82 device_ptr(const device_ptr<T> &other) in device_ptr() function in boost::compute::detail::device_ptr
88 device_ptr<T>& operator=(const device_ptr<T> &other) in operator =()
99 device_ptr(device_ptr<T>&& other) BOOST_NOEXCEPT in device_ptr() function in boost::compute::detail::device_ptr
106 device_ptr<T>& operator=(device_ptr<T>&& other) BOOST_NOEXCEPT in operator =()
117 ~device_ptr() in ~device_ptr()
135 device_ptr<OT> cast() const in cast()
137 return device_ptr<OT>(m_buffer, m_index); in cast()
[all …]
Dbuffer_value.hpp144 detail::device_ptr<T> operator&() const in operator &()
146 return detail::device_ptr<T>(m_buffer, m_index); in operator &()
/third_party/boost/boost/compute/detail/
Ddevice_ptr.hpp61 class device_ptr class
71 device_ptr() in device_ptr() function in boost::compute::detail::device_ptr
76 device_ptr(const buffer &buffer, size_t index = 0) in device_ptr() function in boost::compute::detail::device_ptr
82 device_ptr(const device_ptr<T> &other) in device_ptr() function in boost::compute::detail::device_ptr
88 device_ptr<T>& operator=(const device_ptr<T> &other) in operator =()
99 device_ptr(device_ptr<T>&& other) BOOST_NOEXCEPT in device_ptr() function in boost::compute::detail::device_ptr
106 device_ptr<T>& operator=(device_ptr<T>&& other) BOOST_NOEXCEPT in operator =()
117 ~device_ptr() in ~device_ptr()
135 device_ptr<OT> cast() const in cast()
137 return device_ptr<OT>(m_buffer, m_index); in cast()
[all …]
Dbuffer_value.hpp144 detail::device_ptr<T> operator&() const in operator &()
146 return detail::device_ptr<T>(m_buffer, m_index); in operator &()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_allocator.cc36 void *device_ptr = nullptr; in MallocDeviceMem() local
37 auto cuda_ret = cudaMalloc(&device_ptr, size); in MallocDeviceMem()
47 cuda_ret = cudaFree(device_ptr); in MallocDeviceMem()
55 cuda_tensor_map_[name].data = device_ptr; in MallocDeviceMem()
58 return device_ptr; in MallocDeviceMem()
107 auto device_ptr = current_cuda_tensor.data; in SyncMemInHostAndDevice() local
108 if (device_ptr == nullptr) { in SyncMemInHostAndDevice()
113 void *src_ptr = is_host2device ? host_data : device_ptr; in SyncMemInHostAndDevice()
114 void *dst_ptr = is_host2device ? device_ptr : host_data; in SyncMemInHostAndDevice()
Dtensorrt_subgraph.cc319 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, tensor.DataSize()); in Prepare() local
320 if (device_ptr == nullptr) { in Prepare()
325 tensor_bindings_[index] = device_ptr; in Prepare()
345 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, tensor.DataSize()); in Prepare() local
346 if (device_ptr == nullptr) { in Prepare()
351 tensor_bindings_[index] = device_ptr; in Prepare()
389 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_in_tensor_name_[i], inputs_[i].Dat… in ReSize() local
391 if (device_ptr == nullptr) { in ReSize()
396 tensor_bindings_[index] = device_ptr; in ReSize()
415 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_out_tensor_name_[i], outputs_[i].D… in ReSize() local
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dmemory_scheduler.cc108 auto device_ptr = mem_handler_->MallocDevice(event->mem_size); in PreCompute() local
109 if (device_ptr == nullptr) { in PreCompute()
113 high_priority_device_ptr_[event->key] = device_ptr; in PreCompute()
115 mem_handler_->SwapIn(host_ptr, device_ptr, event->mem_size, stream); in PreCompute()
116 mem_result_[event->key] = device_ptr; in PreCompute()
118 auto device_ptr = mem_handler_->MallocDevice(event->mem_size); in PreCompute() local
119 if (device_ptr == nullptr) { in PreCompute()
122 mem_result_[event->key] = device_ptr; in PreCompute()
130 auto device_ptr = mem_handler_->MallocDevice(event->mem_size); in PreCompute() local
131 if (device_ptr == nullptr) { in PreCompute()
[all …]
Dmemory_manager.h59 virtual void FreeMemFromMemPool(void *device_ptr);
99 void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) override { in SwapIn() argument
100 …MS_LOG(INFO) << "Call default swap in " << host_ptr << "," << device_ptr << "," << mem_size << ","… in SwapIn()
102 void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) override { in SwapOut() argument
103 …MS_LOG(INFO) << "Call default swap out " << host_ptr << "," << device_ptr << "," << mem_size << ",… in SwapOut()
Dmemory_manager.cc146 auto device_ptr = MallocMemFromMemPool(size); in MallocMemFromMemPool() local
147 if (!device_ptr) { in MallocMemFromMemPool()
151 address->ptr_ = device_ptr; in MallocMemFromMemPool()
171 void MemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool() argument
172 if (device_ptr == nullptr) { in FreeMemFromMemPool()
Dmemory_scheduler.h34 virtual void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) = 0;
35 virtual void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) = 0;
/third_party/boost/libs/compute/include/boost/compute/experimental/
Dmalloc.hpp24 using detail::device_ptr;
27 inline device_ptr<T>
32 return device_ptr<T>(buf); in malloc()
35 inline device_ptr<char>
42 inline void free(device_ptr<T> &ptr) in free()
/third_party/boost/boost/compute/experimental/
Dmalloc.hpp24 using detail::device_ptr;
27 inline device_ptr<T>
32 return device_ptr<T>(buf); in malloc()
35 inline device_ptr<char>
42 inline void free(device_ptr<T> &ptr) in free()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_memory_manager.cc131 void AscendMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool() argument
132 AscendMemoryPool::GetInstance().FreeTensorMem(device_ptr); in FreeMemFromMemPool()
225 void AscendMemoryManager::SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *str… in SwapIn() argument
227 … auto ret_rt_memcpy = rtMemcpy(device_ptr, mem_size, host_ptr, mem_size, RT_MEMCPY_HOST_TO_DEVICE); in SwapIn()
232 …auto ret_rt_memcpy = rtMemcpyAsync(device_ptr, mem_size, host_ptr, mem_size, RT_MEMCPY_HOST_TO_DEV… in SwapIn()
242 void AscendMemoryManager::SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *st… in SwapOut() argument
244 … auto ret_rt_memcpy = rtMemcpy(host_ptr, mem_size, device_ptr, mem_size, RT_MEMCPY_DEVICE_TO_HOST); in SwapOut()
249 …auto ret_rt_memcpy = rtMemcpyAsync(host_ptr, mem_size, device_ptr, mem_size, RT_MEMCPY_DEVICE_TO_H… in SwapOut()
Dascend_memory_manager.h36 void FreeMemFromMemPool(void *device_ptr) override;
44 void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) override;
45 void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) override;
Dascend_kernel_runtime.h76 DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format,
78 …DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format, T…
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_manager.cc31 void GPUMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool() argument
32 GPUMemoryAllocator::GetInstance().FreeTensorMem(device_ptr); in FreeMemFromMemPool()
109 auto device_ptr = MallocMemFromMemPool(size); in MallocStaticMem() local
110 if (device_ptr == nullptr) { in MallocStaticMem()
113 return AddressOffset(device_ptr, 0); in MallocStaticMem()
Dgpu_memory_copy_manager.cc37 DeviceMemPtr device_ptr = const_cast<DeviceMemPtr>(device_address->GetPtr()); in AddMemSwapOutTask() local
38 MS_EXCEPTION_IF_NULL(device_ptr); in AddMemSwapOutTask()
42 …CudaDriver::CopyDeviceMemToHostAsync(host_addr.addr, device_ptr, host_addr.size, swap_out_stream_), in AddMemSwapOutTask()
64 DeviceMemPtr device_ptr = const_cast<DeviceMemPtr>(device_address->GetPtr()); in AddMemSwapInTask() local
65 MS_EXCEPTION_IF_NULL(device_ptr); in AddMemSwapInTask()
69 … CudaDriver::CopyHostMemToDeviceAsync(device_ptr, host_addr.addr, host_addr.size, swap_in_stream_), in AddMemSwapInTask()
/third_party/boost/libs/compute/include/boost/compute/allocator/
Dbuffer_allocator.hpp31 typedef detail::device_ptr<T> pointer;
32 typedef const detail::device_ptr<T> const_pointer;
82 return detail::device_ptr<T>(buf); in allocate()
/third_party/boost/boost/compute/allocator/
Dbuffer_allocator.hpp31 typedef detail::device_ptr<T> pointer;
32 typedef const detail::device_ptr<T> const_pointer;
82 return detail::device_ptr<T>(buf); in allocate()
/third_party/gstreamer/gstplugins_bad/sys/nvcodec/
Dcuda-converter.c726 CUdeviceptr device_ptr; member
876 if (convert->fallback_buffer[i].device_ptr) in gst_cuda_converter_free()
877 gst_cuda_result (CuMemFree (convert->fallback_buffer[i].device_ptr)); in gst_cuda_converter_free()
878 if (convert->y444_surface[i].device_ptr) in gst_cuda_converter_free()
879 gst_cuda_result (CuMemFree (convert->y444_surface[i].device_ptr)); in gst_cuda_converter_free()
882 if (convert->unpack_surface.device_ptr) in gst_cuda_converter_free()
883 gst_cuda_result (CuMemFree (convert->unpack_surface.device_ptr)); in gst_cuda_converter_free()
965 if (convert->fallback_buffer[plane].device_ptr) in convert_ensure_fallback_memory()
971 ret = CuMemAllocPitch (&convert->fallback_buffer[plane].device_ptr, in convert_ensure_fallback_memory()
1059 copy_params.dstDevice = convert->fallback_buffer[plane].device_ptr; in convert_create_texture()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc81 auto device_ptr = mem_manager_->MallocMemFromMemPool(size); in AllocateMemory() local
82 if (!device_ptr) { in AllocateMemory()
85 address->ptr_ = device_ptr; in AllocateMemory()
102 DeviceAddressPtr CPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress() argument
104 return std::make_shared<CPUDeviceAddress>(device_ptr, device_size, format, type_id); in CreateDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_memory_manager.h50 …void FreeMemFromMemPool(void *device_ptr) override { CPUMemoryPool::GetInstance().FreeTensorMem(de… in FreeMemFromMemPool() argument
Dcpu_kernel_runtime.h54 DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format,
56 …DeviceAddressPtr CreateDeviceAddress(void *device_ptr, size_t device_size, const string &format, T…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dminmax_update_impl.cu74 thrust::pair<thrust::device_ptr<float>, thrust::device_ptr<float>> tuple; in CalMinMaxPerLayer()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc171 auto device_ptr = mem_manager_->MallocMemFromMemPool(size); in AllocateMemory() local
172 if (!device_ptr) { in AllocateMemory()
175 address->ptr_ = device_ptr; in AllocateMemory()
199 DeviceAddressPtr GPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress() argument
201 return std::make_shared<GPUDeviceAddress>(device_ptr, device_size, format, type_id); in CreateDeviceAddress()

12