2 * Copyright (c) 2015 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
7 * http://www.apache.org/licenses/LICENSE-2.0
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
16 #ifndef included_vector_avx512_h
17 #define included_vector_avx512_h
19 #include <vppinfra/clib.h>
20 #include <x86intrin.h>
23 #define foreach_avx512_vec512i \
24 _(i,8,64,epi8) _(i,16,32,epi16) _(i,32,16,epi32) _(i,64,8,epi64)
25 #define foreach_avx512_vec512u \
26 _(u,8,64,epi8) _(u,16,32,epi16) _(u,32,16,epi32) _(u,64,8,epi64)
27 #define foreach_avx512_vec512f \
28 _(f,32,8,ps) _(f,64,4,pd)
30 /* splat, load_unaligned, store_unaligned, is_all_zero, is_equal,
31 is_all_equal, is_zero_mask */
32 #define _(t, s, c, i) \
33 static_always_inline t##s##x##c \
34 t##s##x##c##_splat (t##s x) \
35 { return (t##s##x##c) _mm512_set1_##i (x); } \
37 static_always_inline t##s##x##c \
38 t##s##x##c##_load_aligned (void *p) \
39 { return (t##s##x##c) _mm512_load_si512 (p); } \
41 static_always_inline void \
42 t##s##x##c##_store_aligned (t##s##x##c v, void *p) \
43 { _mm512_store_si512 ((__m512i *) p, (__m512i) v); } \
45 static_always_inline t##s##x##c \
46 t##s##x##c##_load_unaligned (void *p) \
47 { return (t##s##x##c) _mm512_loadu_si512 (p); } \
49 static_always_inline void \
50 t##s##x##c##_store_unaligned (t##s##x##c v, void *p) \
51 { _mm512_storeu_si512 ((__m512i *) p, (__m512i) v); } \
53 static_always_inline int \
54 t##s##x##c##_is_all_zero (t##s##x##c v) \
55 { return (_mm512_test_epi64_mask ((__m512i) v, (__m512i) v) == 0); } \
57 static_always_inline int \
58 t##s##x##c##_is_equal (t##s##x##c a, t##s##x##c b) \
59 { return t##s##x##c##_is_all_zero (a ^ b); } \
61 static_always_inline int \
62 t##s##x##c##_is_all_equal (t##s##x##c v, t##s x) \
63 { return t##s##x##c##_is_equal (v, t##s##x##c##_splat (x)); } \
65 static_always_inline u##c \
66 t##s##x##c##_is_zero_mask (t##s##x##c v) \
67 { return _mm512_test_##i##_mask ((__m512i) v, (__m512i) v); } \
69 static_always_inline t##s##x##c \
70 t##s##x##c##_interleave_lo (t##s##x##c a, t##s##x##c b) \
71 { return (t##s##x##c) _mm512_unpacklo_##i ((__m512i) a, (__m512i) b); } \
73 static_always_inline t##s##x##c \
74 t##s##x##c##_interleave_hi (t##s##x##c a, t##s##x##c b) \
75 { return (t##s##x##c) _mm512_unpackhi_##i ((__m512i) a, (__m512i) b); } \
78 foreach_avx512_vec512i foreach_avx512_vec512u
82 static_always_inline u32
83 u16x32_msb_mask (u16x32 v)
85 return (u32) _mm512_movepi16_mask ((__m512i) v);
90 always_inline t t##_pack (f lo, f hi) \
92 return (t) fn ((__m512i) lo, (__m512i) hi); \
95 _ (i16x32, i8x64, _mm512_packs_epi16)
96 _ (i16x32, u8x64, _mm512_packus_epi16)
97 _ (i32x16, i16x32, _mm512_packs_epi32)
98 _ (i32x16, u16x32, _mm512_packus_epi32)
101 static_always_inline u32x16
102 u32x16_byte_swap (u32x16 v)
105 3, 2, 1, 0, 7, 6, 5, 4, 11, 10, 9, 8, 15, 14, 13, 12,
106 3, 2, 1, 0, 7, 6, 5, 4, 11, 10, 9, 8, 15, 14, 13, 12,
107 3, 2, 1, 0, 7, 6, 5, 4, 11, 10, 9, 8, 15, 14, 13, 12,
108 3, 2, 1, 0, 7, 6, 5, 4, 11, 10, 9, 8, 15, 14, 13, 12
110 return (u32x16) _mm512_shuffle_epi8 ((__m512i) v, (__m512i) swap);
113 static_always_inline u16x32
114 u16x32_byte_swap (u16x32 v)
117 1, 0, 3, 2, 5, 4, 7, 6, 9, 8, 11, 10, 13, 12, 15, 14,
118 1, 0, 3, 2, 5, 4, 7, 6, 9, 8, 11, 10, 13, 12, 15, 14,
119 1, 0, 3, 2, 5, 4, 7, 6, 9, 8, 11, 10, 13, 12, 15, 14,
120 1, 0, 3, 2, 5, 4, 7, 6, 9, 8, 11, 10, 13, 12, 15, 14
122 return (u16x32) _mm512_shuffle_epi8 ((__m512i) v, (__m512i) swap);
126 static_always_inline t f##_extract_lo (f v) \
128 return (t) _mm512_extracti64x4_epi64 ((__m512i) v, 0); \
130 static_always_inline t f##_extract_hi (f v) \
132 return (t) _mm512_extracti64x4_epi64 ((__m512i) v, 1); \
141 static_always_inline u32
142 u32x16_min_scalar (u32x16 v)
144 return u32x8_min_scalar (u32x8_min (u32x16_extract_lo (v),
145 u32x16_extract_hi (v)));
148 static_always_inline u32x16
149 u32x16_insert_lo (u32x16 r, u32x8 v)
151 return (u32x16) _mm512_inserti64x4 ((__m512i) r, (__m256i) v, 0);
154 static_always_inline u32x16
155 u32x16_insert_hi (u32x16 r, u32x8 v)
157 return (u32x16) _mm512_inserti64x4 ((__m512i) r, (__m256i) v, 1);
160 static_always_inline u64x8
161 u64x8_permute (u64x8 a, u64x8 b, u64x8 mask)
163 return (u64x8) _mm512_permutex2var_epi64 ((__m512i) a, (__m512i) mask,
168 #define u32x16_ternary_logic(a, b, c, d) \
169 (u32x16) _mm512_ternarylogic_epi32 ((__m512i) a, (__m512i) b, (__m512i) c, d)
171 #define u8x64_insert_u8x16(a, b, n) \
172 (u8x64) _mm512_inserti64x2 ((__m512i) (a), (__m128i) (b), n)
174 #define u8x64_extract_u8x16(a, n) \
175 (u8x16) _mm512_extracti64x2_epi64 ((__m512i) (a), n)
177 #define u8x64_word_shift_left(a,n) (u8x64) _mm512_bslli_epi128((__m512i) a, n)
178 #define u8x64_word_shift_right(a,n) (u8x64) _mm512_bsrli_epi128((__m512i) a, n)
180 static_always_inline u8x64
181 u8x64_xor3 (u8x64 a, u8x64 b, u8x64 c)
183 return (u8x64) _mm512_ternarylogic_epi32 ((__m512i) a, (__m512i) b,
187 static_always_inline u8x64
188 u8x64_reflect_u8x16 (u8x64 x)
190 static const u8x64 mask = {
191 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0,
192 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0,
193 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0,
194 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0,
196 return (u8x64) _mm512_shuffle_epi8 ((__m512i) x, (__m512i) mask);
199 static_always_inline u8x64
200 u8x64_shuffle (u8x64 v, u8x64 m)
202 return (u8x64) _mm512_shuffle_epi8 ((__m512i) v, (__m512i) m);
205 #define u8x64_align_right(a, b, imm) \
206 (u8x64) _mm512_alignr_epi8 ((__m512i) a, (__m512i) b, imm)
208 #define u64x8_align_right(a, b, imm) \
209 (u64x8) _mm512_alignr_epi64 ((__m512i) a, (__m512i) b, imm)
211 static_always_inline u32
212 u32x16_sum_elts (u32x16 sum16)
215 sum16 += (u32x16) u8x64_align_right (sum16, sum16, 8);
216 sum16 += (u32x16) u8x64_align_right (sum16, sum16, 4);
217 sum8 = u32x16_extract_hi (sum16) + u32x16_extract_lo (sum16);
218 return sum8[0] + sum8[4];
221 #define _(t, m, p, i, e) \
222 static_always_inline t t##_mask_load (t a, void *p, m mask) \
224 return (t) p##_mask_loadu_##e ((i) a, mask, p); \
226 static_always_inline t t##_mask_load_zero (void *p, m mask) \
228 return (t) p##_maskz_loadu_##e (mask, p); \
230 static_always_inline void t##_mask_store (t a, void *p, m mask) \
232 p##_mask_storeu_##e (p, mask, (i) a); \
235 _ (u8x64, u64, _mm512, __m512i, epi8)
236 _ (u8x32, u32, _mm256, __m256i, epi8)
237 _ (u8x16, u16, _mm, __m128i, epi8)
238 _ (u16x32, u32, _mm512, __m512i, epi16)
239 _ (u16x16, u16, _mm256, __m256i, epi16)
240 _ (u16x8, u8, _mm, __m128i, epi16)
241 _ (u32x16, u16, _mm512, __m512i, epi32)
242 _ (u32x8, u8, _mm256, __m256i, epi32)
243 _ (u32x4, u8, _mm, __m128i, epi32)
244 _ (u64x8, u8, _mm512, __m512i, epi64)
245 _ (u64x4, u8, _mm256, __m256i, epi64)
246 _ (u64x2, u8, _mm, __m128i, epi64)
249 #ifdef CLIB_HAVE_VEC512
250 #define CLIB_HAVE_VEC512_MASK_LOAD_STORE
252 #ifdef CLIB_HAVE_VEC256
253 #define CLIB_HAVE_VEC256_MASK_LOAD_STORE
255 #ifdef CLIB_HAVE_VEC128
256 #define CLIB_HAVE_VEC128_MASK_LOAD_STORE
259 static_always_inline u8x64
260 u8x64_splat_u8x16 (u8x16 a)
262 return (u8x64) _mm512_broadcast_i64x2 ((__m128i) a);
265 static_always_inline u32x16
266 u32x16_splat_u32x4 (u32x4 a)
268 return (u32x16) _mm512_broadcast_i64x2 ((__m128i) a);
271 static_always_inline u32x16
272 u32x16_mask_blend (u32x16 a, u32x16 b, u16 mask)
274 return (u32x16) _mm512_mask_blend_epi32 (mask, (__m512i) a, (__m512i) b);
277 static_always_inline u8x64
278 u8x64_mask_blend (u8x64 a, u8x64 b, u64 mask)
280 return (u8x64) _mm512_mask_blend_epi8 (mask, (__m512i) a, (__m512i) b);
283 #define _(t, m, e, p, it) \
284 static_always_inline m t##_is_equal_mask (t a, t b) \
286 return p##_cmpeq_##e##_mask ((it) a, (it) b); \
288 _ (u8x16, u16, epu8, _mm, __m128i)
289 _ (u16x8, u8, epu16, _mm, __m128i)
290 _ (u32x4, u8, epu32, _mm, __m128i)
291 _ (u64x2, u8, epu64, _mm, __m128i)
293 _ (u8x32, u32, epu8, _mm256, __m256i)
294 _ (u16x16, u16, epu16, _mm256, __m256i)
295 _ (u32x8, u8, epu32, _mm256, __m256i)
296 _ (u64x4, u8, epu64, _mm256, __m256i)
298 _ (u8x64, u64, epu8, _mm512, __m512i)
299 _ (u16x32, u32, epu16, _mm512, __m512i)
300 _ (u32x16, u16, epu32, _mm512, __m512i)
301 _ (u64x8, u8, epu64, _mm512, __m512i)
304 #define _(f, t, fn, it) \
305 static_always_inline t t##_from_##f (f x) { return (t) fn ((it) x); }
306 _ (u16x16, u32x16, _mm512_cvtepi16_epi32, __m256i)
307 _ (u32x16, u16x16, _mm512_cvtusepi32_epi16, __m512i)
308 _ (u32x8, u16x8, _mm256_cvtusepi32_epi16, __m256i)
309 _ (u32x8, u64x8, _mm512_cvtepu32_epi64, __m256i)
312 #define _(vt, mt, p, it, epi) \
313 static_always_inline vt vt##_compress (vt a, mt mask) \
315 return (vt) p##_maskz_compress_##epi (mask, (it) a); \
317 static_always_inline vt vt##_expand (vt a, mt mask) \
319 return (vt) p##_maskz_expand_##epi (mask, (it) a); \
321 static_always_inline void vt##_compress_store (vt v, mt mask, void *p) \
323 p##_mask_compressstoreu_##epi (p, mask, (it) v); \
326 _ (u64x8, u8, _mm512, __m512i, epi64)
327 _ (u32x16, u16, _mm512, __m512i, epi32)
328 _ (u64x4, u8, _mm256, __m256i, epi64)
329 _ (u32x8, u8, _mm256, __m256i, epi32)
330 _ (u64x2, u8, _mm, __m128i, epi64)
331 _ (u32x4, u8, _mm, __m128i, epi32)
332 #ifdef __AVX512VBMI2__
333 _ (u16x32, u32, _mm512, __m512i, epi16)
334 _ (u8x64, u64, _mm512, __m512i, epi8)
335 _ (u16x16, u16, _mm256, __m256i, epi16)
336 _ (u8x32, u32, _mm256, __m256i, epi8)
337 _ (u16x8, u8, _mm, __m128i, epi16)
338 _ (u8x16, u16, _mm, __m128i, epi8)
342 #ifdef CLIB_HAVE_VEC256
343 #define CLIB_HAVE_VEC256_COMPRESS
344 #ifdef __AVX512VBMI2__
345 #define CLIB_HAVE_VEC256_COMPRESS_U8_U16
349 #ifdef CLIB_HAVE_VEC512
350 #define CLIB_HAVE_VEC512_COMPRESS
351 #ifdef __AVX512VBMI2__
352 #define CLIB_HAVE_VEC512_COMPRESS_U8_U16
357 #ifndef __AVX512VBMI2__
358 static_always_inline u16x16
359 u16x16_compress (u16x16 v, u16 mask)
361 return u16x16_from_u32x16 (u32x16_compress (u32x16_from_u16x16 (v), mask));
364 static_always_inline u16x8
365 u16x8_compress (u16x8 v, u8 mask)
367 return u16x8_from_u32x8 (u32x8_compress (u32x8_from_u16x8 (v), mask));
371 static_always_inline void
372 u32x16_transpose (u32x16 m[16])
374 __m512i r[16], a, b, c, d, x, y;
377 __m512i pm1 = (__m512i) (u64x8) { 0, 1, 8, 9, 4, 5, 12, 13};
378 __m512i pm2 = (__m512i) (u64x8) { 2, 3, 10, 11, 6, 7, 14, 15};
379 __m512i pm3 = (__m512i) (u64x8) { 0, 1, 2, 3, 8, 9, 10, 11};
380 __m512i pm4 = (__m512i) (u64x8) { 4, 5, 6, 7, 12, 13, 14, 15};
383 r[0] = _mm512_unpacklo_epi32 ((__m512i) m[0], (__m512i) m[1]);
384 r[1] = _mm512_unpacklo_epi32 ((__m512i) m[2], (__m512i) m[3]);
385 r[2] = _mm512_unpacklo_epi32 ((__m512i) m[4], (__m512i) m[5]);
386 r[3] = _mm512_unpacklo_epi32 ((__m512i) m[6], (__m512i) m[7]);
387 r[4] = _mm512_unpacklo_epi32 ((__m512i) m[8], (__m512i) m[9]);
388 r[5] = _mm512_unpacklo_epi32 ((__m512i) m[10], (__m512i) m[11]);
389 r[6] = _mm512_unpacklo_epi32 ((__m512i) m[12], (__m512i) m[13]);
390 r[7] = _mm512_unpacklo_epi32 ((__m512i) m[14], (__m512i) m[15]);
392 r[8] = _mm512_unpackhi_epi32 ((__m512i) m[0], (__m512i) m[1]);
393 r[9] = _mm512_unpackhi_epi32 ((__m512i) m[2], (__m512i) m[3]);
394 r[10] = _mm512_unpackhi_epi32 ((__m512i) m[4], (__m512i) m[5]);
395 r[11] = _mm512_unpackhi_epi32 ((__m512i) m[6], (__m512i) m[7]);
396 r[12] = _mm512_unpackhi_epi32 ((__m512i) m[8], (__m512i) m[9]);
397 r[13] = _mm512_unpackhi_epi32 ((__m512i) m[10], (__m512i) m[11]);
398 r[14] = _mm512_unpackhi_epi32 ((__m512i) m[12], (__m512i) m[13]);
399 r[15] = _mm512_unpackhi_epi32 ((__m512i) m[14], (__m512i) m[15]);
401 a = _mm512_unpacklo_epi64 (r[0], r[1]);
402 b = _mm512_unpacklo_epi64 (r[2], r[3]);
403 c = _mm512_unpacklo_epi64 (r[4], r[5]);
404 d = _mm512_unpacklo_epi64 (r[6], r[7]);
405 x = _mm512_permutex2var_epi64 (a, pm1, b);
406 y = _mm512_permutex2var_epi64 (c, pm1, d);
407 m[0] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
408 m[8] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
409 x = _mm512_permutex2var_epi64 (a, pm2, b);
410 y = _mm512_permutex2var_epi64 (c, pm2, d);
411 m[4] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
412 m[12] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
414 a = _mm512_unpacklo_epi64 (r[8], r[9]);
415 b = _mm512_unpacklo_epi64 (r[10], r[11]);
416 c = _mm512_unpacklo_epi64 (r[12], r[13]);
417 d = _mm512_unpacklo_epi64 (r[14], r[15]);
418 x = _mm512_permutex2var_epi64 (a, pm1, b);
419 y = _mm512_permutex2var_epi64 (c, pm1, d);
420 m[2] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
421 m[10] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
422 x = _mm512_permutex2var_epi64 (a, pm2, b);
423 y = _mm512_permutex2var_epi64 (c, pm2, d);
424 m[6] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
425 m[14] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
427 a = _mm512_unpackhi_epi64 (r[0], r[1]);
428 b = _mm512_unpackhi_epi64 (r[2], r[3]);
429 c = _mm512_unpackhi_epi64 (r[4], r[5]);
430 d = _mm512_unpackhi_epi64 (r[6], r[7]);
431 x = _mm512_permutex2var_epi64 (a, pm1, b);
432 y = _mm512_permutex2var_epi64 (c, pm1, d);
433 m[1] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
434 m[9] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
435 x = _mm512_permutex2var_epi64 (a, pm2, b);
436 y = _mm512_permutex2var_epi64 (c, pm2, d);
437 m[5] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
438 m[13] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
440 a = _mm512_unpackhi_epi64 (r[8], r[9]);
441 b = _mm512_unpackhi_epi64 (r[10], r[11]);
442 c = _mm512_unpackhi_epi64 (r[12], r[13]);
443 d = _mm512_unpackhi_epi64 (r[14], r[15]);
444 x = _mm512_permutex2var_epi64 (a, pm1, b);
445 y = _mm512_permutex2var_epi64 (c, pm1, d);
446 m[3] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
447 m[11] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
448 x = _mm512_permutex2var_epi64 (a, pm2, b);
449 y = _mm512_permutex2var_epi64 (c, pm2, d);
450 m[7] = (u32x16) _mm512_permutex2var_epi64 (x, pm3, y);
451 m[15] = (u32x16) _mm512_permutex2var_epi64 (x, pm4, y);
456 static_always_inline void
457 u64x8_transpose (u64x8 m[8])
462 __m512i pm1 = (__m512i) (u64x8) { 0, 1, 8, 9, 4, 5, 12, 13};
463 __m512i pm2 = (__m512i) (u64x8) { 2, 3, 10, 11, 6, 7, 14, 15};
464 __m512i pm3 = (__m512i) (u64x8) { 0, 1, 2, 3, 8, 9, 10, 11};
465 __m512i pm4 = (__m512i) (u64x8) { 4, 5, 6, 7, 12, 13, 14, 15};
468 r[0] = _mm512_unpacklo_epi64 ((__m512i) m[0], (__m512i) m[1]);
469 r[1] = _mm512_unpacklo_epi64 ((__m512i) m[2], (__m512i) m[3]);
470 r[2] = _mm512_unpacklo_epi64 ((__m512i) m[4], (__m512i) m[5]);
471 r[3] = _mm512_unpacklo_epi64 ((__m512i) m[6], (__m512i) m[7]);
472 r[4] = _mm512_unpackhi_epi64 ((__m512i) m[0], (__m512i) m[1]);
473 r[5] = _mm512_unpackhi_epi64 ((__m512i) m[2], (__m512i) m[3]);
474 r[6] = _mm512_unpackhi_epi64 ((__m512i) m[4], (__m512i) m[5]);
475 r[7] = _mm512_unpackhi_epi64 ((__m512i) m[6], (__m512i) m[7]);
477 x = _mm512_permutex2var_epi64 (r[0], pm1, r[1]);
478 y = _mm512_permutex2var_epi64 (r[2], pm1, r[3]);
479 m[0] = (u64x8) _mm512_permutex2var_epi64 (x, pm3, y);
480 m[4] = (u64x8) _mm512_permutex2var_epi64 (x, pm4, y);
481 x = _mm512_permutex2var_epi64 (r[0], pm2, r[1]);
482 y = _mm512_permutex2var_epi64 (r[2], pm2, r[3]);
483 m[2] = (u64x8) _mm512_permutex2var_epi64 (x, pm3, y);
484 m[6] = (u64x8) _mm512_permutex2var_epi64 (x, pm4, y);
486 x = _mm512_permutex2var_epi64 (r[4], pm1, r[5]);
487 y = _mm512_permutex2var_epi64 (r[6], pm1, r[7]);
488 m[1] = (u64x8) _mm512_permutex2var_epi64 (x, pm3, y);
489 m[5] = (u64x8) _mm512_permutex2var_epi64 (x, pm4, y);
490 x = _mm512_permutex2var_epi64 (r[4], pm2, r[5]);
491 y = _mm512_permutex2var_epi64 (r[6], pm2, r[7]);
492 m[3] = (u64x8) _mm512_permutex2var_epi64 (x, pm3, y);
493 m[7] = (u64x8) _mm512_permutex2var_epi64 (x, pm4, y);
496 #endif /* included_vector_avx512_h */
498 * fd.io coding-style-patch-verification: ON
501 * eval: (c-set-style "gnu")