avfilter/cuda: fix ptx inflation with large payloads

This commit is contained in:
Timo Rothenpieler 2021-06-24 19:22:24 +02:00
parent aca923b365
commit 91a41a3439
1 changed files with 2 additions and 2 deletions

View File

@ -62,8 +62,8 @@ int ff_cuda_load_module(void *avctx, AVCUDADeviceContext *hwctx, CUmodule *cu_mo
stream.next_out = buf + stream.total_out;
ret = inflate(&stream, Z_FINISH);
if (ret != Z_OK && ret != Z_STREAM_END) {
av_log(avctx, AV_LOG_ERROR, "zlib inflate error: %s\n", stream.msg);
if (ret != Z_OK && ret != Z_STREAM_END && ret != Z_BUF_ERROR) {
av_log(avctx, AV_LOG_ERROR, "zlib inflate error(%d): %s\n", ret, stream.msg);
inflateEnd(&stream);
av_free(buf);
return AVERROR(EINVAL);