blob: 1b558a0404aed444bee77e9cc036721e5bc985a5 [file] [log] [blame]
Adam Nemet9a3ea602014-07-28 17:14:38 +00001/*===---- avx512fintrin.h - AVX2 intrinsics --------------------------------===
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +00002 *
3 * Permission is hereby granted, free of charge, to any person obtaining a copy
4 * of this software and associated documentation files (the "Software"), to deal
5 * in the Software without restriction, including without limitation the rights
6 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7 * copies of the Software, and to permit persons to whom the Software is
8 * furnished to do so, subject to the following conditions:
9 *
10 * The above copyright notice and this permission notice shall be included in
11 * all copies or substantial portions of the Software.
12 *
13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19 * THE SOFTWARE.
20 *
21 *===-----------------------------------------------------------------------===
22 */
23#ifndef __IMMINTRIN_H
24#error "Never use <avx512fintrin.h> directly; include <immintrin.h> instead."
25#endif
26
27#ifndef __AVX512FINTRIN_H
28#define __AVX512FINTRIN_H
29
30typedef double __v8df __attribute__((__vector_size__(64)));
31typedef float __v16sf __attribute__((__vector_size__(64)));
32typedef long long __v8di __attribute__((__vector_size__(64)));
33typedef int __v16si __attribute__((__vector_size__(64)));
34
35typedef float __m512 __attribute__((__vector_size__(64)));
36typedef double __m512d __attribute__((__vector_size__(64)));
37typedef long long __m512i __attribute__((__vector_size__(64)));
38
39typedef unsigned char __mmask8;
40typedef unsigned short __mmask16;
41
42/* Rounding mode macros. */
43#define _MM_FROUND_TO_NEAREST_INT 0x00
44#define _MM_FROUND_TO_NEG_INF 0x01
45#define _MM_FROUND_TO_POS_INF 0x02
46#define _MM_FROUND_TO_ZERO 0x03
47#define _MM_FROUND_CUR_DIRECTION 0x04
48
Adam Nemet0d5bb552014-07-28 17:14:40 +000049/* Create vectors with repeated elements */
50
51static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
52_mm512_setzero_si512(void)
53{
54 return (__m512i)(__v8di){ 0, 0, 0, 0, 0, 0, 0, 0 };
55}
56
57static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
58_mm512_maskz_set1_epi32(__mmask16 __M, int __A)
59{
60 return (__m512i) __builtin_ia32_pbroadcastd512_gpr_mask (__A,
61 (__v16si)
62 _mm512_setzero_si512 (),
63 __M);
64}
65
66static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
67_mm512_maskz_set1_epi64(__mmask8 __M, long long __A)
68{
69#ifdef __x86_64__
70 return (__m512i) __builtin_ia32_pbroadcastq512_gpr_mask (__A,
71 (__v8di)
72 _mm512_setzero_si512 (),
73 __M);
74#else
75 return (__m512i) __builtin_ia32_pbroadcastq512_mem_mask (__A,
76 (__v8di)
77 _mm512_setzero_si512 (),
78 __M);
79#endif
80}
81
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +000082static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +000083_mm512_setzero_ps(void)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +000084{
85 return (__m512){ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
86 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 };
87}
88static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +000089_mm512_setzero_pd(void)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +000090{
91 return (__m512d){ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 };
92}
Adam Nemet0d5bb552014-07-28 17:14:40 +000093
Adam Nemetf42e7a22014-07-30 16:51:22 +000094static __inline __m512 __attribute__((__always_inline__, __nodebug__))
95_mm512_set1_ps(float __w)
96{
97 return (__m512){ __w, __w, __w, __w, __w, __w, __w, __w,
98 __w, __w, __w, __w, __w, __w, __w, __w };
99}
100
101static __inline __m512d __attribute__((__always_inline__, __nodebug__))
102_mm512_set1_pd(double __w)
103{
104 return (__m512d){ __w, __w, __w, __w, __w, __w, __w, __w };
105}
106
107static __inline __m512i __attribute__((__always_inline__, __nodebug__))
108_mm512_set1_epi32(int __s)
109{
110 return (__m512i)(__v16si){ __s, __s, __s, __s, __s, __s, __s, __s,
111 __s, __s, __s, __s, __s, __s, __s, __s };
112}
113
114static __inline __m512i __attribute__((__always_inline__, __nodebug__))
115_mm512_set1_epi64(long long __d)
116{
117 return (__m512i)(__v8di){ __d, __d, __d, __d, __d, __d, __d, __d };
118}
119
Adam Nemet4abc07c2014-08-13 00:29:01 +0000120static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
121_mm512_broadcastss_ps(__m128 __X)
122{
123 float __f = __X[0];
124 return (__v16sf){ __f, __f, __f, __f,
125 __f, __f, __f, __f,
126 __f, __f, __f, __f,
127 __f, __f, __f, __f };
128}
129
130static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
131_mm512_broadcastsd_pd(__m128d __X)
132{
133 double __d = __X[0];
134 return (__v8df){ __d, __d, __d, __d,
135 __d, __d, __d, __d };
136}
137
Adam Nemetc871ff92014-07-30 16:51:24 +0000138/* Cast between vector types */
139
140static __inline __m512d __attribute__((__always_inline__, __nodebug__))
141_mm512_castpd256_pd512(__m256d __a)
142{
143 return __builtin_shufflevector(__a, __a, 0, 1, 2, 3, -1, -1, -1, -1);
144}
145
146static __inline __m512 __attribute__((__always_inline__, __nodebug__))
147_mm512_castps256_ps512(__m256 __a)
148{
149 return __builtin_shufflevector(__a, __a, 0, 1, 2, 3, 4, 5, 6, 7,
150 -1, -1, -1, -1, -1, -1, -1, -1);
151}
152
153static __inline __m128d __attribute__((__always_inline__, __nodebug__))
154_mm512_castpd512_pd128(__m512d __a)
155{
156 return __builtin_shufflevector(__a, __a, 0, 1);
157}
158
159static __inline __m128 __attribute__((__always_inline__, __nodebug__))
160_mm512_castps512_ps128(__m512 __a)
161{
162 return __builtin_shufflevector(__a, __a, 0, 1, 2, 3);
163}
164
Adam Nemet0d5bb552014-07-28 17:14:40 +0000165/* Arithmetic */
166
Adam Nemeta3ebe622014-07-28 17:14:42 +0000167static __inline __m512d __attribute__((__always_inline__, __nodebug__))
168_mm512_add_pd(__m512d __a, __m512d __b)
169{
170 return __a + __b;
171}
172
173static __inline __m512 __attribute__((__always_inline__, __nodebug__))
174_mm512_add_ps(__m512 __a, __m512 __b)
175{
176 return __a + __b;
177}
178
179static __inline __m512d __attribute__((__always_inline__, __nodebug__))
180_mm512_mul_pd(__m512d __a, __m512d __b)
181{
182 return __a * __b;
183}
184
185static __inline __m512 __attribute__((__always_inline__, __nodebug__))
186_mm512_mul_ps(__m512 __a, __m512 __b)
187{
188 return __a * __b;
189}
190
191static __inline __m512d __attribute__((__always_inline__, __nodebug__))
192_mm512_sub_pd(__m512d __a, __m512d __b)
193{
194 return __a - __b;
195}
196
197static __inline __m512 __attribute__((__always_inline__, __nodebug__))
198_mm512_sub_ps(__m512 __a, __m512 __b)
199{
200 return __a - __b;
201}
202
Adam Nemet0d5bb552014-07-28 17:14:40 +0000203static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
204_mm512_max_pd(__m512d __A, __m512d __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000205{
Adam Nemet0d5bb552014-07-28 17:14:40 +0000206 return (__m512d) __builtin_ia32_maxpd512_mask ((__v8df) __A,
207 (__v8df) __B,
208 (__v8df)
209 _mm512_setzero_pd (),
210 (__mmask8) -1,
211 _MM_FROUND_CUR_DIRECTION);
212}
213
214static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
215_mm512_max_ps(__m512 __A, __m512 __B)
216{
217 return (__m512) __builtin_ia32_maxps512_mask ((__v16sf) __A,
218 (__v16sf) __B,
219 (__v16sf)
220 _mm512_setzero_ps (),
221 (__mmask16) -1,
222 _MM_FROUND_CUR_DIRECTION);
223}
224
225static __inline __m512i
226__attribute__ ((__always_inline__, __nodebug__))
227_mm512_max_epi32(__m512i __A, __m512i __B)
228{
229 return (__m512i) __builtin_ia32_pmaxsd512_mask ((__v16si) __A,
230 (__v16si) __B,
231 (__v16si)
232 _mm512_setzero_si512 (),
233 (__mmask16) -1);
234}
235
236static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
237_mm512_max_epu32(__m512i __A, __m512i __B)
238{
239 return (__m512i) __builtin_ia32_pmaxud512_mask ((__v16si) __A,
240 (__v16si) __B,
241 (__v16si)
242 _mm512_setzero_si512 (),
243 (__mmask16) -1);
244}
245
246static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
247_mm512_max_epi64(__m512i __A, __m512i __B)
248{
249 return (__m512i) __builtin_ia32_pmaxsq512_mask ((__v8di) __A,
250 (__v8di) __B,
251 (__v8di)
252 _mm512_setzero_si512 (),
253 (__mmask8) -1);
254}
255
256static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
257_mm512_max_epu64(__m512i __A, __m512i __B)
258{
259 return (__m512i) __builtin_ia32_pmaxuq512_mask ((__v8di) __A,
260 (__v8di) __B,
261 (__v8di)
262 _mm512_setzero_si512 (),
263 (__mmask8) -1);
264}
265
266static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
267_mm512_min_pd(__m512d __A, __m512d __B)
268{
269 return (__m512d) __builtin_ia32_minpd512_mask ((__v8df) __A,
270 (__v8df) __B,
271 (__v8df)
272 _mm512_setzero_pd (),
273 (__mmask8) -1,
274 _MM_FROUND_CUR_DIRECTION);
275}
276
277static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
278_mm512_min_ps(__m512 __A, __m512 __B)
279{
280 return (__m512) __builtin_ia32_minps512_mask ((__v16sf) __A,
281 (__v16sf) __B,
282 (__v16sf)
283 _mm512_setzero_ps (),
284 (__mmask16) -1,
285 _MM_FROUND_CUR_DIRECTION);
286}
287
288static __inline __m512i
289__attribute__ ((__always_inline__, __nodebug__))
290_mm512_min_epi32(__m512i __A, __m512i __B)
291{
292 return (__m512i) __builtin_ia32_pminsd512_mask ((__v16si) __A,
293 (__v16si) __B,
294 (__v16si)
295 _mm512_setzero_si512 (),
296 (__mmask16) -1);
297}
298
299static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
300_mm512_min_epu32(__m512i __A, __m512i __B)
301{
302 return (__m512i) __builtin_ia32_pminud512_mask ((__v16si) __A,
303 (__v16si) __B,
304 (__v16si)
305 _mm512_setzero_si512 (),
306 (__mmask16) -1);
307}
308
309static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
310_mm512_min_epi64(__m512i __A, __m512i __B)
311{
312 return (__m512i) __builtin_ia32_pminsq512_mask ((__v8di) __A,
313 (__v8di) __B,
314 (__v8di)
315 _mm512_setzero_si512 (),
316 (__mmask8) -1);
317}
318
319static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
320_mm512_min_epu64(__m512i __A, __m512i __B)
321{
322 return (__m512i) __builtin_ia32_pminuq512_mask ((__v8di) __A,
323 (__v8di) __B,
324 (__v8di)
325 _mm512_setzero_si512 (),
326 (__mmask8) -1);
327}
328
329static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
330_mm512_mul_epi32(__m512i __X, __m512i __Y)
331{
332 return (__m512i) __builtin_ia32_pmuldq512_mask ((__v16si) __X,
333 (__v16si) __Y,
334 (__v8di)
335 _mm512_setzero_si512 (),
336 (__mmask8) -1);
337}
338
339static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
340_mm512_mul_epu32(__m512i __X, __m512i __Y)
341{
342 return (__m512i) __builtin_ia32_pmuludq512_mask ((__v16si) __X,
343 (__v16si) __Y,
344 (__v8di)
345 _mm512_setzero_si512 (),
346 (__mmask8) -1);
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000347}
348
349static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
350_mm512_sqrt_pd(__m512d a)
351{
352 return (__m512d)__builtin_ia32_sqrtpd512_mask((__v8df)a,
353 (__v8df) _mm512_setzero_pd (),
354 (__mmask8) -1,
355 _MM_FROUND_CUR_DIRECTION);
356}
357
358static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
359_mm512_sqrt_ps(__m512 a)
360{
361 return (__m512)__builtin_ia32_sqrtps512_mask((__v16sf)a,
362 (__v16sf) _mm512_setzero_ps (),
363 (__mmask16) -1,
364 _MM_FROUND_CUR_DIRECTION);
365}
366
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000367static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
368_mm512_rsqrt14_pd(__m512d __A)
369{
370 return (__m512d) __builtin_ia32_rsqrt14pd512_mask ((__v8df) __A,
371 (__v8df)
372 _mm512_setzero_pd (),
373 (__mmask8) -1);}
374
375static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
376_mm512_rsqrt14_ps(__m512 __A)
377{
378 return (__m512) __builtin_ia32_rsqrt14ps512_mask ((__v16sf) __A,
379 (__v16sf)
380 _mm512_setzero_ps (),
381 (__mmask16) -1);
382}
383
384static __inline__ __m128 __attribute__((__always_inline__, __nodebug__))
385_mm_rsqrt14_ss(__m128 __A, __m128 __B)
386{
387 return (__m128) __builtin_ia32_rsqrt14ss_mask ((__v4sf) __A,
388 (__v4sf) __B,
389 (__v4sf)
390 _mm_setzero_ps (),
391 (__mmask8) -1);
392}
393
394static __inline__ __m128d __attribute__((__always_inline__, __nodebug__))
395_mm_rsqrt14_sd(__m128d __A, __m128d __B)
396{
397 return (__m128d) __builtin_ia32_rsqrt14sd_mask ((__v2df) __A,
398 (__v2df) __B,
399 (__v2df)
400 _mm_setzero_pd (),
401 (__mmask8) -1);
402}
403
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000404static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
405_mm512_rcp14_pd(__m512d __A)
406{
407 return (__m512d) __builtin_ia32_rcp14pd512_mask ((__v8df) __A,
408 (__v8df)
409 _mm512_setzero_pd (),
410 (__mmask8) -1);
411}
412
413static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
414_mm512_rcp14_ps(__m512 __A)
415{
416 return (__m512) __builtin_ia32_rcp14ps512_mask ((__v16sf) __A,
417 (__v16sf)
418 _mm512_setzero_ps (),
419 (__mmask16) -1);
420}
421static __inline__ __m128 __attribute__((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000422_mm_rcp14_ss(__m128 __A, __m128 __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000423{
424 return (__m128) __builtin_ia32_rcp14ss_mask ((__v4sf) __A,
425 (__v4sf) __B,
426 (__v4sf)
427 _mm_setzero_ps (),
428 (__mmask8) -1);
429}
430
431static __inline__ __m128d __attribute__((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000432_mm_rcp14_sd(__m128d __A, __m128d __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000433{
434 return (__m128d) __builtin_ia32_rcp14sd_mask ((__v2df) __A,
435 (__v2df) __B,
436 (__v2df)
437 _mm_setzero_pd (),
438 (__mmask8) -1);
439}
440
Adam Nemet0d5bb552014-07-28 17:14:40 +0000441static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
442_mm512_floor_ps(__m512 __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000443{
Adam Nemet0d5bb552014-07-28 17:14:40 +0000444 return (__m512) __builtin_ia32_rndscaleps_mask ((__v16sf) __A,
445 _MM_FROUND_FLOOR,
446 (__v16sf) __A, -1,
447 _MM_FROUND_CUR_DIRECTION);
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000448}
449
Adam Nemet0d5bb552014-07-28 17:14:40 +0000450static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
451_mm512_floor_pd(__m512d __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000452{
Adam Nemet0d5bb552014-07-28 17:14:40 +0000453 return (__m512d) __builtin_ia32_rndscalepd_mask ((__v8df) __A,
454 _MM_FROUND_FLOOR,
455 (__v8df) __A, -1,
456 _MM_FROUND_CUR_DIRECTION);
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000457}
458
Adam Nemet0d5bb552014-07-28 17:14:40 +0000459static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
460_mm512_ceil_ps(__m512 __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000461{
Adam Nemet0d5bb552014-07-28 17:14:40 +0000462 return (__m512) __builtin_ia32_rndscaleps_mask ((__v16sf) __A,
463 _MM_FROUND_CEIL,
464 (__v16sf) __A, -1,
465 _MM_FROUND_CUR_DIRECTION);
466}
467
468static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
469_mm512_ceil_pd(__m512d __A)
470{
471 return (__m512d) __builtin_ia32_rndscalepd_mask ((__v8df) __A,
472 _MM_FROUND_CEIL,
473 (__v8df) __A, -1,
474 _MM_FROUND_CUR_DIRECTION);
475}
476
477static __inline __m512i __attribute__ (( __always_inline__, __nodebug__))
478_mm512_abs_epi64(__m512i __A)
479{
480 return (__m512i) __builtin_ia32_pabsq512_mask ((__v8di) __A,
481 (__v8di)
482 _mm512_setzero_si512 (),
483 (__mmask8) -1);
484}
485
486static __inline __m512i __attribute__ (( __always_inline__, __nodebug__))
487_mm512_abs_epi32(__m512i __A)
488{
489 return (__m512i) __builtin_ia32_pabsd512_mask ((__v16si) __A,
490 (__v16si)
491 _mm512_setzero_si512 (),
492 (__mmask16) -1);
493}
494
495static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
496_mm512_roundscale_ps(__m512 __A, const int __imm)
497{
498 return (__m512) __builtin_ia32_rndscaleps_mask ((__v16sf) __A, __imm,
499 (__v16sf) __A, -1,
500 _MM_FROUND_CUR_DIRECTION);
501}
502static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
503_mm512_roundscale_pd(__m512d __A, const int __imm)
504{
505 return (__m512d) __builtin_ia32_rndscalepd_mask ((__v8df) __A, __imm,
506 (__v8df) __A, -1,
507 _MM_FROUND_CUR_DIRECTION);
508}
509
510/* Vector permutations */
511
512static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
513_mm512_permutex2var_epi32(__m512i __A, __m512i __I, __m512i __B)
514{
515 return (__m512i) __builtin_ia32_vpermt2vard512_mask ((__v16si) __I
516 /* idx */ ,
517 (__v16si) __A,
518 (__v16si) __B,
519 (__mmask16) -1);
520}
521static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
522_mm512_permutex2var_epi64(__m512i __A, __m512i __I, __m512i __B)
523{
524 return (__m512i) __builtin_ia32_vpermt2varq512_mask ((__v8di) __I
525 /* idx */ ,
526 (__v8di) __A,
527 (__v8di) __B,
528 (__mmask8) -1);
529}
530
531static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
532_mm512_permutex2var_pd(__m512d __A, __m512i __I, __m512d __B)
533{
534 return (__m512d) __builtin_ia32_vpermt2varpd512_mask ((__v8di) __I
535 /* idx */ ,
536 (__v8df) __A,
537 (__v8df) __B,
538 (__mmask8) -1);
539}
540static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
541_mm512_permutex2var_ps(__m512 __A, __m512i __I, __m512 __B)
542{
543 return (__m512) __builtin_ia32_vpermt2varps512_mask ((__v16si) __I
544 /* idx */ ,
545 (__v16sf) __A,
546 (__v16sf) __B,
547 (__mmask16) -1);
548}
549
Adam Nemet5bf7baa2014-08-05 17:28:23 +0000550static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
551_mm512_valign_epi64(__m512i __A, __m512i __B, const int __I)
552{
553 return (__m512i) __builtin_ia32_alignq512_mask((__v8di)__A,
554 (__v8di)__B,
555 __I,
556 (__v8di)_mm512_setzero_si512(),
557 (__mmask8) -1);
558}
559
560static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
561_mm512_valign_epi32(__m512i __A, __m512i __B, const int __I)
562{
563 return (__m512i)__builtin_ia32_alignd512_mask((__v16si)__A,
564 (__v16si)__B,
565 __I,
566 (__v16si)_mm512_setzero_si512(),
567 (__mmask16) -1);
568}
569
Adam Nemet0d5bb552014-07-28 17:14:40 +0000570/* Vector Blend */
571
572static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
573_mm512_mask_blend_pd(__mmask8 __U, __m512d __A, __m512d __W)
574{
575 return (__m512d) __builtin_ia32_blendmpd_512_mask ((__v8df) __A,
576 (__v8df) __W,
577 (__mmask8) __U);
578}
579
580static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
581_mm512_mask_blend_ps(__mmask16 __U, __m512 __A, __m512 __W)
582{
583 return (__m512) __builtin_ia32_blendmps_512_mask ((__v16sf) __A,
584 (__v16sf) __W,
585 (__mmask16) __U);
586}
587
588static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
589_mm512_mask_blend_epi64(__mmask8 __U, __m512i __A, __m512i __W)
590{
591 return (__m512i) __builtin_ia32_blendmq_512_mask ((__v8di) __A,
592 (__v8di) __W,
593 (__mmask8) __U);
594}
595
596static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
597_mm512_mask_blend_epi32(__mmask16 __U, __m512i __A, __m512i __W)
598{
599 return (__m512i) __builtin_ia32_blendmd_512_mask ((__v16si) __A,
600 (__v16si) __W,
601 (__mmask16) __U);
602}
603
604/* Compare */
605
606static __inline __mmask16 __attribute__ ((__always_inline__, __nodebug__))
607_mm512_cmp_ps_mask(__m512 a, __m512 b, const int p)
608{
609 return (__mmask16) __builtin_ia32_cmpps512_mask ((__v16sf) a,
610 (__v16sf) b, p, (__mmask16) -1,
611 _MM_FROUND_CUR_DIRECTION);
612}
613
614static __inline __mmask8 __attribute__ ((__always_inline__, __nodebug__))
615_mm512_cmp_pd_mask(__m512d __X, __m512d __Y, const int __P)
616{
617 return (__mmask8) __builtin_ia32_cmppd512_mask ((__v8df) __X,
618 (__v8df) __Y, __P,
619 (__mmask8) -1,
620 _MM_FROUND_CUR_DIRECTION);
621}
622
623/* Conversion */
624
625static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
626_mm512_cvttps_epu32(__m512 __A)
627{
628 return (__m512i) __builtin_ia32_cvttps2udq512_mask ((__v16sf) __A,
629 (__v16si)
630 _mm512_setzero_si512 (),
631 (__mmask16) -1,
632 _MM_FROUND_CUR_DIRECTION);
633}
634
635static __inline __m512 __attribute__ (( __always_inline__, __nodebug__))
636_mm512_cvt_roundepi32_ps(__m512i __A, const int __R)
637{
638 return (__m512) __builtin_ia32_cvtdq2ps512_mask ((__v16si) __A,
639 (__v16sf)
640 _mm512_setzero_ps (),
641 (__mmask16) -1,
642 __R);
643}
644
645static __inline __m512 __attribute__ (( __always_inline__, __nodebug__))
646_mm512_cvt_roundepu32_ps(__m512i __A, const int __R)
647{
648 return (__m512) __builtin_ia32_cvtudq2ps512_mask ((__v16si) __A,
649 (__v16sf)
650 _mm512_setzero_ps (),
651 (__mmask16) -1,
652 __R);
653}
654
655static __inline __m512d __attribute__ (( __always_inline__, __nodebug__))
656_mm512_cvtepi32_pd(__m256i __A)
657{
658 return (__m512d) __builtin_ia32_cvtdq2pd512_mask ((__v8si) __A,
659 (__v8df)
660 _mm512_setzero_pd (),
661 (__mmask8) -1);
662}
663
664static __inline __m512d __attribute__ (( __always_inline__, __nodebug__))
665_mm512_cvtepu32_pd(__m256i __A)
666{
667 return (__m512d) __builtin_ia32_cvtudq2pd512_mask ((__v8si) __A,
668 (__v8df)
669 _mm512_setzero_pd (),
670 (__mmask8) -1);
671}
672static __inline __m256 __attribute__ (( __always_inline__, __nodebug__))
673_mm512_cvt_roundpd_ps(__m512d __A, const int __R)
674{
675 return (__m256) __builtin_ia32_cvtpd2ps512_mask ((__v8df) __A,
676 (__v8sf)
677 _mm256_setzero_ps (),
678 (__mmask8) -1,
679 __R);
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000680}
681
682static __inline __m256i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000683_mm512_cvtps_ph(__m512 __A, const int __I)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000684{
685 return (__m256i) __builtin_ia32_vcvtps2ph512_mask ((__v16sf) __A,
686 __I,
687 (__v16hi)
688 _mm256_setzero_si256 (),
689 -1);
690}
691
692static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000693_mm512_cvtph_ps(__m256i __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000694{
695 return (__m512) __builtin_ia32_vcvtph2ps512_mask ((__v16hi) __A,
696 (__v16sf)
697 _mm512_setzero_ps (),
698 (__mmask16) -1,
699 _MM_FROUND_CUR_DIRECTION);
700}
701
702static __inline __m512i __attribute__((__always_inline__, __nodebug__))
703_mm512_cvttps_epi32(__m512 a)
704{
705 return (__m512i)
706 __builtin_ia32_cvttps2dq512_mask((__v16sf) a,
707 (__v16si) _mm512_setzero_si512 (),
708 (__mmask16) -1, _MM_FROUND_CUR_DIRECTION);
709}
710
711static __inline __m256i __attribute__((__always_inline__, __nodebug__))
712_mm512_cvttpd_epi32(__m512d a)
713{
714 return (__m256i)__builtin_ia32_cvttpd2dq512_mask((__v8df) a,
715 (__v8si)_mm256_setzero_si256(),
716 (__mmask8) -1,
717 _MM_FROUND_CUR_DIRECTION);
718}
719
720static __inline __m256i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000721_mm512_cvtt_roundpd_epi32(__m512d __A, const int __R)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000722{
723 return (__m256i) __builtin_ia32_cvttpd2dq512_mask ((__v8df) __A,
724 (__v8si)
725 _mm256_setzero_si256 (),
726 (__mmask8) -1,
727 __R);
728}
729static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000730_mm512_cvtt_roundps_epi32(__m512 __A, const int __R)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000731{
732 return (__m512i) __builtin_ia32_cvttps2dq512_mask ((__v16sf) __A,
733 (__v16si)
734 _mm512_setzero_si512 (),
735 (__mmask16) -1,
736 __R);
737}
738
739static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000740_mm512_cvt_roundps_epi32(__m512 __A, const int __R)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000741{
742 return (__m512i) __builtin_ia32_cvtps2dq512_mask ((__v16sf) __A,
743 (__v16si)
744 _mm512_setzero_si512 (),
745 (__mmask16) -1,
746 __R);
747}
748static __inline __m256i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000749_mm512_cvt_roundpd_epi32(__m512d __A, const int __R)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000750{
751 return (__m256i) __builtin_ia32_cvtpd2dq512_mask ((__v8df) __A,
752 (__v8si)
753 _mm256_setzero_si256 (),
754 (__mmask8) -1,
755 __R);
756}
757static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000758_mm512_cvt_roundps_epu32(__m512 __A, const int __R)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000759{
760 return (__m512i) __builtin_ia32_cvtps2udq512_mask ((__v16sf) __A,
761 (__v16si)
762 _mm512_setzero_si512 (),
763 (__mmask16) -1,
764 __R);
765}
766static __inline __m256i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000767_mm512_cvt_roundpd_epu32(__m512d __A, const int __R)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000768{
769 return (__m256i) __builtin_ia32_cvtpd2udq512_mask ((__v8df) __A,
770 (__v8si)
771 _mm256_setzero_si256 (),
772 (__mmask8) -1,
773 __R);
774}
775
Adam Nemet0d5bb552014-07-28 17:14:40 +0000776/* Bit Test */
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000777
778static __inline __mmask16 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000779_mm512_test_epi32_mask(__m512i __A, __m512i __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000780{
781 return (__mmask16) __builtin_ia32_ptestmd512 ((__v16si) __A,
782 (__v16si) __B,
783 (__mmask16) -1);
784}
785
786static __inline __mmask8 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000787_mm512_test_epi64_mask(__m512i __A, __m512i __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000788{
789 return (__mmask8) __builtin_ia32_ptestmq512 ((__v8di) __A,
790 (__v8di) __B,
791 (__mmask8) -1);
792}
793
Adam Nemet0d5bb552014-07-28 17:14:40 +0000794/* SIMD load ops */
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000795
796static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000797_mm512_maskz_loadu_epi32(__mmask16 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000798{
799 return (__m512i) __builtin_ia32_loaddqusi512_mask ((const __v16si *)__P,
800 (__v16si)
801 _mm512_setzero_si512 (),
802 (__mmask16) __U);
803}
804
805static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000806_mm512_maskz_loadu_epi64(__mmask8 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000807{
808 return (__m512i) __builtin_ia32_loaddqudi512_mask ((const __v8di *)__P,
809 (__v8di)
810 _mm512_setzero_si512 (),
811 (__mmask8) __U);
812}
813
814static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000815_mm512_maskz_loadu_ps(__mmask16 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000816{
817 return (__m512) __builtin_ia32_loadups512_mask ((const __v16sf *)__P,
818 (__v16sf)
819 _mm512_setzero_ps (),
820 (__mmask16) __U);
821}
822
823static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000824_mm512_maskz_loadu_pd(__mmask8 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000825{
826 return (__m512d) __builtin_ia32_loadupd512_mask ((const __v8df *)__P,
827 (__v8df)
828 _mm512_setzero_pd (),
829 (__mmask8) __U);
830}
831
Adam Nemetda82bcc2014-07-31 04:00:39 +0000832static __inline __m512d __attribute__((__always_inline__, __nodebug__))
833_mm512_loadu_pd(double const *__p)
834{
835 struct __loadu_pd {
836 __m512d __v;
837 } __attribute__((packed, may_alias));
838 return ((struct __loadu_pd*)__p)->__v;
839}
840
841static __inline __m512 __attribute__((__always_inline__, __nodebug__))
842_mm512_loadu_ps(float const *__p)
843{
844 struct __loadu_ps {
845 __m512 __v;
846 } __attribute__((packed, may_alias));
847 return ((struct __loadu_ps*)__p)->__v;
848}
849
Adam Nemet0d5bb552014-07-28 17:14:40 +0000850/* SIMD store ops */
851
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000852static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000853_mm512_mask_storeu_epi64(void *__P, __mmask8 __U, __m512i __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000854{
855 __builtin_ia32_storedqudi512_mask ((__v8di *)__P, (__v8di) __A,
856 (__mmask8) __U);
857}
858
859static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000860_mm512_mask_storeu_epi32(void *__P, __mmask16 __U, __m512i __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000861{
862 __builtin_ia32_storedqusi512_mask ((__v16si *)__P, (__v16si) __A,
863 (__mmask16) __U);
864}
865
866static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000867_mm512_mask_storeu_pd(void *__P, __mmask8 __U, __m512d __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000868{
869 __builtin_ia32_storeupd512_mask ((__v8df *)__P, (__v8df) __A, (__mmask8) __U);
870}
871
872static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemetfce1ad02014-07-28 17:14:45 +0000873_mm512_storeu_pd(void *__P, __m512d __A)
874{
875 __builtin_ia32_storeupd512_mask((__v8df *)__P, (__v8df)__A, (__mmask8)-1);
876}
877
878static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000879_mm512_mask_storeu_ps(void *__P, __mmask16 __U, __m512 __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000880{
881 __builtin_ia32_storeups512_mask ((__v16sf *)__P, (__v16sf) __A,
882 (__mmask16) __U);
883}
884
Adam Nemetfce1ad02014-07-28 17:14:45 +0000885static __inline void __attribute__ ((__always_inline__, __nodebug__))
886_mm512_storeu_ps(void *__P, __m512 __A)
887{
888 __builtin_ia32_storeups512_mask((__v16sf *)__P, (__v16sf)__A, (__mmask16)-1);
889}
890
891static __inline void __attribute__ ((__always_inline__, __nodebug__))
892_mm512_store_ps(void *__P, __m512 __A)
893{
894 *(__m512*)__P = __A;
895}
896
897static __inline void __attribute__ ((__always_inline__, __nodebug__))
898_mm512_store_pd(void *__P, __m512d __A)
899{
900 *(__m512d*)__P = __A;
901}
902
Adam Nemet2db1d2f2014-07-30 16:51:27 +0000903/* Mask ops */
904
905static __inline __mmask16 __attribute__ ((__always_inline__, __nodebug__))
906_mm512_knot(__mmask16 __M)
907{
908 return __builtin_ia32_knothi(__M);
909}
910
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000911#endif // __AVX512FINTRIN_H