mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-29 11:40:38 +00:00
73cd1aa9dc
Since cuda-tools 9.0, nvcuvid.h is replaced by dynlink_nvcuvid.h. This patch changes nvdec to use run-time dynamic linking if cuda-tools version >= 9. nvenc does not require any change since its necessary headers are still available. https://bugzilla.gnome.org/show_bug.cgi?id=791724
122 lines
4.6 KiB
C
122 lines
4.6 KiB
C
/*
|
|
* Copyright 2018 Red Hat, Inc. and/or its affiliates.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer
|
|
* in the documentation and/or other materials provided with the
|
|
* distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <gst/gst.h>
|
|
#include <gst/gl/gstglfuncs.h>
|
|
#include <dynlink_cuda.h>
|
|
#include <gmodule.h>
|
|
|
|
/* Types that are missing from included headers */
|
|
typedef CUresult CUDAAPI tcuGetErrorName (CUresult error, const char **pStr);
|
|
typedef CUresult CUDAAPI tcuGetErrorString (CUresult error, const char **pStr);
|
|
typedef CUresult CUDAAPI tcuGraphicsGLRegisterImage (CUgraphicsResource *
|
|
pCudaResource, GLuint image, GLenum target, unsigned int Flags);
|
|
|
|
tcuInit *_cuInit;
|
|
tcuDriverGetVersion *cuDriverGetVersion;
|
|
tcuCtxCreate *cuCtxCreate;
|
|
tcuCtxDestroy *cuCtxDestroy;
|
|
tcuCtxPopCurrent *cuCtxPopCurrent;
|
|
tcuMemcpy2D *cuMemcpy2D;
|
|
tcuGraphicsUnregisterResource *cuGraphicsUnregisterResource;
|
|
tcuGraphicsSubResourceGetMappedArray *cuGraphicsSubResourceGetMappedArray;
|
|
tcuGraphicsMapResources *cuGraphicsMapResources;
|
|
tcuGraphicsUnmapResources *cuGraphicsUnmapResources;
|
|
|
|
tcuGetErrorName *cuGetErrorName;
|
|
tcuGetErrorString *cuGetErrorString;
|
|
tcuGraphicsGLRegisterImage *cuGraphicsGLRegisterImage;
|
|
|
|
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
|
static char CudaLibName[] = "nvcuda.dll";
|
|
#elif defined(__unix__)
|
|
static char CudaLibName[] = "libcuda.so";
|
|
#endif
|
|
|
|
#define STRINGIFY(X) #X
|
|
#define GET_PROC(name) \
|
|
if (!g_module_symbol (module, #name, (gpointer *)&name)) { \
|
|
GST_ERROR("%s", g_module_error()); \
|
|
g_module_close(module); \
|
|
return CUDA_ERROR_UNKNOWN; \
|
|
}
|
|
|
|
#define GET_PROC_V2(name) \
|
|
if(!g_module_symbol (module, STRINGIFY(name##_v2), (gpointer *)&name)) { \
|
|
GST_ERROR("%s", g_module_error()); \
|
|
g_module_close(module); \
|
|
return CUDA_ERROR_UNKNOWN; \
|
|
}
|
|
|
|
CUresult CUDAAPI
|
|
cuInit (unsigned int Flags, int cudaVersion, void *pHandleDriver)
|
|
{
|
|
GModule *module;
|
|
int driverVer = 1000;
|
|
|
|
module = g_module_open (CudaLibName, G_MODULE_BIND_LAZY);
|
|
if (module == NULL) {
|
|
GST_ERROR ("%s", g_module_error ());
|
|
return CUDA_ERROR_UNKNOWN;
|
|
}
|
|
//Init
|
|
g_module_symbol (module, "cuInit", (gpointer *) & _cuInit);
|
|
if (_cuInit == NULL || _cuInit (Flags) != CUDA_SUCCESS) {
|
|
GST_ERROR ("Failed to init cuda\n");
|
|
return CUDA_ERROR_UNKNOWN;
|
|
}
|
|
|
|
GET_PROC (cuDriverGetVersion);
|
|
if (cuDriverGetVersion == NULL
|
|
|| cuDriverGetVersion (&driverVer) != CUDA_SUCCESS) {
|
|
GST_ERROR ("Failed to get cuda version\n");
|
|
return CUDA_ERROR_UNKNOWN;
|
|
}
|
|
|
|
if (cudaVersion < 4000 || __CUDA_API_VERSION < 4000) {
|
|
GST_ERROR ("cuda version or cuda api version is too old\n");
|
|
return CUDA_ERROR_UNKNOWN;
|
|
}
|
|
|
|
GET_PROC (cuCtxDestroy);
|
|
GET_PROC (cuCtxPopCurrent);
|
|
GET_PROC (cuGetErrorName);
|
|
GET_PROC (cuGetErrorString);
|
|
|
|
GET_PROC_V2 (cuCtxDestroy);
|
|
GET_PROC_V2 (cuCtxPopCurrent);
|
|
GET_PROC_V2 (cuCtxCreate);
|
|
GET_PROC_V2 (cuMemcpy2D);
|
|
GET_PROC (cuGraphicsGLRegisterImage);
|
|
GET_PROC (cuGraphicsGLRegisterImage);
|
|
GET_PROC (cuGraphicsUnregisterResource);
|
|
GET_PROC (cuGraphicsSubResourceGetMappedArray);
|
|
GET_PROC (cuGraphicsMapResources);
|
|
GET_PROC (cuGraphicsUnmapResources);
|
|
|
|
return CUDA_SUCCESS;
|
|
}
|