diff options
| -rw-r--r-- | vecmathlib.h | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/vecmathlib.h b/vecmathlib.h index 0d72add..5fe2832 100644 --- a/vecmathlib.h +++ b/vecmathlib.h @@ -141,7 +141,12 @@ class type_info; #include "vec_avx_fp16_16.h" #include "vec_avx_float8.h" #include "vec_avx_double4.h" -#define VML_CONFIG_AVX " AVX" +#if defined __AVX2__ +#define VML_CONFIG_AVX2 " AVX2" +#else +#define VML_CONFIG_AVX2 +#endif +#define VML_CONFIG_AVX " AVX" VML_CONFIG_AVX2 #else #define VML_CONFIG_AVX #endif |
