From e5a6e60f5b4aa98420bbcd934440744411240ed7 Mon Sep 17 00:00:00 2001
From: JbR <90027771+jbr-smtg@users.noreply.github.com>
Date: Mon, 4 Apr 2022 11:17:57 +0200
Subject: [PATCH] Fixing compilation issue with Visual Studio 2022 in Arm64EC
 (#1450)

---
 Include/arm_math_types.h | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/Include/arm_math_types.h b/Include/arm_math_types.h
index 775b33c4..c615e667 100755
--- a/Include/arm_math_types.h
+++ b/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)