fix: simd binary ip for 32bit

kemingy · kemingy · commit ab291922900a · 2024-12-03T21:05:26.000+08:00
Signed-off-by: Keming &lt;kemingy94@gmail.com&gt;
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
 }

Original file line number	Diff line number	Diff line change
`@@ -377,7 +377,8 @@ pub unsafe fn binary_dot_product(lhs: &[u64], rhs: &[u64]) -> u32 {`
`377`	`377`	`_mm256_castsi256_si128(sum256),`
`378`	`378`	`_mm256_extracti128_si256(sum256, 1),`
`379`	`379`	`);`
`380`		`- sum += _mm_cvtsi128_si64(_mm_add_epi64(xa, _mm_shuffle_epi32(xa, 78))) as u32;`
	`380`	`+ // this assumes the length is less than 2 ** 31`
	`381`	`+ sum += _mm_cvtsi128_si32(_mm_add_epi64(xa, _mm_shuffle_epi32(xa, 78))) as u32;`
`381`	`382`
`382`	`383`	`sum`
`383`	`384`	`}`