mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-13 12:51:16 +00:00
cuda: Fix runtime compiler loading with old CUDA tookit
Fallback to PTX if CUBIN symbol is unavailable Fixes: https://gitlab.freedesktop.org/gstreamer/gstreamer/-/issues/3685 Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7220>
This commit is contained in:
parent
3e4c8f1b16
commit
0b285fc1a1
1 changed files with 19 additions and 3 deletions
|
@ -45,10 +45,19 @@ GST_DEBUG_CATEGORY_STATIC (gst_cuda_nvrtc_debug);
|
|||
} \
|
||||
} G_STMT_END;
|
||||
|
||||
#define LOAD_OPTIONAL_SYMBOL(name,func) G_STMT_START { \
|
||||
if (!g_module_symbol (module, G_STRINGIFY (name), (gpointer *) &vtable->func)) { \
|
||||
GST_WARNING ("Failed to load '%s', %s", G_STRINGIFY (name), g_module_error()); \
|
||||
return TRUE; \
|
||||
} \
|
||||
} G_STMT_END;
|
||||
|
||||
|
||||
/* *INDENT-OFF* */
|
||||
typedef struct _GstCudaNvrtcVTable
|
||||
{
|
||||
gboolean loaded;
|
||||
gboolean have_cubin;
|
||||
|
||||
nvrtcResult (*NvrtcCompileProgram) (nvrtcProgram prog, int numOptions,
|
||||
const char **options);
|
||||
|
@ -154,6 +163,7 @@ gst_cuda_nvrtc_load_library_once (void)
|
|||
}
|
||||
|
||||
vtable = &gst_cuda_nvrtc_vtable;
|
||||
vtable->have_cubin = FALSE;
|
||||
|
||||
LOAD_SYMBOL (nvrtcCompileProgram, NvrtcCompileProgram);
|
||||
LOAD_SYMBOL (nvrtcCreateProgram, NvrtcCreateProgram);
|
||||
|
@ -162,11 +172,12 @@ gst_cuda_nvrtc_load_library_once (void)
|
|||
LOAD_SYMBOL (nvrtcGetPTXSize, NvrtcGetPTXSize);
|
||||
LOAD_SYMBOL (nvrtcGetProgramLog, NvrtcGetProgramLog);
|
||||
LOAD_SYMBOL (nvrtcGetProgramLogSize, NvrtcGetProgramLogSize);
|
||||
LOAD_SYMBOL (nvrtcGetCUBINSize, NvrtcGetCUBINSize);
|
||||
LOAD_SYMBOL (nvrtcGetCUBIN, NvrtcGetCUBIN);
|
||||
|
||||
vtable->loaded = TRUE;
|
||||
|
||||
LOAD_OPTIONAL_SYMBOL (nvrtcGetCUBINSize, NvrtcGetCUBINSize);
|
||||
LOAD_OPTIONAL_SYMBOL (nvrtcGetCUBIN, NvrtcGetCUBIN);
|
||||
vtable->have_cubin = TRUE;
|
||||
|
||||
return TRUE;
|
||||
|
||||
error:
|
||||
|
@ -387,6 +398,11 @@ gst_cuda_nvrtc_compile_cubin (const gchar * source, gint device)
|
|||
if (!gst_cuda_nvrtc_load_library ())
|
||||
return nullptr;
|
||||
|
||||
if (!gst_cuda_nvrtc_vtable.have_cubin) {
|
||||
GST_DEBUG ("CUBIN related symbols are unavailable");
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
GST_TRACE ("CUDA kernel source \n%s", source);
|
||||
|
||||
curet = CuDeviceGetAttribute (&major,
|
||||
|
|
Loading…
Reference in a new issue