nbody_util: fix compile on Windows ARM64

Signed-off-by: Steven Noonan <[email protected]>
tycho · Jun 21, 2024 · 9abfe21 · 9abfe21
1 parent 4514906
commit 9abfe21
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/nbody_util.cpp b/src/nbody_util.cpp
@@ -43,12 +43,8 @@
 #ifdef _WIN32
 #include <windows.h>
 #else
-#if !defined(__aarch64__)
+#if !defined(__aarch64__) && !defined(_M_ARM64)
 #include <mm_malloc.h>
-#else
-#include <stdlib.h>
-#define _mm_malloc(x,y) malloc(x)
-#define _mm_free free
 #endif
 #include <math.h>
 #include <stdio.h>
@@ -57,6 +53,11 @@
 #include <fcntl.h>
 #include <sys/select.h>
 #endif
+#if defined(__arch64__) || defined(_M_ARM64)
+#include <stdlib.h>
+#define _mm_malloc(x,y) malloc(x)
+#define _mm_free free
+#endif
 #include <random>
 
 #include "nbody_util.h"