Compute Shader写入交换链图像时的图像布局验证错误

核心问题根源 #

Swapchain图像默认处于VK_IMAGE_LAYOUT_PRESENT_SRC_KHR布局，但Storage Image要求图像必须处于VK_IMAGE_LAYOUT_GENERAL布局才能被Compute Shader写入；而present操作又要求图像回到VK_IMAGE_LAYOUT_PRESENT_SRC_KHR布局。冲突的本质是图像布局没有在合适的时机完成转换。

具体解决步骤 #

• 步骤1：获取Swapchain图像后，先转至GENERAL布局
  在执行Compute Shader绘制前，插入图像内存屏障，将图像从展示布局转换为通用布局：

  VkImageMemoryBarrier barrier = {};
  barrier.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER;
  barrier.oldLayout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;
  barrier.newLayout = VK_IMAGE_LAYOUT_GENERAL;
  barrier.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
  barrier.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
  barrier.image = swapchainImage;
  barrier.subresourceRange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
  barrier.subresourceRange.baseMipLevel = 0;
  barrier.subresourceRange.levelCount = 1;
  barrier.subresourceRange.baseArrayLayer = 0;
  barrier.subresourceRange.layerCount = 1;
  barrier.srcAccessMask = 0;
  barrier.dstAccessMask = VK_ACCESS_SHADER_WRITE_BIT;
  
  vkCmdPipelineBarrier(
      cmdBuffer,
      VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT,
      VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
      0,
      0, nullptr,
      0, nullptr,
      1, &barrier
  );
  

  转换完成后，将DescriptorImageInfo的布局设为VK_IMAGE_LAYOUT_GENERAL即可通过验证层检查。

• 步骤2：Compute Shader写入完成后，转回PRESENT_SRC_KHR布局
  在Compute Shader dispatch执行完毕后，插入第二道屏障，将图像从通用布局转换回展示布局，确保present操作能正常执行：

  VkImageMemoryBarrier barrier = {};
  barrier.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER;
  barrier.oldLayout = VK_IMAGE_LAYOUT_GENERAL;
  barrier.newLayout = VK_IMAGE_LAYOUT_PRESENT_SRC_KHR;
  barrier.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
  barrier.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
  barrier.image = swapchainImage;
  barrier.subresourceRange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT;
  barrier.subresourceRange.baseMipLevel = 0;
  barrier.subresourceRange.levelCount = 1;
  barrier.subresourceRange.baseArrayLayer = 0;
  barrier.subresourceRange.layerCount = 1;
  barrier.srcAccessMask = VK_ACCESS_SHADER_WRITE_BIT;
  barrier.dstAccessMask = VK_ACCESS_MEMORY_READ_BIT;
  
  vkCmdPipelineBarrier(
      cmdBuffer,
      VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
      VK_PIPELINE_STAGE_BOTTOM_OF_PIPE_BIT,
      0,
      0, nullptr,
      0, nullptr,
      1, &barrier
  );
  

关键注意事项 #

• 创建Swapchain时，除了VK_IMAGE_USAGE_STORAGE_BIT，还需添加VK_IMAGE_USAGE_TRANSFER_DST_BIT和VK_IMAGE_USAGE_TRANSFER_SRC_BIT，用于支持布局转换操作。
• DescriptorImageInfo中指定的布局必须与图像当前的实际布局完全一致，否则会触发验证层错误。

内容的提问来源于stack exchange，提问作者RefuX