From 4d2d625432e8fdaaaa33042f31ceb6071eef6809 Mon Sep 17 00:00:00 2001 From: nihuini Date: Wed, 22 Jul 2020 16:44:39 +0800 Subject: [PATCH] fix avx2 build, second try, fix #1953 --- src/layer/x86/cast_x86.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/layer/x86/cast_x86.cpp b/src/layer/x86/cast_x86.cpp index 7b296052f..64a2b56db 100644 --- a/src/layer/x86/cast_x86.cpp +++ b/src/layer/x86/cast_x86.cpp @@ -22,6 +22,17 @@ #endif // __AVX__ #if __AVX__ +typedef union m128i +{ + __m128i vec; + uint16_t m128i_u16[8]; +} m128; + +typedef union m256i +{ + __m256i vec; + uint32_t m256i_u32[8]; +} m256; static inline __m256 bfloat2float_avx(__m128i v0) { __m128i zero = _mm_set1_epi32(0);