From 9abfe212f856f02a29c220a081df8c5a0b0b706c Mon Sep 17 00:00:00 2001
From: Steven Noonan <steven@uplinklabs.net>
Date: Fri, 21 Jun 2024 04:32:55 -0700
Subject: [PATCH] nbody_util: fix compile on Windows ARM64

Signed-off-by: Steven Noonan <steven@uplinklabs.net>
---
 src/nbody_util.cpp | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/nbody_util.cpp b/src/nbody_util.cpp
index cdc624a..1ffcb2c 100644
--- 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"