Merge pull request #10 from seijikun/master

Fix build for aarch64
ralfbiedert · Jun 22, 2024 · ca9701d · ca9701d
2 parents 4f33f15 + 1eca5d8
commit ca9701d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/svm/kernel/rbf.rs b/src/svm/kernel/rbf.rs
@@ -1,4 +1,5 @@
 use std::convert::{From, TryFrom};
+use std::arch;
 
 use super::{KernelDense, KernelSparse};
 use crate::{
@@ -66,7 +67,7 @@ fn compute(rbf: Rbf, vectors: &MatrixD<f32s, Rows>, feature: &VectorD<f32s>, out
 #[cfg(target_arch = "aarch64")]
 #[inline]
 fn compute(rbf: Rbf, vectors: &MatrixD<f32s, Rows>, feature: &VectorD<f32s>, output: &mut [f64]) {
-    if is_aarch64_feature_detected!("neon") {
+    if arch::is_aarch64_feature_detected!("neon") {
         unsafe { compute_neon(rbf, vectors, feature, output) }
     } else {
         compute_core(rbf, vectors, feature, output)