igl | vulkan | Check storage images in ensureShaderModule()

Reviewed By: rudybear, rameshviswanathan Differential Revision: D69157243 fbshipit-source-id: 1c196a6931a98608cb37e10529ba991a77d920ac
facebook · Feb 5, 2025 · 5e9d9b5 · 5e9d9b5
1 parent 5a9ff24
commit 5e9d9b5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/igl/vulkan/Common.cpp b/src/igl/vulkan/Common.cpp
@@ -545,6 +545,15 @@ void ensureShaderModule(IShaderModule* sm) {
       continue;
     }
   }
+  for (const auto& i : info.images) {
+    if (!IGL_DEBUG_VERIFY(i.descriptorSet == kBindPoint_StorageImages)) {
+      IGL_LOG_ERROR(
+          "Missing descriptor set id for storage images: the shader should contain \"layout(set = "
+          "%u, ...)\"",
+          kBindPoint_StorageImages);
+      continue;
+    }
+  }
 }
 
 uint32_t getNumImagePlanes(VkFormat format) {