macos aarch64 => fp16 support

sonos · Oct 11, 2023 · 9b86386 · 9b86386
1 parent f6e97e2
commit 9b86386
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/linalg/src/arm64.rs b/linalg/src/arm64.rs
@@ -64,7 +64,7 @@ lazy_static::lazy_static! {
 
 #[inline]
 pub fn has_fp16() -> bool {
-    cfg!(feature_cpu = "fp16") || *KIND == Kind::CortexA55 || *KIND == Kind::CortexA75 || *HAS_FP16
+    cfg!(target_os = "macos") || cfg!(feature_cpu = "fp16") || *KIND == Kind::CortexA55 || *KIND == Kind::CortexA75 || *HAS_FP16
 }
 
 #[target_feature(enable = "fp16")]