fix: cross compilation

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

Author: usamoi

crates/c/src/c.c

@@ -1,7 +1,12 @@
 #include "c.h"
-#include <immintrin.h>
 #include <math.h>
 
+#if defined(__x86_64__)
+#include <immintrin.h>
+#endif
+
+#if defined(__x86_64__)
+
 __attribute__((target("avx512fp16,bmi2"))) extern float
 v_f16_cosine_axv512(_Float16 const *restrict a, _Float16 const *restrict b,
                     size_t n) {
@@ -115,3 +120,5 @@ v_f16_sl2_axv2(_Float16 const *restrict a, _Float16 const *restrict b,
   }
   return dd;
 }
+
+#endif

crates/c/src/c.h

@@ -1,9 +1,13 @@
 #include <stddef.h>
 #include <stdint.h>
 
+#if defined(__x86_64__)
+
 extern float v_f16_cosine_axv512(_Float16 const *, _Float16 const *, size_t n);
 extern float v_f16_dot_axv512(_Float16 const *, _Float16 const *, size_t n);
 extern float v_f16_sl2_axv512(_Float16 const *, _Float16 const *, size_t n);
 extern float v_f16_cosine_axv2(_Float16 const *, _Float16 const *, size_t n);
 extern float v_f16_dot_axv2(_Float16 const *, _Float16 const *, size_t n);
 extern float v_f16_sl2_axv2(_Float16 const *, _Float16 const *, size_t n);
+
+#endif