diff --git a/libavutil/hwcontext_vulkan.c b/libavutil/hwcontext_vulkan.c index 1c298a0b18..6402030afe 100644 --- a/libavutil/hwcontext_vulkan.c +++ b/libavutil/hwcontext_vulkan.c @@ -143,6 +143,9 @@ typedef struct VulkanDevicePriv { /* Disable multiplane images */ int disable_multiplane; + /* Maximum queues */ + int limit_queues; + /* Nvidia */ int dev_is_nvidia; } VulkanDevicePriv; @@ -1509,9 +1512,14 @@ static int setup_queue_families(AVHWDeviceContext *ctx, VkDeviceCreateInfo *cd) if (i == hwctx->nb_qf) { \ hwctx->qf[i].idx = idx; \ hwctx->qf[i].num = qf[idx].queueFamilyProperties.queueCount; \ - if (dprops.driverID == VK_DRIVER_ID_NVIDIA_PROPRIETARY) { \ + if (p->limit_queues || \ + dprops.driverID == VK_DRIVER_ID_NVIDIA_PROPRIETARY) { \ + int max = p->limit_queues; \ if (type == VK_QUEUE_GRAPHICS_BIT) \ - hwctx->qf[i].num = FFMIN(hwctx->qf[i].num, 1); \ + hwctx->qf[i].num = FFMIN(hwctx->qf[i].num, \ + max ? max : 1); \ + else if (max) \ + hwctx->qf[i].num = FFMIN(hwctx->qf[i].num, max); \ } \ hwctx->qf[i].flags = type; \ hwctx->qf[i].video_caps = vid_op; \ @@ -1718,6 +1726,11 @@ static int vulkan_device_create_internal(AVHWDeviceContext *ctx, if (opt_d) p->use_linear_images = strtol(opt_d->value, NULL, 10); + /* Limit queues to a given number if needed */ + opt_d = av_dict_get(opts, "limit_queues", NULL, 0); + if (opt_d) + p->limit_queues = strtol(opt_d->value, NULL, 10); + /* The disable_multiplane argument takes precedent over the option */ p->disable_multiplane = disable_multiplane; if (!p->disable_multiplane) {