[FFmpeg-cvslog] avcodec/cuviddec: dynamically set max decode surfaces

[Random Joe]

ffmpeg | branch: master | Random Joe <ff.random.joe at gmail.com> | Sun Aug 13 19:13:01 2023 +0200| [176ed008bd0f8c6ec7f0474b6503680e7ea6c451] | committer: Timo Rothenpieler

avcodec/cuviddec: dynamically set max decode surfaces

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=176ed008bd0f8c6ec7f0474b6503680e7ea6c451
---

 libavcodec/cuviddec.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/libavcodec/cuviddec.c b/libavcodec/cuviddec.c
index 814bc53f70..da37207cf2 100644
--- a/libavcodec/cuviddec.c
+++ b/libavcodec/cuviddec.c
@@ -360,6 +360,11 @@ static int CUDAAPI cuvid_handle_video_sequence(void *opaque, CUVIDEOFORMAT* form
         }
     }
 
+    if(ctx->cuparseinfo.ulMaxNumDecodeSurfaces != cuinfo.ulNumDecodeSurfaces) {
+        ctx->cuparseinfo.ulMaxNumDecodeSurfaces = cuinfo.ulNumDecodeSurfaces;
+        return cuinfo.ulNumDecodeSurfaces;
+    }
+
     return 1;
 }
 
@@ -1022,7 +1027,7 @@ static av_cold int cuvid_decode_init(AVCodecContext *avctx)
         goto error;
     }
 
-    ctx->cuparseinfo.ulMaxNumDecodeSurfaces = ctx->nb_surfaces;
+    ctx->cuparseinfo.ulMaxNumDecodeSurfaces = 1;
     ctx->cuparseinfo.ulMaxDisplayDelay = (avctx->flags & AV_CODEC_FLAG_LOW_DELAY) ? 0 : CUVID_MAX_DISPLAY_DELAY;
     ctx->cuparseinfo.pUserData = avctx;
     ctx->cuparseinfo.pfnSequenceCallback = cuvid_handle_video_sequence;