From 2d9e30c373b4ea9372ee3906a06aa24a31fa81a6 Mon Sep 17 00:00:00 2001 From: Ivan Belyavtsev Date: Tue, 16 Jul 2024 13:48:19 +0300 Subject: [PATCH] set SSE4 intrinsic level for NEON --- MathLib.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/MathLib.h b/MathLib.h index 59ea5fb..876abe2 100644 --- a/MathLib.h +++ b/MathLib.h @@ -89,6 +89,7 @@ #include #elif( defined(__arm__) || defined(__aarch64__) || defined(_M_ARM64) || defined(_M_ARM) ) #include "External/sse2neon/sse2neon.h" + #define __NEON__ #else #include #endif @@ -131,7 +132,7 @@ #define ML_INTRINSIC_LEVEL ML_INTRINSIC_AVX2 #elif defined ( __AVX__ ) #define ML_INTRINSIC_LEVEL ML_INTRINSIC_AVX1 - #elif defined ( __SSE4_2__ ) || defined ( __SSE4_1__ ) + #elif defined ( __SSE4_2__ ) || defined ( __SSE4_1__ ) || defined ( __NEON__ ) #define ML_INTRINSIC_LEVEL ML_INTRINSIC_SSE4 #else #define ML_INTRINSIC_LEVEL ML_INTRINSIC_SSE3