WEBKT

Vulkan延迟渲染管线中隐式MSAA的实现方案与避坑指南

25 0 0 0

在传统的延迟渲染（Deferred Shading）管线中，多重采样抗锯齿（MSAA）一直是性能杀手。由于 G-Buffer（几何缓冲区）包含了深度、法线、材质等多种信息，如果直接对 G-Buffer 进行物理上的多重采样（如 4x MSAA），会导致显存带宽和容量暴涨，这在移动端（TBDR 架构）或带宽敏感的设备上是不可接受的。

为了解决这一痛点，Vulkan 提供了两种实现“隐式”或“在片（On-Chip）”多重采样的方案：

  1. 基于 Subpass + Transient Attachment（瞬态附件）的在片 MSAA 方案（利用 TBDR 硬件特性）。
  2. 使用扩展 VK_EXT_multisampled_render_to_single_sampled（显式声明，隐式解析）。

下面将详细探讨这两种方案的具体实现机制、核心代码配置以及在延迟渲染中应用时的关键限制。

方案一：利用 Subpass + Transient Attachment 实现在片 MSAA

这是移动端（如 ARM Mali, PowerVR, Adreno）最经典的延迟渲染 MSAA 优化方案。其核心思想是：多重采样的 G-Buffer 只存在于 GPU 的 Tile Memory（片上高速缓存）中，绝不写回 VRAM（显存），在 Shading Subpass 中直接消费这些多重采样数据，并输出单采样的 Color Buffer 到 VRAM。

1. 创建瞬态（Transient）G-Buffer 图像

在创建 G-Buffer 的 VkImage 时，必须使用 VK_IMAGE_USAGE_TRANSIENT_ATTACHMENT_BIT，并且在分配内存时，寻找支持 VK_MEMORY_PROPERTY_LAZILY_ALLOCATED_BIT 的内存类型。

VkImageCreateInfo imageInfo{};
imageInfo.sType = VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO;
imageInfo.imageType = VK_IMAGE_TYPE_2D;
imageInfo.format = VK_FORMAT_R16G16B16A16_SFLOAT; // 以法线/粗糙度贴图为例
imageInfo.extent = { width, height, 1 };
imageInfo.mipLevels = 1;
imageInfo.arrayLayers = 1;
imageInfo.samples = VK_SAMPLE_COUNT_4_BIT; // 开启 4x MSAA
imageInfo.tiling = VK_IMAGE_TILING_OPTIMAL;
// 关键：声明为瞬态附件，且只能作为输入附件和颜色附件
imageInfo.usage = VK_IMAGE_USAGE_COLOR_ATTACHMENT_BIT | VK_IMAGE_USAGE_TRANSIENT_ATTACHMENT_BIT | VK_IMAGE_USAGE_INPUT_ATTACHMENT_BIT;

2. 配置 Render Pass 与 Subpasses

定义一个包含两个 Subpass 的 Render Pass：

  • Subpass 0 (Geometry Pass)：渲染到多重采样的瞬态 G-Buffer（Color Attachment）。
  • Subpass 1 (Shading Pass)：将多重采样的 G-Buffer 作为 Input Attachment 读入，进行光照计算，最后输出到单采样的 Framebuffer（Color Attachment）。
// Subpass 0 描述
VkSubpassDescription subpass0{};
subpass0.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS;
subpass0.colorAttachmentCount = gBufferAttachmentCount;
subpass0.pColorAttachments = m_msaaGBufferReferences; // 4x MSAA 瞬态 G-Buffer

// Subpass 1 描述
VkSubpassDescription subpass1{};
subpass1.pipelineBindPoint = VK_PIPELINE_BIND_POINT_GRAPHICS;
subpass1.colorAttachmentCount = 1;
subpass1.pColorAttachments = &m_singleSampledColorReference; // 1x 最终输出
subpass1.inputAttachmentCount = gBufferAttachmentCount;
subpass1.pInputAttachments = m_msaaGBufferInputReferences; // 读取 4x MSAA G-Buffer

3. 片元着色器中的多重采样处理

在 Subpass 1 的片元着色器中，必须使用 subpassInputMS 代替普通的 subpassInput，并根据需要对每个 sample 进行迭代或选择性采样。

#version 450

// 声明多重采样输入附件
layout(input_attachment_index = 0, set = 0, binding = 0) uniform subpassInputMS u_GBufferNormal;
layout(input_attachment_index = 1, set = 0, binding = 1) uniform subpassInputMS u_GBufferDepth;

layout(location = 0) out vec4 outColor;

void main() {
    vec4 accumulatedColor = vec4(0.0);
    int numSamples = 4; // 与管线设置的采样率一致

    for (int i = 0; i < numSamples; ++i) {
        vec3 normal = subpassLoad(u_GBufferNormal, i).rgb;
        float depth = subpassLoad(u_GBufferDepth, i).r;
        
        // 执行光照计算...
        vec4 sampleColor = calculateLighting(normal, depth);
        accumulatedColor += sampleColor;
    }

    // 在片上完成 Resolve（解析）并输出
    outColor = accumulatedColor / float(numSamples);
}

