Commit a52976c0 authored by Timo Rothenpieler's avatar Timo Rothenpieler Committed by Luca Barbato

nvenc: make gpu indices independent of supported capabilities

Do not allocate a CUDA context for every available gpu.
Signed-off-by: 's avatarLuca Barbato <lu_zero@gentoo.org>
parent 77c23704
......@@ -359,6 +359,9 @@ static int nvenc_check_device(AVCodecContext *avctx, int idx)
if (((major << 4) | minor) < NVENC_CAP)
goto fail;
if (ctx->device != idx && ctx->device != ANY_DEVICE)
return -1;
ret = nvel->cu_ctx_create(&ctx->cu_context_internal, 0, cu_device);
if (ret != CUDA_SUCCESS)
goto fail;
......@@ -377,7 +380,7 @@ static int nvenc_check_device(AVCodecContext *avctx, int idx)
av_log(avctx, loglevel, "supports NVENC\n");
if (ctx->device == cu_device || ctx->device == ANY_DEVICE)
if (ctx->device == idx || ctx->device == ANY_DEVICE)
return 0;
fail3:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment