Commit aac7d6b2 authored by Anton Khirnov's avatar Anton Khirnov

nvenc: flush the encoder before closing it, as required by the docs

Otherwise, closing the encoder can crash.
parent ee359c72
......@@ -825,6 +825,14 @@ av_cold int ff_nvenc_encode_close(AVCodecContext *avctx)
NV_ENCODE_API_FUNCTION_LIST *nv = &ctx->nvel.nvenc_funcs;
int i;
/* the encoder has to be flushed before it can be closed */
if (ctx->nvenc_ctx) {
NV_ENC_PIC_PARAMS params = { .version = NV_ENC_PIC_PARAMS_VER,
.encodePicFlags = NV_ENC_PIC_FLAG_EOS };
nv->nvEncEncodePicture(ctx->nvenc_ctx, &params);
}
av_fifo_free(ctx->timestamps);
av_fifo_free(ctx->pending);
av_fifo_free(ctx->ready);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment