Packages
evision
0.2.6
1.0.1-rc.0
1.0.0
1.0.0-rc.0
0.2.17
0.2.17-rc2
0.2.17-rc1
0.2.16
0.2.16-pre.2
0.2.16-pre
0.2.15
0.2.14
0.2.13
0.2.12
0.2.11
0.2.9
0.2.8
0.2.7
0.2.6
0.2.5
0.2.4
0.2.3
0.2.2
0.2.2-rc2
0.2.1
0.2.0
0.1.39
0.1.38
0.1.37
0.1.36
0.1.35
0.1.34
0.1.33
0.1.32
retired
0.1.31
0.1.30
0.1.29
0.1.28
0.1.27
0.1.26
0.1.26-rc3
0.1.26-rc2
0.1.26-rc1
0.1.26-rc0
0.1.25
0.1.24
0.1.23
0.1.22
0.1.21
0.1.20
0.1.19
0.1.18
0.1.17
0.1.16
0.1.15
0.1.14
0.1.13
0.1.12
0.1.11
0.1.10
0.1.9
0.1.8
0.1.7
OpenCV-Erlang/Elixir binding.
Current section
Files
Jump to
Current section
Files
c_src/modules/evision_cuda.cc
#include "evision_cuda.h"
#ifdef CUDA_ENABLED
#include <cuda_runtime.h>
#include <cstring>
#include <cstdint>
#include <vector>
#include <iostream>
std::pair<std::vector<unsigned char>, int> get_cuda_ipc_handle(std::uintptr_t ptr) {
cudaIpcMemHandle_t ipc_handle;
cudaError_t status = cudaIpcGetMemHandle(&ipc_handle, reinterpret_cast<void*>(ptr));
// Assuming sizeof(cudaIpcMemHandle_t) is constant
const size_t size = sizeof(cudaIpcMemHandle_t);
// Copy the memory handle to a byte array
std::vector<unsigned char> result(size);
memcpy(result.data(), &ipc_handle, size);
return std::make_pair(result, status != cudaSuccess);
}
std::pair<void*, int> get_pointer_for_ipc_handle(std::vector<int64_t> handle_list, int device_id) {
if (handle_list.size() != sizeof(cudaIpcMemHandle_t)) {
printf("Error: Invalid CUDA IPC memory handle size\n");
return std::make_pair(nullptr, 1); // Return with error status
}
unsigned char ipc_handle_data[sizeof(cudaIpcMemHandle_t)];
for (int i = 0; i < sizeof(cudaIpcMemHandle_t); i++) {
ipc_handle_data[i] = (uint8_t)handle_list[i];
}
cudaIpcMemHandle_t ipc_handle;
memcpy(&ipc_handle, ipc_handle_data, sizeof(cudaIpcMemHandle_t));
int* ptr;
cudaError_t cuda_status = cudaSetDevice(device_id); // Assuming device 0, change as needed
if (cuda_status != cudaSuccess) {
printf("Error setting CUDA device: %s\n", cudaGetErrorString(cuda_status));
return std::make_pair(nullptr, 1); // Return with error status
}
cuda_status = cudaIpcOpenMemHandle((void**)&ptr, ipc_handle, cudaIpcMemLazyEnablePeerAccess);
if (cuda_status != cudaSuccess) {
printf("Error opening CUDA IPC memory handle: %s\n", cudaGetErrorString(cuda_status));
return std::make_pair(nullptr, 1); // Return with error status
}
return std::make_pair(ptr, cuda_status != cudaSuccess);
}
#else
std::pair<std::vector<unsigned char>, int> get_cuda_ipc_handle(std::uintptr_t ptr) {
return std::make_pair(std::vector<unsigned char>(0), 1);
}
std::pair<void*, int> get_pointer_for_ipc_handle(std::vector<int64_t> handle_list, int device_id) {
return std::make_pair(nullptr, 1);
}
#endif