启用MSAA后Metal图像块存储半透明片段出现artifacts问题求助

问题背景 #

我用Metal的image block以3层栈结构存储半透明片段，结构体定义如下：

#define TRANSPARENCY_LAYERS 3
struct TransparentFragmentValues {
    rgba8unorm<float4> color [[raster_order_group(0)]] [TRANSPARENCY_LAYERS];
    float depth [[raster_order_group(0)]] [TRANSPARENCY_LAYERS];
};

这些半透明片段会在最终阶段与其余渲染内容混合，原本运行正常，但启用*4倍多重采样抗锯齿（MSAA）*后，屏幕出现大量artifacts。

我的瓦片尺寸为16x16，每个片段占用64字节，无MSAA时总内存16kB，4xMSAA时为64kB，已达GPU内存限制。

我怀疑问题出在image block的初始化函数——原函数未考虑每个线程需初始化4个TransparentFragmentValues而非1个，但线程数必须保持16x16，请问如何正确索引这些样本？

原初始化函数代码：

kernel void transparent_fragment_store_init (
    imageblock<TransparentFragmentValues, imageblock_layout_explicit> blockData,
    ushort2 localThreadID[[thread_position_in_threadgroup]])
{
    threadgroup_imageblock TransparentFragmentValues* fragmentValues = blockData.data(localThreadID);
    for (short i = 0; i < TRANSPARENCY_LAYERS; i++) {
        fragmentValues->color[i] = 0.0;
        fragmentValues->depth[i] = INFINITY;
    }
}

解决方案 #

启用MSAA后，每个像素位置对应多个样本（4xMSAA即4个样本），image block中每个线程对应的像素位置会包含这4个样本的TransparentFragmentValues数据。你需要在初始化时遍历每个样本，通过blockData.data的重载方法传入样本索引来访问对应样本的结构体指针。

修改后的初始化函数如下：

kernel void transparent_fragment_store_init (
    imageblock<TransparentFragmentValues, imageblock_layout_explicit> blockData,
    ushort2 localThreadID[[thread_position_in_threadgroup]])
{
    // 遍历4xMSAA的所有样本
    for (ushort sampleIndex = 0; sampleIndex < 4; sampleIndex++) {
        threadgroup_imageblock TransparentFragmentValues* fragmentValues = blockData.data(localThreadID, sampleIndex);
        for (short i = 0; i < TRANSPARENCY_LAYERS; i++) {
            fragmentValues->color[i] = float4(0.0);
            fragmentValues->depth[i] = INFINITY;
        }
    }
}

关键说明 #

• blockData.data(localThreadID, sampleIndex)：该重载方法可以指定样本索引，获取对应样本的TransparentFragmentValues指针，确保每个样本的颜色和深度都被正确初始化
• 保持16x16的线程组规模，每个线程负责处理对应像素位置的所有4个样本，无需额外增加线程数
• 原代码只初始化了第一个样本，未初始化的样本数据会包含垃圾值，导致后续混合阶段出现artifacts；修改后所有样本都被初始化为透明黑色（颜色0）和最大深度（INFINITY），符合半透明片段栈的初始状态

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