Added missing image layout deduction for mutable desc. set's setImmed…

…iate()
Dolkar · Jun 12, 2024 · f2402fa · f2402fa
1 parent 8985aac
commit f2402fa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/tephra/utils/mutable_descriptor_set.cpp b/src/tephra/utils/mutable_descriptor_set.cpp
@@ -166,9 +166,15 @@ namespace utils {
         ScratchVector<VkBufferView> vkBufferViews;
 
         if (descriptors[0].vkResolveDescriptorImageInfo() != nullptr) {
+            // Deduce image layout
+            VkImageLayout imageLayout = vkGetImageLayoutForDescriptor(
+                descriptorBinding->descriptorType,
+                descriptorBinding->flags.contains(DescriptorBindingFlag::AliasStorageImage));
+
             vkImageInfos.reserve(descriptors.size());
             for (const tp::Descriptor& descriptor : descriptors) {
                 vkImageInfos.push_back(*descriptor.vkResolveDescriptorImageInfo());
+                vkImageInfos.back().imageLayout = imageLayout;
             }
             descriptorWrite.pImageInfo = vkImageInfos.data();
         } else if (descriptors[0].vkResolveDescriptorBufferInfo() != nullptr) {