diff options
Diffstat (limited to 'intgemm/vec_traits.h')
-rw-r--r-- | intgemm/vec_traits.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/intgemm/vec_traits.h b/intgemm/vec_traits.h index 86265b2..948dae1 100644 --- a/intgemm/vec_traits.h +++ b/intgemm/vec_traits.h @@ -18,11 +18,13 @@ template <> struct vector_s<CPUType::SSSE3, int16_t> { using type = __m128i; }; template <> struct vector_s<CPUType::SSSE3, int> { using type = __m128i; }; template <> struct vector_s<CPUType::SSSE3, float> { using type = __m128; }; template <> struct vector_s<CPUType::SSSE3, double> { using type = __m128d; }; +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 template <> struct vector_s<CPUType::AVX2, int8_t> { using type = __m256i; }; template <> struct vector_s<CPUType::AVX2, int16_t> { using type = __m256i; }; template <> struct vector_s<CPUType::AVX2, int> { using type = __m256i; }; template <> struct vector_s<CPUType::AVX2, float> { using type = __m256; }; template <> struct vector_s<CPUType::AVX2, double> { using type = __m256d; }; +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW template <> struct vector_s<CPUType::AVX512BW, int8_t> { using type = __m512i; }; template <> struct vector_s<CPUType::AVX512BW, int16_t> { using type = __m512i; }; |