From 91a41a34398afd5678c20c5f5025562a41cf5bd4 Mon Sep 17 00:00:00 2001
From: Timo Rothenpieler <timo@rothenpieler.org>
Date: Thu, 24 Jun 2021 19:22:24 +0200
Subject: [PATCH] avfilter/cuda: fix ptx inflation with large payloads

---
 libavfilter/cuda/load_helper.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/libavfilter/cuda/load_helper.c b/libavfilter/cuda/load_helper.c
index 62d644c29a..280e596ecb 100644
--- a/libavfilter/cuda/load_helper.c
+++ b/libavfilter/cuda/load_helper.c
@@ -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);