WGPU中Buffer创建写入读取问题及CPU/GPU可访问性咨询

一、读取MAP_READ类型Buffer的方法 #

WGPU中带MAP_READ标记的Buffer无法同步读取，必须通过异步映射获取CPU可访问的内存视图，具体步骤如下：

1. 调用buffer.map_async发起映射请求，指定映射范围（通常为整个Buffer）；
2. 等待映射完成（需用async/await或Future机制）；
3. 提取映射后的内存切片，将数据复制到CPU侧变量；
4. 调用buffer.unmap()解除映射，释放资源。

针对你的代码，完整读取示例：

use futures_lite::future; // 可替换为tokio等异步运行时

async fn read_buffer_data(buffer: &wgpu::Buffer) -> Vec<i32> {
    // 发起全Buffer的读映射请求
    buffer.map_async(wgpu::MapMode::Read, 0, buffer.size()).await.expect("映射Buffer失败");
    
    // 将内存视图转换为i32切片并收集为Vec
    let data_slice = buffer.get_mapped_range(0, buffer.size()).as_slice();
    let data: Vec<i32> = data_slice.chunks_exact(std::mem::size_of::<i32>())
        .map(|chunk| i32::from_ne_bytes(chunk.try_into().unwrap()))
        .collect();
    
    // 解除映射
    buffer.unmap();
    
    data
}

// 在异步上下文执行读取
let result = future::block_on(read_buffer_data(&array_buffer));
println!("Buffer内容: {:?}", result);

注意：queue.write_buffer是异步提交给GPU的操作，读取前建议调用queue.submit(None)等待队列任务完成，避免读取到未写入的旧数据。

二、不同CPU/GPU访问性的Buffer定义示例 #

Buffer的访问权限完全由usage字段控制，以下是三种核心场景的定义：

1. GPU独占Buffer（仅GPU可读写，性能最优） #

适用于顶点、索引、静态着色器存储等无需CPU干预的数据：

let gpu_only_buffer = device.create_buffer(&wgpu::BufferDescriptor {
    label: Some("GPU独占Buffer"),
    size: 1024 * std::mem::size_of::<f32>() as u64, // 4KB的f32数据
    usage: wgpu::BufferUsages::VERTEX | wgpu::BufferUsages::STORAGE,
    mapped_at_creation: false,
});

2. CPU写→GPU读Buffer（CPU频繁更新，GPU读取） #

适用于Uniform变量、动态顶点数据等场景：

let cpu_write_gpu_read_buffer = device.create_buffer(&wgpu::BufferDescriptor {
    label: Some("CPU写GPU读Buffer"),
    size: 64 * std::mem::size_of::<f32>() as u64, // 256字节Uniform数据
    usage: wgpu::BufferUsages::MAP_WRITE | wgpu::BufferUsages::UNIFORM | wgpu::BufferUsages::COPY_DST,
    mapped_at_creation: false,
});

// 异步写入示例
async fn write_to_buffer(buffer: &wgpu::Buffer, data: &[f32]) {
    buffer.map_async(wgpu::MapMode::Write, 0, buffer.size()).await.expect("映射Buffer失败");
    let mut data_slice = buffer.get_mapped_range_mut(0, buffer.size()).as_mut_slice();
    data_slice.copy_from_slice(bytemuck::cast_slice(data));
    buffer.unmap();
}

3. CPU读←GPU写Buffer（GPU写入后CPU读取） #

即你当前使用的类型，用于获取GPU计算结果：

let cpu_read_gpu_write_buffer = device.create_buffer(&wgpu::BufferDescriptor {
    label: Some("CPU读GPU写Buffer"),
    size: 128 * std::mem::size_of::<i32>() as u64,
    usage: wgpu::BufferUsages::MAP_READ | wgpu::BufferUsages::COPY_DST,
    mapped_at_creation: false,
});

注：GPU需先将数据复制到该Buffer（比如从GPU独占的Storage Buffer复制），CPU才能读取到有效数据。

另外，你代码中的size计算可能存在问题：如果arr是Vec<i32>，应该用arr.len() * std::mem::size_of::<i32>()，而非arr.len() * std::mem::size_of::<[i32;5]>，否则会导致Buffer大小远超实际数据需求，读取时出现冗余垃圾数据。

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