Merge pull request #376 from robertknight/avx512-fix

Fix AVX-512 build of im2col.rs
robertknight · Oct 8, 2024 · b56f459 · b56f459
2 parents 099268d + 9fac2af
commit b56f459
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops/conv/im2col.rs b/src/ops/conv/im2col.rs
@@ -304,7 +304,7 @@ impl<'a> VirtualIm2Col<'a, f32> {
         cols: Range<usize>,
     ) {
         use std::arch::x86_64::__m512i;
-        const NR_REGS: usize = vec_count::<__m512>(KERNEL_AVX512_NR);
+        const NR_REGS: usize = vec_count::<__m512i>(KERNEL_AVX512_NR);
         self.pack_b_impl::<__m512i, NR_REGS>(out, panel_width, rows.clone(), cols.clone());
     }
 }