使用bytemuck与WGPU向GPU传递Rust结构体的对齐问题

以下是原始及修改后的代码，核心问题为内存对齐规则不匹配：

原始代码（出现数据错乱） #

Rust端 #

const MAX_SPHERES: usize = 4;

#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct Sphere {
    pub position: [f32; 3],
    pub color: [f32; 3],
    pub radius: f32,
    _padding: u32,
}
impl Sphere {
    pub fn new(position: [f32; 3], radius: f32, color: [f32; 3]) -> Self {
        Self {
            position,
            radius,
            color,
            _padding: 0,
        }
    }
}

#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct GlobalUniform {
    pub resolution: [u32; 2],
    pub time: f32,

    sphere_count: u32,
    spheres: [Sphere; MAX_SPHERES],
}
impl GlobalUniform {
    pub fn new(resolution: [u32; 2]) -> Self {
        Self {
            resolution,
            time: 0.0,

            sphere_count: 0,
            spheres: [Sphere::new([0.0; 3], 0.0, [0.0; 3]); MAX_SPHERES],
        }
    }
}

WGSL端 #

const MAX_SPHERES: u32 = 4u;

@group(1)
@binding(0)
var<uniform> globals: Globals;
struct Globals {
    resolution: vec2<u32>,
    time: f32,

    sphere_count: u32,
    spheres: array<Sphere, MAX_SPHERES>,
};

struct Sphere {
    position: vec3<f32>,
    color: vec3<f32>,
    radius: f32,
};

修改后的代码（问题仍存在） #

Rust端 #

const MAX_SPHERES: usize = 4;

#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct GeometryUniform {
    sphere_count: u32,
    spheres: [Sphere; MAX_SPHERES],
    _padding: [u32; 3],
}
impl GeometryUniform {
    pub fn new() -> Self {
        Self {
            sphere_count: 0,
            spheres: [Sphere::new([0.0; 3], 0.0, [0.0; 3]); MAX_SPHERES],
            _padding: [0; 3],
        }
    }

    pub fn push_sphere(&mut self, sphere: Sphere) {
        if self.sphere_count < MAX_SPHERES as u32 {
            self.spheres[self.sphere_count as usize] = sphere;
            self.sphere_count += 1;
        }
    }
}

#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct Sphere {
    pub position: [f32; 3],
    pub color: [f32; 3],
    pub radius: f32,
    _padding: u32,
}
impl Sphere {
    pub fn new(position: [f32; 3], radius: f32, color: [f32; 3]) -> Self {
        Self {
            position,
            radius,
            color,
            _padding: 0,
        }
    }
}

WGSL端 #

const MAX_SPHERES: u32 = 4u;

@group(2)
@binding(0)
var<uniform> geometry: Geometry;
struct Geometry {
    sphere_count: u32,
    spheres: array<Sphere, MAX_SPHERES>,
};
struct Sphere {
    position: vec3<f32>,
    color: vec3<f32>,
    radius: f32,
};

问题根源：WGSL与C语言（Rust #[repr(C)]）的内存对齐规则差异 #

1. Sphere结构体的布局差异 #

• WGSL规则：vec3<f32>的对齐要求是16字节（等价于vec4<f32>的对齐），因此每个vec3<f32>成员后会自动填充4字节，确保下一个成员对齐到16字节边界。最终WGSL中Sphere的内存布局为：
  • position: 0-11字节 → 补4字节到16字节
  • color: 16-27字节 → 补4字节到32字节
  • radius: 32-35字节 → 补4字节到32字节（结构体总大小为32字节，满足16字节对齐）
• Rust #[repr(C)]规则：[f32;3]不会自动填充，成员紧密排列，你的Rust Sphere布局为：
  • position: 0-11字节
  • color: 12-23字节
  • radius:24-27字节
  • _padding:28-31字节
    两者布局完全错位，导致数据解析错乱。

2. GeometryUniform的数组对齐问题 #

WGSL中，数组的起始地址必须对齐到其元素类型的对齐值（这里是16字节）。sphere_count是4字节的u32，WGSL会自动在其后填充12字节，让spheres数组起始于16字节边界；但你的Rust代码直接将spheres放在sphere_count之后，没有补充填充，导致数组起始位置偏移。

修正方案：让Rust结构体严格匹配WGSL布局 #

修正后的Rust代码 #

const MAX_SPHERES: usize = 4;

#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct Sphere {
    pub position: [f32; 3],
    _pad1: u32, // 匹配WGSL中vec3的自动填充
    pub color: [f32; 3],
    _pad2: u32, // 匹配WGSL中vec3的自动填充
    pub radius: f32,
    _pad3: u32, // 确保结构体总大小为32字节，满足16字节对齐
}

impl Sphere {
    pub fn new(position: [f32; 3], radius: f32, color: [f32; 3]) -> Self {
        Self {
            position,
            color,
            radius,
            _pad1: 0,
            _pad2: 0,
            _pad3: 0,
        }
    }
}

#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct GeometryUniform {
    sphere_count: u32,
    _pad: [u32; 3], // 补12字节，让spheres数组对齐到16字节边界
    spheres: [Sphere; MAX_SPHERES],
}

impl GeometryUniform {
    pub fn new() -> Self {
        Self {
            sphere_count: 0,
            spheres: [Sphere::new([0.0; 3], 0.0, [0.0; 3]); MAX_SPHERES],
            _pad: [0; 3],
        }
    }

    pub fn push_sphere(&mut self, sphere: Sphere) {
        if self.sphere_count < MAX_SPHERES as u32 {
            self.spheres[self.sphere_count as usize] = sphere;
            self.sphere_count += 1;
        }
    }
}

WGSL代码保持不变 #

WGSL会自动处理内部填充，无需修改。

验证方法 #

• 使用bytemuck::bytes_of打印Rust结构体的字节序列，检查大小和填充是否符合预期
• 使用naga-cli validate命令分析WGSL代码，查看结构体的内存布局和偏移量，确保与Rust端一致

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