WGSL片段着色器critical section使用纹理采样值报错咨询

在WGSL片段着色器的临界区中使用依赖纹理采样值的变量时出现错误。已知路径分歧情况下不应在临界区中采样纹理，因此已将fragData的初始化移至临界区外，让getPBRColor()在临界区外执行，但fragData仍依赖采样得到的color，在临界区中使用时依然报错。疑问：是否完全无法在临界区中使用纹理采样值？

注：忽略aBuffer的实际用法，重点是无法将颜色写入其中。

着色器代码 #

struct FragData {
    color: vec4<f32>,
    depth: f32,
}

@group(0) @binding(0) var baseColorTexture: texture_2d<f32>;
@group(0) @binding(1) var baseColorSampler: sampler;

// 用于写入片段颜色和深度的累积缓冲区
@group(1) @binding(0) var<storage, read_write> aBuffer: array<FragData>;

// 每个像素的自旋锁，字节大小 = resolution.x * resolution.y * sizeOf(u32)
@group(1) @binding(1) var<storage, read_write> spinLock: array<atomic<u32>>;

@fragment
fn_main(input: FragmentInput) {
    let color: vec4<f32> = getPBRColor(input); // 采样baseColorTexture获取颜色
    let fragDepth = input.position.z;
    let fragData = FragData(color, fragDepth); // 导致崩溃
    //let fragData = FragData(vec4(0.0f), fragDepth); // 运行正常

    let pixelIndex = input.position.y * resolution.x + input.position.x;

    var keepWaiting = true;
    while (keepWaiting) {
        // 临界区：尝试获取锁
        if (atomicCompareExchangeWeak(&spinLock[pixelIndex], 0u, 1u).exchanged) {
            // 根据fragData.color是否使用采样值，此处会触发错误
            aBuffer[pixelIndex] = fragData;
        
            // 释放锁
            atomicStore(&spinLock[pixelIndex], 0u);
            keepWaiting = false;
        }
    }
}

报错信息 #

GLDRendererMetal命令缓冲区完成错误：Error Domain=MTLCommandBufferErrorDomain Code=1 "已丢弃（GPU错误/恢复的受害者）(00000005:kIOGPUCommandBufferCallbackErrorInnocentVictim)" UserInfo={NSLocalizedDescription=已丢弃（GPU错误/恢复的受害者）(00000005:kIOGPUCommandBufferCallbackErrorInnocentVictim)}

核心原因 #

GPU的内存模型和同步规则限制了临界区（自旋锁保护的代码段）内的操作：

• 纹理采样属于异步/延迟操作，采样结果可能依赖后续的内存访问或指令调度，而临界区要求操作是完全同步、无依赖的原子性执行。
• 即使采样在临界区外完成，采样结果的内存位置可能仍处于GPU的临时缓存中，并未完成写入到通用寄存器或可安全访问的内存区域，在临界区中引用这类值会触发内存访问冲突，导致GPU命令缓冲区被丢弃。

解决方法 #

1. 强制采样结果可见性：在进入临界区前，将采样得到的颜色值显式写入临时本地变量，通过无意义的算术操作触发编译器将值固化到寄存器，避免延迟求值：

let color: vec4<f32> = getPBRColor(input);
// 强制编译器确认color的值，固化到寄存器
var tempColor = color;
tempColor += vec4(0.0); 
let fragData = FragData(tempColor, fragDepth);

2. 重构临界区逻辑：将写入操作拆分，只在临界区内执行必要的原子写操作，避免传递依赖采样的复杂结构体：

@fragment
fn_main(input: FragmentInput) {
    let color: vec4<f32> = getPBRColor(input);
    let fragDepth = input.position.z;
    let pixelIndex = input.position.y * resolution.x + input.position.x;

    var keepWaiting = true;
    while (keepWaiting) {
        if (atomicCompareExchangeWeak(&spinLock[pixelIndex], 0u, 1u).exchanged) {
            // 直接写入拆分后的数据，规避结构体传递的依赖问题
            aBuffer[pixelIndex].color = color;
            aBuffer[pixelIndex].depth = fragDepth;
        
            atomicStore(&spinLock[pixelIndex], 0u);
            keepWaiting = false;
        }
    }
}

3. 替换自旋锁方案：GPU线程模型不适合CPU式自旋锁，极易引发死锁或内存冲突。推荐使用WGSL内置的原子操作+内存作用域实现像素同步，或采用顺序无关透明度（OIT）类的累积缓冲区方案，完全规避自旋锁的使用。

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