FaSTTUBe/GST-Tensordecoder-ov_ep
35
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

cuda: Protect gst_cuda_ensure_element_context() with lock

Browse Source 
Serialize the order of context setup since it can be called
from multiple threads

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/2666>
This commit is contained in:
Seungha Yang 2022-06-27 20:50:15 +09:00
parent ce82c0570e
commit bcacfc7ced
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
subprojects/gst-plugins-bad/gst-libs/gst/cuda/gstcudautils.c
View File

@ -197,18 +197,27 @@ gst_cuda_ensure_element_context (GstElement * element, gint device_id,
GstCudaContext ** cuda_ctx) GstCudaContext ** cuda_ctx)
{ {
guint target_device_id = 0; guint target_device_id = 0;
gboolean ret = TRUE;
static GRecMutex lock;
static gsize init_lock_once = 0;
g_return_val_if_fail (element != NULL, FALSE); g_return_val_if_fail (element != NULL, FALSE);
g_return_val_if_fail (cuda_ctx != NULL, FALSE); g_return_val_if_fail (cuda_ctx != NULL, FALSE);
_init_debug (); _init_debug ();
if (g_once_init_enter (&init_lock_once)) {
g_rec_mutex_init (&lock);
g_once_init_leave (&init_lock_once, 1);
}
g_rec_mutex_lock (&lock);
if (*cuda_ctx) if (*cuda_ctx)
return TRUE; goto out;
find_cuda_context (element, cuda_ctx); find_cuda_context (element, cuda_ctx);
if (*cuda_ctx) if (*cuda_ctx)
return TRUE; goto out;
if (device_id > 0) if (device_id > 0)
target_device_id = device_id; target_device_id = device_id;
@ -219,7 +228,7 @@ gst_cuda_ensure_element_context (GstElement * element, gint device_id,
if (*cuda_ctx == NULL) { if (*cuda_ctx == NULL) {
GST_CAT_ERROR_OBJECT (GST_CAT_CONTEXT, element, GST_CAT_ERROR_OBJECT (GST_CAT_CONTEXT, element,
"Failed to create CUDA context with device-id %d", device_id); "Failed to create CUDA context with device-id %d", device_id);
return FALSE; ret = FALSE;
} else { } else {
GstContext *context; GstContext *context;
GstMessage *msg; GstMessage *msg;
@ -238,7 +247,10 @@ gst_cuda_ensure_element_context (GstElement * element, gint device_id,
gst_element_post_message (GST_ELEMENT_CAST (element), msg); gst_element_post_message (GST_ELEMENT_CAST (element), msg);
} }
return TRUE; out:
g_rec_mutex_unlock (&lock);
return ret;
} }
/** /**