diff options
author | Kenneth Heafield <github@kheafield.com> | 2020-03-08 22:22:24 +0300 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2020-03-08 22:22:24 +0300 |
commit | 261a5fbcf7558fc3c2ac22b33fe0c2930d440fc3 (patch) | |
tree | f912e6a3cec9691e509d3e63eb3672ae6a357d2c /interleave.h | |
parent | 1323f245adff16b351b61310625b743b22e90059 (diff) |
Change to INTGEMM_COMPILER_SUPPORTS_AVX512BW and update test
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 |