avcodec/cuviddec: explicitly synchronize cuMemcpy calls

7 years ago · 93d1756af2
parent 9b82e333b7
commit 93d1756af2
1 changed files with 5 additions and 1 deletions
--- a/libavcodec/cuviddec.c
+++ b/libavcodec/cuviddec.c
@ -550,12 +550,16 @@ static int cuvid_output_frame(AVCodecContext *avctx, AVFrame *frame)
                    .Height        = avctx->height >> (i ? 1 : 0),
                };

-                ret = CHECK_CU(ctx->cudl->cuMemcpy2D(&cpy));
+                ret = CHECK_CU(ctx->cudl->cuMemcpy2DAsync(&cpy, device_hwctx->stream));
                if (ret < 0)
                    goto error;

                offset += avctx->height;
            }
+
+            ret = CHECK_CU(ctx->cudl->cuStreamSynchronize(device_hwctx->stream));
+            if (ret < 0)
+                goto error;
        } else if (avctx->pix_fmt == AV_PIX_FMT_NV12 ||
                   avctx->pix_fmt == AV_PIX_FMT_P010 ||
                   avctx->pix_fmt == AV_PIX_FMT_P016) {