From 30b28f9a83ec095f5684fecb04a42619a6675989 Mon Sep 17 00:00:00 2001 From: Timo Rothenpieler Date: Mon, 6 Apr 2020 19:33:01 +0200 Subject: [PATCH] avutil/hwcontext_cuda: Only handle CUDA hardware frames --- libavutil/hwcontext_cuda.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/libavutil/hwcontext_cuda.c b/libavutil/hwcontext_cuda.c index 95402a1460..58d128a280 100644 --- a/libavutil/hwcontext_cuda.c +++ b/libavutil/hwcontext_cuda.c @@ -212,6 +212,10 @@ static int cuda_transfer_data(AVHWFramesContext *ctx, AVFrame *dst, CUcontext dummy; int i, ret; + if ((src->hw_frames_ctx && ((AVHWFramesContext*)src->hw_frames_ctx->data)->format != AV_PIX_FMT_CUDA) || + (dst->hw_frames_ctx && ((AVHWFramesContext*)dst->hw_frames_ctx->data)->format != AV_PIX_FMT_CUDA)) + return AVERROR(ENOSYS); + ret = CHECK_CU(cu->cuCtxPushCurrent(hwctx->cuda_ctx)); if (ret < 0) return ret;