diff options
Diffstat (limited to 'interleave.h')
-rw-r--r-- | interleave.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/interleave.h b/interleave.h index 9bd4fdd..231be46 100644 --- a/interleave.h +++ b/interleave.h @@ -28,7 +28,7 @@ INTGEMM_INTERLEAVE_N(target, type, 64) INTGEMM_INTERLEAVE(INTGEMM_SSE2, __m128i) INTGEMM_INTERLEAVE(INTGEMM_AVX2, __m256i) -#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512 +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW INTGEMM_INTERLEAVE(INTGEMM_AVX512BW, __m512i) #endif @@ -44,7 +44,7 @@ target static inline void Swap(Register &a, Register &b) { \ INTGEMM_SWAP(INTGEMM_SSE2, __m128i) INTGEMM_SWAP(INTGEMM_AVX2, __m256i) -#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512 +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW /* Only INTGEMM_AVX512F is necessary but due to GCC 5.4 bug we have to set INTGEMM_AVX512BW */ INTGEMM_SWAP(INTGEMM_AVX512BW, __m512i) #endif @@ -97,7 +97,7 @@ target static inline void Transpose16InLane(Register &r0, Register &r1, Register INTGEMM_TRANSPOSE16(INTGEMM_SSE2, __m128i) INTGEMM_TRANSPOSE16(INTGEMM_AVX2, __m256i) -#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512 +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW /* Only INTGEMM_AVX512F is necessary but due to GCC 5.4 bug we have to set INTGEMM_AVX512BW */ INTGEMM_TRANSPOSE16(INTGEMM_AVX512BW, __m512i) #endif |