方案二：使用 VK_EXT_multisampled_render_to_single_sampled 扩展

该扩展允许应用在一个单采样（1x）的 Attachment 上进行多重采样渲染。GPU 内部会自动创建多重采样缓冲区，并在渲染结束写回 VRAM 时，**隐式地（implicitly）**将其解析（Resolve）为单采样图像。

1. 启用扩展与特性

在创建逻辑设备（Logical Device）时，需要启用 VK_EXT_multisampled_render_to_single_sampled 扩展。

2. 配置 Render Pass 中的隐式 Resolve

在创建 Render Pass 时，通过在 VkRenderPassCreateInfo 的 pNext 链中插入 VkRenderPassCreationControlEXT 或在 Subpass 中使用特定的多重采样渲染结构体：

VkMultisampledRenderToSingleSampledInfoEXT msInfo{};
msInfo.sType = VK_STRUCTURE_TYPE_MULTISAMPLED_RENDER_TO_SINGLE_SAMPLED_INFO_EXT;
msInfo.multisampledRenderToSingleSampledEnable = VK_TRUE;
msInfo.rasterizationSamples = VK_SAMPLE_COUNT_4_BIT; // 硬件内部进行 4x MSAA

// 将此结构体挂载到具体 Subpass 的 pNext 链上
VkSubpassDescription2 subpass{};
subpass.pNext = &msInfo;
// ... 配置其他属性

关键限制与避坑指南

这两种隐式 MSAA 方案虽然能极大节省显存带宽，但在延迟渲染管线中应用时，存在以下不可忽视的硬性限制：

1. 桌面级 IMR 架构与移动级 TBDR 架构的鸿沟

  • 在 TBDR（移动端）上：方案一（Transient Attachment）是完美的。由于 VK_MEMORY_PROPERTY_LAZILY_ALLOCATED_BIT 的存在，G-Buffer 真的不会占用物理显存，完全在 GPU 缓存中完成分配、写入、读取和销毁。
  • 在 IMR（桌面端如 NVIDIA/AMD）上：IMR 架构没有物理上的 Tile Memory 来保留整个瞬态图像。即使你声明了 LAZILY_ALLOCATED，驱动程序大机率仍会在 VRAM 中为其分配物理空间。这意味着在桌面端使用此方案，并不能减少显存带宽消耗，它仅仅减少了不必要的 vkCmdResolveImage 调用。

2. G-Buffer 深度与法线的非线性 Resolve 灾难

如果使用方案二（VK_EXT_multisampled_render_to_single_sampled）直接作用于 G-Buffer 附件，会带来严重的渲染错误。

  • 原因：该扩展是在 Subpass 结束时自动对 Attachment 进行均值 Resolve。
  • 后果：对于颜色值，均值 Resolve 是正确的；但对于深度（Depth）、法线（Normal）和材质 ID（Material ID），直接取平均值是灾难性的。例如：两个相邻三角形交界处的法线平均值可能会指向一个完全错误的方向，导致光照计算出现黑边或白边。
  • 对策：在延迟渲染中，千万不要直接对 G-Buffer Attachment 应用隐式 Resolve 扩展。必须使用方案一，在 Shading Subpass 中手动读取每个 sample（subpassInputMS），计算完光照后再对颜色进行 Resolve。

3. Subpass 布局的严格约束

使用方案一（基于 Subpass 交互）时，受限于 Vulkan Subpass 的硬件实现：

  • 不能进行跨像素操作：在 Subpass 1 中，你只能使用 subpassInput 读取当前片元（pixel）坐标对应的 G-Buffer 数据，无法进行类似 textureOffset 的邻域采样。这意味着屏幕空间遮挡（SSAO）、屏幕空间反射（SSR）等需要邻域采样的后处理算法无法在这个包含在片 MSAA 的 Render Pass 内完成。
  • 解决方案：必须在 Subpass 1 输出 resolved 的单采样 Color/Depth 图像到 VRAM 后，再新建一个普通的 Render Pass 或 Compute Pass 来处理后处理特效。

4. 性能与功耗的权衡 (Sample Shading Rate)

在片元着色器中对 subpassInputMS 进行循环采样并计算光照，实际上退化成了 SSAA（超级采样抗锯齿），因为每个 sample 都执行了一次完整的光照计算。

  • 如果只想对几何边缘进行 MSAA，而内部像素保持单采样光照，需要借助 VK_EXT_sample_locations 或在管线中启用 sampleShadingEnable = VK_FALSE，通过硬件的 centroid 采样或特定标记来避免多余的着色计算。否则，移动端芯片会迅速因发热而降频。
GfxArch Vulkan延迟渲染MSAA

评论点评