From 986059694e7b39ead5e789ee42bd0bb62322b623 Mon Sep 17 00:00:00 2001
From: Ahmet Inan <inan@aicodix.de>
Date: Fri, 5 May 2023 08:03:34 +0200
Subject: [PATCH] added support for aarch64

__ARM_NEON__ is not defined on gcc anymore ..
---
 polar_helper.hh | 6 +++---
 rotate.hh       | 2 +-
 simd.hh         | 2 +-
 tests/Makefile  | 3 +++
 4 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/polar_helper.hh b/polar_helper.hh
index b31b5d4..84a02ba 100644
--- a/polar_helper.hh
+++ b/polar_helper.hh
@@ -129,7 +129,7 @@ struct PolarHelper<SIMD<int8_t, WIDTH>>
 	}
 	static TYPE qmul(TYPE a, TYPE b)
 	{
-#ifdef __ARM_NEON__
+#ifdef __ARM_NEON
 		return vmul(a, b);
 #else
 		return vsign(a, b);
@@ -137,7 +137,7 @@ struct PolarHelper<SIMD<int8_t, WIDTH>>
 	}
 	static TYPE prod(TYPE a, TYPE b)
 	{
-#ifdef __ARM_NEON__
+#ifdef __ARM_NEON
 		return vmul(vmul(vsignum(a), vsignum(b)), vmin(vqabs(a), vqabs(b)));
 #else
 		return vsign(vmin(vqabs(a), vqabs(b)), vsign(vsignum(a), b));
@@ -145,7 +145,7 @@ struct PolarHelper<SIMD<int8_t, WIDTH>>
 	}
 	static TYPE madd(TYPE a, TYPE b, TYPE c)
 	{
-#ifdef __ARM_NEON__
+#ifdef __ARM_NEON
 		return vqadd(vmul(a, vmax(b, vdup<TYPE>(-127))), c);
 #else
 		return vqadd(vsign(vmax(b, vdup<TYPE>(-127)), a), c);
diff --git a/rotate.hh b/rotate.hh
index f06963e..93fc188 100644
--- a/rotate.hh
+++ b/rotate.hh
@@ -30,7 +30,7 @@ public:
 	}
 };
 
-#ifdef __ARM_NEON__
+#ifdef __ARM_NEON
 template <int WIDTH>
 class Rotate<SIMD<int8_t, 16>, WIDTH>
 {
diff --git a/simd.hh b/simd.hh
index 21dc163..9f557e3 100644
--- a/simd.hh
+++ b/simd.hh
@@ -1504,7 +1504,7 @@ static inline SIMD<double, WIDTH> vshuf(SIMD<double, WIDTH> a, SIMD<uint64_t, WI
 #endif
 #endif
 
-#ifdef __ARM_NEON__
+#ifdef __ARM_NEON
 #include "neon.hh"
 #endif
 #endif
diff --git a/tests/Makefile b/tests/Makefile
index 0871848..f83ac34 100644
--- a/tests/Makefile
+++ b/tests/Makefile
@@ -8,6 +8,9 @@ CXX = clang++ -stdlib=libc++ -march=native
 #CXX = armv7a-hardfloat-linux-gnueabi-g++ -static -mfpu=neon -march=armv7-a
 #QEMU = qemu-arm
 
+#CXX = aarch64-unknown-linux-gnu-g++ -static
+#QEMU = qemu-aarch64
+
 .PHONY: clean test
 
 tests := $(basename $(wildcard *_test.*))