aboutsummaryrefslogtreecommitdiff
path: root/include/CudaLibrary.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/CudaLibrary.hpp')
-rw-r--r--include/CudaLibrary.hpp148
1 files changed, 47 insertions, 101 deletions
diff --git a/include/CudaLibrary.hpp b/include/CudaLibrary.hpp
index 134d19c..e45aa44 100644
--- a/include/CudaLibrary.hpp
+++ b/include/CudaLibrary.hpp
@@ -1,42 +1,29 @@
#pragma once
+#include "LibraryLoader.hpp"
+
#include <cuda.h>
-#include <cudaGL.h>
#include <dlfcn.h>
#include <stdio.h>
-typedef CUresult CUDAAPI (*CUINIT)(unsigned int Flags);
-typedef CUresult CUDAAPI (*CUDEVICEGETCOUNT)(int *count);
-typedef CUresult CUDAAPI (*CUDEVICEGET)(CUdevice *device, int ordinal);
-typedef CUresult CUDAAPI (*CUCTXCREATE_V2)(CUcontext *pctx, unsigned int flags, CUdevice dev);
-typedef CUresult CUDAAPI (*CUCTXPUSHCURRENT_V2)(CUcontext ctx);
-typedef CUresult CUDAAPI (*CUCTXPOPCURRENT_V2)(CUcontext *pctx);
-typedef CUresult CUDAAPI (*CUGETERRORSTRING)(CUresult error, const char **pStr);
-typedef CUresult CUDAAPI (*CUMEMSETD8_V2)(CUdeviceptr dstDevice, unsigned char uc, size_t N);
-typedef CUresult CUDAAPI (*CUMEMCPY2D_V2)(const CUDA_MEMCPY2D *pCopy);
-
-typedef CUresult CUDAAPI (*CUGRAPHICSGLREGISTERIMAGE)(CUgraphicsResource *pCudaResource, GLuint image, GLenum target, unsigned int Flags);
-typedef CUresult CUDAAPI (*CUGRAPHICSRESOURCESETMAPFLAGS)(CUgraphicsResource resource, unsigned int flags);
-typedef CUresult CUDAAPI (*CUGRAPHICSMAPRESOURCES)(unsigned int count, CUgraphicsResource *resources, CUstream hStream);
-typedef CUresult CUDAAPI (*CUGRAPHICSUNREGISTERRESOURCE)(CUgraphicsResource resource);
-typedef CUresult CUDAAPI (*CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY)(CUarray *pArray, CUgraphicsResource resource, unsigned int arrayIndex, unsigned int mipLevel);
+typedef struct CUgraphicsResource_st *CUgraphicsResource;
struct Cuda {
- CUINIT cuInit;
- CUDEVICEGETCOUNT cuDeviceGetCount;
- CUDEVICEGET cuDeviceGet;
- CUCTXCREATE_V2 cuCtxCreate_v2;
- CUCTXPUSHCURRENT_V2 cuCtxPushCurrent_v2;
- CUCTXPOPCURRENT_V2 cuCtxPopCurrent_v2;
- CUGETERRORSTRING cuGetErrorString;
- CUMEMSETD8_V2 cuMemsetD8_v2;
- CUMEMCPY2D_V2 cuMemcpy2D_v2;
-
- CUGRAPHICSGLREGISTERIMAGE cuGraphicsGLRegisterImage;
- CUGRAPHICSRESOURCESETMAPFLAGS cuGraphicsResourceSetMapFlags;
- CUGRAPHICSMAPRESOURCES cuGraphicsMapResources;
- CUGRAPHICSUNREGISTERRESOURCE cuGraphicsUnregisterResource;
- CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY cuGraphicsSubResourceGetMappedArray;
+ CUresult CUDAAPI (*cuInit)(unsigned int Flags);
+ CUresult CUDAAPI (*cuDeviceGetCount)(int *count);
+ CUresult CUDAAPI (*cuDeviceGet)(CUdevice *device, int ordinal);
+ CUresult CUDAAPI (*cuCtxCreate_v2)(CUcontext *pctx, unsigned int flags, CUdevice dev);
+ CUresult CUDAAPI (*cuCtxPushCurrent_v2)(CUcontext ctx);
+ CUresult CUDAAPI (*cuCtxPopCurrent_v2)(CUcontext *pctx);
+ CUresult CUDAAPI (*cuGetErrorString)(CUresult error, const char **pStr);
+ CUresult CUDAAPI (*cuMemsetD8_v2)(CUdeviceptr dstDevice, unsigned char uc, size_t N);
+ CUresult CUDAAPI (*cuMemcpy2D_v2)(const CUDA_MEMCPY2D *pCopy);
+
+ CUresult CUDAAPI (*cuGraphicsGLRegisterImage)(CUgraphicsResource *pCudaResource, unsigned int image, unsigned int target, unsigned int Flags);
+ CUresult CUDAAPI (*cuGraphicsResourceSetMapFlags)(CUgraphicsResource resource, unsigned int flags);
+ CUresult CUDAAPI (*cuGraphicsMapResources)(unsigned int count, CUgraphicsResource *resources, CUstream hStream);
+ CUresult CUDAAPI (*cuGraphicsUnregisterResource)(CUgraphicsResource resource);
+ CUresult CUDAAPI (*cuGraphicsSubResourceGetMappedArray)(CUarray *pArray, CUgraphicsResource resource, unsigned int arrayIndex, unsigned int mipLevel);
~Cuda() {
if(library)
@@ -52,80 +39,39 @@ struct Cuda {
if(!lib) {
lib = dlopen("libcuda.so.1", RTLD_LAZY);
if(!lib) {
- fprintf(stderr, "Error: failed to load libcuda.so and libcuda.so.1, error: %s\n", dlerror());
+ fprintf(stderr, "Error: failed to load libcuda.so/libcuda.so.1, error: %s\n", dlerror());
return false;
}
}
- cuInit = (CUINIT)load_symbol(lib, "cuInit");
- if(!cuInit)
- goto fail;
-
- cuDeviceGetCount = (CUDEVICEGETCOUNT)load_symbol(lib, "cuDeviceGetCount");
- if(!cuDeviceGetCount)
- goto fail;
-
- cuDeviceGet = (CUDEVICEGET)load_symbol(lib, "cuDeviceGet");
- if(!cuDeviceGet)
- goto fail;
-
- cuCtxCreate_v2 = (CUCTXCREATE_V2)load_symbol(lib, "cuCtxCreate_v2");
- if(!cuCtxCreate_v2)
- goto fail;
-
- cuCtxPushCurrent_v2 = (CUCTXPUSHCURRENT_V2)load_symbol(lib, "cuCtxPushCurrent_v2");
- if(!cuCtxPushCurrent_v2)
- goto fail;
-
- cuCtxPopCurrent_v2 = (CUCTXPOPCURRENT_V2)load_symbol(lib, "cuCtxPopCurrent_v2");
- if(!cuCtxPopCurrent_v2)
- goto fail;
-
- cuGetErrorString = (CUGETERRORSTRING)load_symbol(lib, "cuGetErrorString");
- if(!cuGetErrorString)
- goto fail;
-
- cuMemsetD8_v2 = (CUMEMSETD8_V2)load_symbol(lib, "cuMemsetD8_v2");
- if(!cuMemsetD8_v2)
- goto fail;
-
- cuMemcpy2D_v2 = (CUMEMCPY2D_V2)load_symbol(lib, "cuMemcpy2D_v2");
- if(!cuMemcpy2D_v2)
- goto fail;
-
- cuGraphicsGLRegisterImage = (CUGRAPHICSGLREGISTERIMAGE)load_symbol(lib, "cuGraphicsGLRegisterImage");
- if(!cuGraphicsGLRegisterImage)
- goto fail;
-
- cuGraphicsResourceSetMapFlags = (CUGRAPHICSRESOURCESETMAPFLAGS)load_symbol(lib, "cuGraphicsResourceSetMapFlags");
- if(!cuGraphicsResourceSetMapFlags)
- goto fail;
-
- cuGraphicsMapResources = (CUGRAPHICSMAPRESOURCES)load_symbol(lib, "cuGraphicsMapResources");
- if(!cuGraphicsMapResources)
- goto fail;
-
- cuGraphicsUnregisterResource = (CUGRAPHICSUNREGISTERRESOURCE)load_symbol(lib, "cuGraphicsUnregisterResource");
- if(!cuGraphicsUnregisterResource)
- goto fail;
-
- cuGraphicsSubResourceGetMappedArray = (CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY)load_symbol(lib, "cuGraphicsSubResourceGetMappedArray");
- if(!cuGraphicsSubResourceGetMappedArray)
- goto fail;
-
- library = lib;
- return true;
-
- fail:
- dlclose(lib);
- return false;
- }
-private:
- void* load_symbol(void *library, const char *symbol) {
- void *sym = dlsym(library, symbol);
- if(!sym)
- fprintf(stderr, "Error: missing required symbol %s from libcuda.so\n", symbol);
- return sym;
+ dlsym_assign required_dlsym[] = {
+ { (void**)&cuInit, "cuInit" },
+ { (void**)&cuDeviceGetCount, "cuDeviceGetCount" },
+ { (void**)&cuDeviceGet, "cuDeviceGet" },
+ { (void**)&cuCtxCreate_v2, "cuCtxCreate_v2" },
+ { (void**)&cuCtxPushCurrent_v2, "cuCtxPushCurrent_v2" },
+ { (void**)&cuCtxPopCurrent_v2, "cuCtxPopCurrent_v2" },
+ { (void**)&cuGetErrorString, "cuGetErrorString" },
+ { (void**)&cuMemsetD8_v2, "cuMemsetD8_v2" },
+ { (void**)&cuMemcpy2D_v2, "cuMemcpy2D_v2" },
+
+ { (void**)&cuGraphicsGLRegisterImage, "cuGraphicsGLRegisterImage" },
+ { (void**)&cuGraphicsResourceSetMapFlags, "cuGraphicsResourceSetMapFlags" },
+ { (void**)&cuGraphicsMapResources, "cuGraphicsMapResources" },
+ { (void**)&cuGraphicsUnregisterResource, "cuGraphicsUnregisterResource" },
+ { (void**)&cuGraphicsSubResourceGetMappedArray, "cuGraphicsSubResourceGetMappedArray" },
+
+ { NULL, NULL }
+ };
+
+ if(dlsym_load_list(lib, required_dlsym)) {
+ library = lib;
+ return true;
+ } else {
+ fprintf(stderr, "Error: missing required symbols in libcuda.so\n");
+ dlclose(lib);
+ return false;
+ }
}
private:
void *library = nullptr;
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-04 23:57:13 +0000