nvenc: Don't leak CUDA device memory

Allocated device memory should be freed with cuMemFree

nvenc: Don't leak CUDA device memory
Allocated device memory should be freed with cuMemFree
5898acce · Seungha Yang · Tim-Philipp Müller · 1410f521 · 5898acce
Commit 5898acce authored Mar 26, 2019 by Seungha Yang 🐑 Committed by Tim-Philipp Müller May 02, 2019
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

sys/nvenc/gstnvbaseenc.c sys/nvenc/gstnvbaseenc.c +6 -0

No files found.
--- a/sys/nvenc/gstnvbaseenc.c
+++ b/sys/nvenc/gstnvbaseenc.c
@@ -1012,6 +1012,12 @@ gst_nv_base_enc_free_buffers (GstNvBaseEnc * nvenc)
        GST_ERROR_OBJECT (nvenc, "Failed to unregister resource %p, ret %d",
            in_gl_resource, nv_ret);

+      nv_ret = cuMemFree ((CUdeviceptr) in_gl_resource->cuda_pointer);
+      if (nv_ret != NV_ENC_SUCCESS) {
+        GST_ERROR_OBJECT (nvenc, "Failed to free CUDA device memory, ret %d",
+            nv_ret);
+      }
+
      g_free (in_gl_resource);
      cuCtxPopCurrent (NULL);
    } else