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

cudabasetransform: Update CUDA stream on context update

Browse Source 
CUDA stream must be associated with updated CUDA context

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/3618>
This commit is contained in:
Seungha Yang 2022-12-22 00:22:46 +09:00 committed by GStreamer Marge Bot
parent ad0ba3ab7d
commit 15b2cd6565
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
subprojects/gst-plugins-bad/sys/nvcodec/gstcudabasetransform.c
View File

@ -340,9 +340,20 @@ gst_cuda_base_transform_before_transform (GstBaseTransform * trans,
GST_INFO_OBJECT (self, "Updating device %" GST_PTR_FORMAT " -> %" GST_INFO_OBJECT (self, "Updating device %" GST_PTR_FORMAT " -> %"
GST_PTR_FORMAT, self->context, cmem->context); GST_PTR_FORMAT, self->context, cmem->context);
if (self->cuda_stream) {
gst_cuda_context_push (self->context);
CuStreamDestroy (self->cuda_stream);
gst_cuda_context_pop (NULL);
self->cuda_stream = NULL;
}
gst_object_unref (self->context); gst_object_unref (self->context);
self->context = gst_object_ref (cmem->context); self->context = gst_object_ref (cmem->context);
gst_cuda_context_push (self->context);
CuStreamCreate (&self->cuda_stream, CU_STREAM_DEFAULT);
gst_cuda_context_pop (NULL);
/* subclass will update internal object. /* subclass will update internal object.
* Note that gst_base_transform_reconfigure() might not trigger this * Note that gst_base_transform_reconfigure() might not trigger this
* unless caps was changed meanwhile */ * unless caps was changed meanwhile */