fix: simd binary ip for 32bit (#6)

Signed-off-by: Keming <[email protected]>
kemingy · Dec 3, 2024 · 6fd34f0 · 6fd34f0
1 parent 1007380
commit 6fd34f0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/simd.rs b/src/simd.rs
@@ -377,7 +377,8 @@ pub unsafe fn binary_dot_product(lhs: &[u64], rhs: &[u64]) -> u32 {
         _mm256_castsi256_si128(sum256),
         _mm256_extracti128_si256(sum256, 1),
     );
-    sum += _mm_cvtsi128_si64(_mm_add_epi64(xa, _mm_shuffle_epi32(xa, 78))) as u32;
+    // this assumes the length is less than 2 ** 31
+    sum += _mm_cvtsi128_si32(_mm_add_epi64(xa, _mm_shuffle_epi32(xa, 78))) as u32;
 
     sum
 }