crt_helpers.h MSVC

flintlib · Nov 17, 2023 · 22c836b · 22c836b
1 parent 0a3aa59
commit 22c836b
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/crt_helpers.h b/src/crt_helpers.h
@@ -12,10 +12,18 @@
 #ifndef CRT_HELPERS_H
 #define CRT_HELPERS_H
 
-#if defined (__AVX2__)
-# include <x86intrin.h>
-#else
-# include <arm_neon.h>
+#if defined(__GNUC__)
+# if defined(__AVX2__)
+#  include <x86intrin.h>
+# elif defined(__ARM_NEON)
+#  include <arm_neon.h>
+# endif
+#elif defined(_MSC_VER)
+# if defined(__AVX2__)
+#  include <intrin.h>
+# elif defined(_M_ARM64)
+#  include <arm_neon.h>
+# endif
 #endif
 
 #include "flint.h"