Przeglądaj źródła

Fixing compilation issue with Visual Studio 2022 in Arm64EC (#1450)

JbR 3 lat temu
rodzic
commit
e12ff84cd1
1 zmienionych plików z 9 dodań i 5 usunięć
  1. 9 5
      CMSIS/DSP/Include/arm_math_types.h

+ 9 - 5
CMSIS/DSP/Include/arm_math_types.h

@@ -95,12 +95,16 @@ extern "C"
 #endif
 
 #if defined(ARM_MATH_NEON)
-#include <arm_neon.h>
-#if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
-  #if !defined(ARM_MATH_NEON_FLOAT16)
-  #define ARM_MATH_NEON_FLOAT16
+  #if defined(_MSC_VER) && defined(_M_ARM64EC)
+    #include <arm64_neon.h>
+  #else
+    #include <arm_neon.h>
+  #endif
+  #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
+    #if !defined(ARM_MATH_NEON_FLOAT16)
+      #define ARM_MATH_NEON_FLOAT16
+    #endif
   #endif
-#endif
 #endif
 
 #if !defined(ARM_MATH_AUTOVECTORIZE)