diff options
Diffstat (limited to 'include')
| -rw-r--r-- | include/psimd.h | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/include/psimd.h b/include/psimd.h index 3cd5ca8..b8aede6 100644 --- a/include/psimd.h +++ b/include/psimd.h @@ -28,8 +28,36 @@ #define PSIMD_INTRINSIC static #endif -#if defined(__ARM_NEON__) - #include <arm_neon.h> +#if defined(__GNUC__) + #if defined(__ARM_NEON__) || defined(__ARM_NEON) + #include <arm_neon.h> + #endif + + #if defined(__SSE2__) + #include <emmintrin.h> + #endif + + #if defined(__SSE3__) + #include <pmmintrin.h> + #endif + + #if defined(__SSSE3__) + #include <tmmintrin.h> + #endif + + #if defined(__SSE4_1__) + #include <smmintrin.h> + #endif + + #if defined(__SSE4_2__) + #include <nmmintrin.h> + #endif + + #if defined(__AVX__) + #include <immintrin.h> + #endif +#elif defined(_MSC_VER) + #include <intrin.h> #endif #if defined(__cplusplus) |