fix: x86 simd (#5)

Signed-off-by: Keming <[email protected]>

src/simd.rs

@@ -81,6 +81,9 @@ pub unsafe fn l2_squared_distance(lhs: &[f32], rhs: &[f32]) -> f32 {
 #[target_feature(enable = "avx,avx2")]
 #[inline]
 pub unsafe fn vector_binarize_query(vec: &[u8], binary: &mut [u64]) {
+    #[cfg(target_arch = "x86")]
+    use std::arch::x86::*;
+    #[cfg(target_arch = "x86_64")]
     use std::arch::x86_64::*;
 
     let length = vec.len();
@@ -112,6 +115,9 @@ pub unsafe fn vector_binarize_query(vec: &[u8], binary: &mut [u64]) {
 #[target_feature(enable = "avx")]
 #[inline]
 pub unsafe fn min_max_residual(res: &mut [f32], x: &[f32], y: &[f32]) -> (f32, f32) {
+    #[cfg(target_arch = "x86")]
+    use std::arch::x86::*;
+    #[cfg(target_arch = "x86_64")]
     use std::arch::x86_64::*;
 
     let mut min_32x8 = _mm256_set1_ps(f32::MAX);
@@ -182,6 +188,9 @@ pub unsafe fn scalar_quantize(
     lower_bound: f32,
     multiplier: f32,
 ) -> u32 {
+    #[cfg(target_arch = "x86")]
+    use std::arch::x86::*;
+    #[cfg(target_arch = "x86_64")]
     use std::arch::x86_64::*;
 
     let mut quantize_ptr = quantized.as_mut_ptr() as *mut u64;
@@ -246,6 +255,9 @@ pub unsafe fn scalar_quantize(
 #[target_feature(enable = "fma,avx,avx2")]
 #[inline]
 pub unsafe fn vector_dot_product(lhs: &[f32], rhs: &[f32]) -> f32 {
+    #[cfg(target_arch = "x86")]
+    use std::arch::x86::*;
+    #[cfg(target_arch = "x86_64")]
     use std::arch::x86_64::*;
 
     let mut lhs_ptr = lhs.as_ptr();
@@ -312,6 +324,9 @@ pub unsafe fn vector_dot_product(lhs: &[f32], rhs: &[f32]) -> f32 {
 #[target_feature(enable = "sse2,avx,avx2")]
 #[inline]
 pub unsafe fn binary_dot_product(lhs: &[u64], rhs: &[u64]) -> u32 {
+    #[cfg(target_arch = "x86")]
+    use std::arch::x86::*;
+    #[cfg(target_arch = "x86_64")]
     use std::arch::x86_64::*;
 
     let mut sum = 0;