From cb782763f03b146df43344992a82007ad616c8b9 Mon Sep 17 00:00:00 2001
From: Faker0715 <949506563@qq.com>
Date: Thu, 5 Sep 2024 11:08:08 +0800
Subject: More cleanup hwdevice ctx

---
 src/encoder/video/cuda.c  | 1 +
 src/encoder/video/vaapi.c | 1 +
 2 files changed, 2 insertions(+)

(limited to 'src')

diff --git a/src/encoder/video/cuda.c b/src/encoder/video/cuda.c
index 9681418..a323940 100644
--- a/src/encoder/video/cuda.c
+++ b/src/encoder/video/cuda.c
@@ -62,6 +62,7 @@ static bool gsr_video_encoder_cuda_setup_context(gsr_video_encoder_cuda *self, A
 
     self->cuda_stream = cuda_device_context->stream;
     video_codec_context->hw_frames_ctx = av_buffer_ref(frame_context);
+    av_buffer_unref(&frame_context);
     return true;
 }
 
diff --git a/src/encoder/video/vaapi.c b/src/encoder/video/vaapi.c
index e96319f..7a2abfc 100644
--- a/src/encoder/video/vaapi.c
+++ b/src/encoder/video/vaapi.c
@@ -63,6 +63,7 @@ static bool gsr_video_encoder_vaapi_setup_context(gsr_video_encoder_vaapi *self,
     }
 
     video_codec_context->hw_frames_ctx = av_buffer_ref(frame_context);
+    av_buffer_unref(&frame_context);
     return true;
 }
 
-- 
cgit v1.2.3