From ff088c256d3dae156777fc9399b0f1bac8f60652 Mon Sep 17 00:00:00 2001
From: Radzivon Bartoshyk <radzivon.bartoshyk@proton.me>
Date: Sat, 4 Jan 2025 15:35:30 +0000
Subject: [PATCH] RGBA u16 12<bp AVX-VNNI

---
 src/dispatch_group_u16.rs | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/src/dispatch_group_u16.rs b/src/dispatch_group_u16.rs
index d497ca9..4eeb1b4 100644
--- a/src/dispatch_group_u16.rs
+++ b/src/dispatch_group_u16.rs
@@ -133,21 +133,21 @@ pub(crate) fn convolve_horizontal_dispatch_u16<const CHANNELS: usize>(
             });
     } else {
         let approx = filter_weights.numerical_approximation_i16::<PRECISION>(0);
-        // dst.chunks_exact_mut(dst_stride * 4)
-        //     .zip(src.chunks_exact(src_stride * 4))
-        //     .for_each(|(dst, src)| {
-        //         u16::handle_fixed_row_4::<i32, CHANNELS>(
-        //             src,
-        //             src_stride,
-        //             dst,
-        //             dst_stride,
-        //             &approx,
-        //             bit_depth as u32,
-        //         );
-        //     });
+        dst.chunks_exact_mut(dst_stride * 4)
+            .zip(src.chunks_exact(src_stride * 4))
+            .for_each(|(dst, src)| {
+                u16::handle_fixed_row_4::<i32, CHANNELS>(
+                    src,
+                    src_stride,
+                    dst,
+                    dst_stride,
+                    &approx,
+                    bit_depth as u32,
+                );
+            });
 
-        let remainder = dst;//dst.chunks_exact_mut(dst_stride * 4).into_remainder();
-        let src_remainder = src;//;src.chunks_exact(src_stride * 4).remainder();
+        let remainder = dst.chunks_exact_mut(dst_stride * 4).into_remainder();
+        let src_remainder = src.chunks_exact(src_stride * 4).remainder();
 
         remainder
             .chunks_exact_mut(dst_stride)