Fix NATIVE_FP16 macro in our fork of half.h

Simple build unbreak for Raspberry Pi 5 -- fp16 is an optional aarch64 feature.
pytorch · Dec 27, 2024 · 31cc29b · 31cc29b
1 parent ed15042
commit 31cc29b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime/core/portable_type/half.h b/runtime/core/portable_type/half.h
@@ -15,7 +15,7 @@
 #include <ostream>
 
 #if defined(__GNUC__) || defined(__clang__)
-#if defined(__aarch64__)
+#if defined(__aarch64__) && defined(__ARM_FEATURE_FP16)
 #ifndef __ARM_V8_ONLY__
 #define NATIVE_FP16 1
 #endif // __ARM_V8_ONLY__