Windows10环境下wgpu WGSL计算着色器运行后返回全零缓冲区问题

问题原因及解决方案 #

你遇到的读取结果全零的问题，核心是GPU操作的同步逻辑缺失，以及可能的版本兼容性问题，以下是对应解决方法：

1. 手动插入内存屏障（最常见原因） #

wgpu不会自动在计算着色器写入存储缓冲、后续拷贝缓冲两个操作之间插入同步屏障，GPU可能在计算写入完成前就执行了拷贝操作，导致读取到未初始化的staging buffer内容。
在计算通道结束、copy_buffer_to_buffer调用前添加如下屏障代码即可：

// 针对0.12及更早版本的wgpu
encoder.push_buffer_barrier(
    &wgpu::BufferBarrier {
        buffer: &storage_buffer,
        src_access: wgpu::Access::STORAGE_WRITE,
        dst_access: wgpu::Access::COPY_READ,
    },
    wgpu::PipelineStages::COMPUTE_SHADER,
    wgpu::PipelineStages::COPY,
);

2. 验证缓冲初始化逻辑 #

可以先注释掉计算调度的代码，直接拷贝存储缓冲到staging缓冲读取：

• 如果读取结果为[1,2,3]，说明缓冲初始化、拷贝逻辑正常，问题确实出在计算同步环节
• 如果读取结果还是全零，说明create_buffer_init没有正确把初始数据上传到GPU，可替换为手动写入队列的方式验证：

// 替换create_buffer_init的手动上传代码
let storage_buffer = device.create_buffer(&BufferDescriptor {
    label: Some("storage buffer"),
    size,
    usage: BufferUsages::STORAGE | BufferUsages::COPY_DST | BufferUsages::COPY_SRC,
    mapped_at_creation: false,
});
queue.write_buffer(&storage_buffer, 0, cast_slice(&buffer));

3. 调整设备等待逻辑 #

部分Windows DX12后端对device.poll的处理存在差异，可调整等待逻辑，确保所有队列操作完成后再映射缓冲：

queue.submit(Some(encoder.finish()));
// 提交后立即等待所有操作完成，再执行映射
device.poll(Maintain::Wait);

let buf_slice = staging_buffer.slice(..);
let buf_future = buf_slice.map_async(MapMode::Read);
// 等待映射完成
block_on(buf_future).unwrap();

4. 版本兼容性验证 #

你使用的是旧版WGSL的[[attribute]]语法，如果你的wgpu版本在0.13及以上，官方已切换为@attribute的新WGSL语法，可替换着色器代码验证：

struct Numbers {
    @stride(4) data: array<u32>;
};

@group(0) @binding(0)
var<storage, read_write> numbers: Numbers;

@compute @workgroup_size(1)
fn main() {
    numbers.data[0] = numbers.data[0] + 1u;
    numbers.data[1] = numbers.data[1] + 1u;
    numbers.data[2] = numbers.data[2] + 1u;
}

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