Function compute_num_workgroups_3d 

pub fn compute_num_workgroups_3d(
    total: [u32; 3],
    workgroup_size: [u32; 3],
) -> [u32; 3]

Compute workgroup counts for a 3D dispatch.