Fixed refactored GPU code.

zlogic · Jan 21, 2024 · d608476 · d608476
1 parent 8df68df
commit d608476
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/correlation.rs b/src/correlation.rs
@@ -970,7 +970,7 @@ mod gpu {
         ) -> Result<(), Box<dyn error::Error>> {
             let max_width = img1.width().max(img2.width());
             let max_height = img1.height().max(img2.height());
-            let max_shape = (max_height, max_width);
+            let max_shape = (max_width, max_height);
             let img1_shape = (img1.width(), img1.height());
             let out_shape = match dir {
                 CorrelationDirection::Forward => self.img1_shape,
@@ -1143,7 +1143,7 @@ mod gpu {
                 .device
                 .create_command_encoder(&wgpu::CommandEncoderDescriptor { label: None });
             {
-                let workgroup_size = ((shape.0 + 15) / 16, ((shape.1 + 15) / 16));
+                let workgroup_size = ((shape.1 + 15) / 16, ((shape.0 + 15) / 16));
                 let mut cpass = encoder.begin_compute_pass(&wgpu::ComputePassDescriptor {
                     label: None,
                     timestamp_writes: None,