[FFmpeg-devel] [PATCH] avfilter/vf_scale_cuda: unload cuModule when uninit

[leozhang]

Signed-off-by: leozhang <nowerzt at gmail.com>
---
 libavfilter/vf_scale_cuda.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libavfilter/vf_scale_cuda.c b/libavfilter/vf_scale_cuda.c
index 1ffb73f831..46d69b00ee 100644
--- a/libavfilter/vf_scale_cuda.c
+++ b/libavfilter/vf_scale_cuda.c
@@ -119,7 +119,15 @@ static av_cold int cudascale_init(AVFilterContext *ctx)
 static av_cold void cudascale_uninit(AVFilterContext *ctx)
 {
     CUDAScaleContext *s = ctx->priv;
+    CUcontext dummy;
 
+    if (s->hwctx && s->cu_module) {
+        CudaFunctions *cu = s->hwctx->internal->cuda_dl;
+        CHECK_CU(cu->cuCtxPushCurrent(s->hwctx->cuda_ctx));
+        CHECK_CU(cu->cuModuleUnload(s->cu_module));
+        s->cu_module = NULL;
+        CHECK_CU(cu->cuCtxPopCurrent(&dummy));
+    }
     av_frame_free(&s->frame);
     av_buffer_unref(&s->frames_ctx);
     av_frame_free(&s->tmp_frame);
-- 
2.11.0