mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2025-01-25 16:48:11 +00:00
Code Issues 1 Projects Releases Wiki Activity

nvcodec: Enhance CUDA runtime compiler library loading on Windows

Browse source 
The name of installed CUDA runtime compiler library is formed like
nvrtc64_{major-version}{minor-version}_0.dll on Windows
(which is differnt from documented in https://docs.nvidia.com/cuda/nvrtc/index.html)

And minor version might not be exactly same as that of CUDA.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/2362>
This commit is contained in:
Seungha Yang 2021-06-27 01:15:49 +09:00 committed by GStreamer Marge Bot
parent 44e3399bf8
commit c8ffd39dfe
1 changed files with 29 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
sys/nvcodec/gstnvrtcloader.c
View file

@ -26,6 +26,9 @@
#include <gmodule.h> #include <gmodule.h>
GST_DEBUG_CATEGORY_EXTERN (gst_nvcodec_debug);
#define GST_CAT_DEFAULT gst_nvcodec_debug
#ifndef G_OS_WIN32 #ifndef G_OS_WIN32
#define NVRTC_LIBNAME "libnvrtc.so" #define NVRTC_LIBNAME "libnvrtc.so"
#else #else
@ -80,14 +83,33 @@ gst_nvrtc_load_library (void)
if (!module) { if (!module) {
#ifndef G_OS_WIN32 #ifndef G_OS_WIN32
filename = g_strdup (NVRTC_LIBNAME); filename = g_strdup (NVRTC_LIBNAME);
#else
/* (major version * 1000) + (minor version * 10) */
filename = g_strdup_printf (NVRTC_LIBNAME, cuda_version / 1000,
(cuda_version % 1000) / 10);
#endif
module = g_module_open (filename, G_MODULE_BIND_LAZY);
fname = filename; fname = filename;
module = g_module_open (filename, G_MODULE_BIND_LAZY);
#else
/* XXX: On Windows, minor version of nvrtc library might not be exactly
* same as CUDA library */
{
gint cuda_major_version = cuda_version / 1000;
gint cuda_minor_version = (cuda_version % 1000) / 10;
gint minor_version;
for (minor_version = cuda_minor_version; minor_version >= 0;
minor_version--) {
g_free (filename);
filename = g_strdup_printf (NVRTC_LIBNAME, cuda_major_version,
minor_version);
fname = filename;
module = g_module_open (filename, G_MODULE_BIND_LAZY);
if (module) {
GST_INFO ("%s is available", filename);
break;
}
GST_DEBUG ("Couldn't open library %s", filename);
}
}
#endif
} }
if (module == NULL) { if (module == NULL) {