pub fn linear_index_3d( x: usize, y: usize, z: usize, dim_x: usize, dim_y: usize, ) -> usize
Flatten a 3D dispatch (x, y, z) into a linear index, given grid dimensions.