diff options
| author | Marat Dukhan <marat@fb.com> | 2018-09-06 19:11:46 +0300 |
|---|---|---|
| committer | Marat Dukhan <marat@fb.com> | 2018-09-06 19:11:46 +0300 |
| commit | 90a938f30ba414ada2f4b00674ee9631d7d85e19 (patch) | |
| tree | bf841b72bad785bb7405d6b427b023b15578ad69 | |
| parent | c7f1f995f37251aadf39ad5966fd4acedfc3afe8 (diff) | |
| download | psimd-90a938f30ba414ada2f4b00674ee9631d7d85e19.tar.gz | |
Include SSE/AVX intrinsics headers
| -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) |