X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=ffmpeg_cuvid.c;h=baf6eee9f8c46821900783fd686bf3fb8bc70508;hb=ba632efa93eff2550bcc465190efd05a46116ce6;hp=766878f58bc3c75a32fd8498e4915e0d7dc21009;hpb=1bc6cdf2fcbd3caea7b78d0a617c6e338606d756;p=ffmpeg

diff --git a/ffmpeg_cuvid.c b/ffmpeg_cuvid.c
index 766878f58bc..baf6eee9f8c 100644
--- a/ffmpeg_cuvid.c
+++ b/ffmpeg_cuvid.c
@@ -135,7 +135,7 @@ int cuvid_transcode_init(OutputStream *ost)
          */
         hwframe_ctx = (AVHWFramesContext*)ctx->hw_frames_ctx->data;
         hwframe_ctx->format = AV_PIX_FMT_CUDA;
-        hwframe_ctx->sw_format = AV_PIX_FMT_NV12;
+        hwframe_ctx->sw_format = ist->st->codecpar->format == AV_PIX_FMT_YUV420P10 ? AV_PIX_FMT_P010 : AV_PIX_FMT_NV12;
     }
 
     return 0;