From 0ccbdf0baa31464daaf56dcfcf71d3b2eedd8ea7 Mon Sep 17 00:00:00 2001
From: Dhairya Malhotra <dhairya.malhotra@gmail.com>
Date: Thu, 23 May 2024 15:19:48 -0400
Subject: [PATCH] Fix xmmintrin.h not found errors

---
 include/fmm_pts.txx | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/include/fmm_pts.txx b/include/fmm_pts.txx
index 49cfc2b..f1e1d23 100644
--- a/include/fmm_pts.txx
+++ b/include/fmm_pts.txx
@@ -17,17 +17,20 @@
 #include <sys/stat.h>
 #endif
 
-#ifdef __SSE__
-#include <xmmintrin.h>
-#endif
-#ifdef __SSE2__
-#include <emmintrin.h>
-#endif
-#ifdef __SSE3__
-#include <pmmintrin.h>
-#endif
-#ifdef __AVX__
-#include <immintrin.h>
+#if defined(__ARM_NEON)
+#  include SCTL_INCLUDE(sse2neon.h)
+#  define __SSE__
+#  define __SSE2__
+#  define __SSE3__
+#  define __SSE4__
+#  define __SSE4_1__
+#  define __SSE4_2__
+#elif defined(__SSE4_2__) || defined(__AVX__) || defined(__AVX512F__)
+#  ifdef _MSC_VER
+#    include <intrin.h>
+#  else
+#    include <x86intrin.h>
+#  endif
 #endif
 #if defined(__MIC__)
 #include <immintrin.h>