From 1eca5d8ea1423c7da364531d3fed4fc23e62a23a Mon Sep 17 00:00:00 2001 From: Markus Ebner Date: Sat, 22 Jun 2024 18:25:53 +0200 Subject: [PATCH] Fix build for aarch64 --- src/svm/kernel/rbf.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/svm/kernel/rbf.rs b/src/svm/kernel/rbf.rs index 67c88e9..bf0a1b0 100644 --- 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, feature: &VectorD, out #[cfg(target_arch = "aarch64")] #[inline] fn compute(rbf: Rbf, vectors: &MatrixD, feature: &VectorD, 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)