Fixed image layout transition for non-sampled images

corporateshark · Dec 2, 2024 · 4a7946d · 4a7946d
1 parent c1b784f
commit 4a7946d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lvk/vulkan/VulkanClasses.cpp b/lvk/vulkan/VulkanClasses.cpp
@@ -1870,7 +1870,7 @@ void lvk::CommandBuffer::transitionToShaderReadOnly(TextureHandle handle) const
     }
     // set the result of the previous render pass
     img.transitionLayout(wrapper_->cmdBuf_,
-                         VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL,
+                         img.isSampledImage() ? VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL : VK_IMAGE_LAYOUT_GENERAL,
                          srcStage,
                          VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT | VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT, // wait for subsequent
                                                                                                        // fragment/compute shaders