mirror of https://git.ffmpeg.org/ffmpeg.git
nvenc: make gpu indices independent of supported capabilities
Do not allocate a CUDA context for every available gpu. Signed-off-by: Luca Barbato <lu_zero@gentoo.org>
This commit is contained in:
parent
77c23704c7
commit
a52976c0fe
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue