From 14983a76cddb88da451f54fd1c4f5bf48eff4c67 Mon Sep 17 00:00:00 2001
From: baldurk <baldurk@baldurk.org>
Date: Mon, 22 Feb 2021 16:47:09 +0000
Subject: [PATCH] Fix missing handling of variable descriptor counts in usage

---
 renderdoc/driver/vulkan/vk_core.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/renderdoc/driver/vulkan/vk_core.cpp b/renderdoc/driver/vulkan/vk_core.cpp
index 269eaafd0..de3be29c0 100644
--- a/renderdoc/driver/vulkan/vk_core.cpp
+++ b/renderdoc/driver/vulkan/vk_core.cpp
@@ -4272,7 +4272,11 @@ void WrappedVulkan::AddUsage(VulkanDrawcallTreeNode &drawNode, rdcarray<DebugMes
           continue;
         }
 
-        if(layout.bindings[bind].descriptorCount > 1000)
+        uint32_t descriptorCount = layout.bindings[bind].descriptorCount;
+        if(layout.bindings[bind].variableSize)
+          descriptorCount = descset.data.variableDescriptorCount;
+
+        if(descriptorCount > 1000)
         {
           if(!hugeRangeWarned)
             RDCWARN("Skipping large, most likely 'bindless', descriptor range");
@@ -4280,8 +4284,11 @@ void WrappedVulkan::AddUsage(VulkanDrawcallTreeNode &drawNode, rdcarray<DebugMes
           continue;
         }
 
-        for(uint32_t a = 0; a < layout.bindings[bind].descriptorCount; a++)
+        for(uint32_t a = 0; a < descriptorCount; a++)
         {
+          if(!descset.data.binds[bind])
+            continue;
+
           DescriptorSetSlot &slot = descset.data.binds[bind][a];
 
           ResourceId id;