igl | vulkan | More stages handled in transitionLayout()

Summary: More pipeline stages are handled in `transitionLayout()`. Reviewed By: pixelperfect3 Differential Revision: D49659344 fbshipit-source-id: 83e0ea37d26226f819c7ef82b09c079d72fc5680
facebook · Sep 27, 2023 · 9abe2e5 · 9abe2e5
1 parent 544985c
commit 9abe2e5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/igl/vulkan/VulkanImage.cpp b/src/igl/vulkan/VulkanImage.cpp
@@ -663,6 +663,7 @@ void VulkanImage::transitionLayout(VkCommandBuffer cmdBuf,
   case VK_PIPELINE_STAGE_TRANSFER_BIT:
   case VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT:
   case VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT:
+  case VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT | VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT:
     break;
   default:
     IGL_ASSERT_MSG(
@@ -679,9 +680,11 @@ void VulkanImage::transitionLayout(VkCommandBuffer cmdBuf,
     srcAccessMask |= VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT;
   }
   if (srcStageMask & VK_PIPELINE_STAGE_TRANSFER_BIT) {
+    srcAccessMask |= VK_ACCESS_TRANSFER_READ_BIT;
     srcAccessMask |= VK_ACCESS_TRANSFER_WRITE_BIT;
   }
   if (srcStageMask & VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT) {
+    srcAccessMask |= VK_ACCESS_SHADER_READ_BIT;
     srcAccessMask |= VK_ACCESS_SHADER_WRITE_BIT;
   }
   if (srcStageMask & VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT) {