Merge pull request alibaba#2038 from DaydreamCoding/patch-11

Fix VecHalf.hpp not include neon header
runngezhang · Sep 5, 2022 · 41db47f · 41db47f
2 parents 5fbfa5a + 206f73e
commit 41db47f
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/source/backend/cpu/bf16/VecHalf.hpp b/source/backend/cpu/bf16/VecHalf.hpp
@@ -12,6 +12,18 @@
 #include <stdint.h>
 #include <array>
 #include <algorithm>  // supply std::max and std::min
+
+#ifdef MNN_USE_NEON
+#include <arm_neon.h>
+#endif
+#ifdef MNN_USE_SSE
+#if defined(_MSC_VER)
+#include <intrin.h>
+#else
+#include <x86intrin.h>
+#endif
+#endif
+
 namespace MNN {
 namespace Math {
 
@@ -160,11 +172,6 @@ struct VecHalf {
 };
 
 #if defined(MNN_USE_SSE)
-#if defined(_MSC_VER)
-#include <intrin.h>
-#else
-#include <x86intrin.h>
-#endif
 
 template<>
 struct VecHalf<4> {
@@ -310,7 +317,6 @@ struct VecHalf<4> {
 #endif
 
 #if defined(MNN_USE_NEON)
-#include <arm_neon.h>
 
 template<>
 struct VecHalf<4> {