blob: c6d46cb0ab245b5c7600d3cfe41accecb791fb8e [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
Craig Topper72c7d512015-02-01 07:35:35 +0000495#define _mm512_roundscale_ps(A, B) __extension__ ({ \
496 (__m512)__builtin_ia32_rndscaleps_mask((__v16sf)(A), (B), (__v16sf)(A), \
497 -1, _MM_FROUND_CUR_DIRECTION); })
498
499#define _mm512_roundscale_pd(A, B) __extension__ ({ \
500 (__m512d)__builtin_ia32_rndscalepd_mask((__v8df)(A), (B), (__v8df)(A), \
501 -1, _MM_FROUND_CUR_DIRECTION); })
Adam Nemet0d5bb552014-07-28 17:14:40 +0000502
Adam Nemet2278fcb2014-08-14 17:17:57 +0000503static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
504_mm512_fmadd_pd(__m512d __A, __m512d __B, __m512d __C)
505{
506 return (__m512d)
507 __builtin_ia32_vfmaddpd512_mask(__A,
508 __B,
509 __C,
510 (__mmask8) -1,
511 _MM_FROUND_CUR_DIRECTION);
512}
513
514static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
515_mm512_fmsub_pd(__m512d __A, __m512d __B, __m512d __C)
516{
517 return (__m512d)
518 __builtin_ia32_vfmsubpd512_mask(__A,
519 __B,
520 __C,
521 (__mmask8) -1,
522 _MM_FROUND_CUR_DIRECTION);
523}
524
525static __inline__ __m512d __attribute__((__always_inline__, __nodebug__))
526_mm512_fnmadd_pd(__m512d __A, __m512d __B, __m512d __C)
527{
528 return (__m512d)
529 __builtin_ia32_vfnmaddpd512_mask(__A,
530 __B,
531 __C,
532 (__mmask8) -1,
533 _MM_FROUND_CUR_DIRECTION);
534}
535
536static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
537_mm512_fmadd_ps(__m512 __A, __m512 __B, __m512 __C)
538{
539 return (__m512)
540 __builtin_ia32_vfmaddps512_mask(__A,
541 __B,
542 __C,
543 (__mmask16) -1,
544 _MM_FROUND_CUR_DIRECTION);
545}
546
547static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
548_mm512_fmsub_ps(__m512 __A, __m512 __B, __m512 __C)
549{
550 return (__m512)
551 __builtin_ia32_vfmsubps512_mask(__A,
552 __B,
553 __C,
554 (__mmask16) -1,
555 _MM_FROUND_CUR_DIRECTION);
556}
557
558static __inline__ __m512 __attribute__((__always_inline__, __nodebug__))
559_mm512_fnmadd_ps(__m512 __A, __m512 __B, __m512 __C)
560{
561 return (__m512)
562 __builtin_ia32_vfnmaddps512_mask(__A,
563 __B,
564 __C,
565 (__mmask16) -1,
566 _MM_FROUND_CUR_DIRECTION);
567}
568
Adam Nemet0d5bb552014-07-28 17:14:40 +0000569/* Vector permutations */
570
571static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
572_mm512_permutex2var_epi32(__m512i __A, __m512i __I, __m512i __B)
573{
574 return (__m512i) __builtin_ia32_vpermt2vard512_mask ((__v16si) __I
575 /* idx */ ,
576 (__v16si) __A,
577 (__v16si) __B,
578 (__mmask16) -1);
579}
580static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
581_mm512_permutex2var_epi64(__m512i __A, __m512i __I, __m512i __B)
582{
583 return (__m512i) __builtin_ia32_vpermt2varq512_mask ((__v8di) __I
584 /* idx */ ,
585 (__v8di) __A,
586 (__v8di) __B,
587 (__mmask8) -1);
588}
589
590static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
591_mm512_permutex2var_pd(__m512d __A, __m512i __I, __m512d __B)
592{
593 return (__m512d) __builtin_ia32_vpermt2varpd512_mask ((__v8di) __I
594 /* idx */ ,
595 (__v8df) __A,
596 (__v8df) __B,
597 (__mmask8) -1);
598}
599static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
600_mm512_permutex2var_ps(__m512 __A, __m512i __I, __m512 __B)
601{
602 return (__m512) __builtin_ia32_vpermt2varps512_mask ((__v16si) __I
603 /* idx */ ,
604 (__v16sf) __A,
605 (__v16sf) __B,
606 (__mmask16) -1);
607}
608
Craig Topper67826a52015-02-01 07:35:40 +0000609#define _mm512_alignr_epi64(A, B, I) __extension__ ({ \
610 (__m512i)__builtin_ia32_alignq512_mask((__v8di)(__m512i)(A), \
611 (__v8di)(__m512i)(B), \
612 (I), (__v8di)_mm512_setzero_si512(), \
613 (__mmask8)-1); })
Adam Nemet5bf7baa2014-08-05 17:28:23 +0000614
Craig Topper67826a52015-02-01 07:35:40 +0000615#define _mm512_alignr_epi32(A, B, I) __extension__ ({ \
616 (__m512i)__builtin_ia32_alignd512_mask((__v16si)(__m512i)(A), \
617 (__v16si)(__m512i)(B), \
618 (I), (__v16si)_mm512_setzero_si512(), \
619 (__mmask16)-1); })
Adam Nemet5bf7baa2014-08-05 17:28:23 +0000620
Adam Nemetf893ede2015-01-19 20:12:05 +0000621/* Vector Extract */
622
623#define _mm512_extractf64x4_pd(A, I) __extension__ ({ \
624 __m512d __A = (A); \
625 (__m256d) \
626 __builtin_ia32_extractf64x4_mask((__v8df)__A, \
627 (I), \
628 (__v4df)_mm256_setzero_si256(), \
629 (__mmask8) -1); })
630
631#define _mm512_extractf32x4_ps(A, I) __extension__ ({ \
632 __m512 __A = (A); \
633 (__m128) \
634 __builtin_ia32_extractf32x4_mask((__v16sf)__A, \
635 (I), \
636 (__v4sf)_mm_setzero_ps(), \
637 (__mmask8) -1); })
638
Adam Nemet0d5bb552014-07-28 17:14:40 +0000639/* Vector Blend */
640
641static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
642_mm512_mask_blend_pd(__mmask8 __U, __m512d __A, __m512d __W)
643{
644 return (__m512d) __builtin_ia32_blendmpd_512_mask ((__v8df) __A,
645 (__v8df) __W,
646 (__mmask8) __U);
647}
648
649static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
650_mm512_mask_blend_ps(__mmask16 __U, __m512 __A, __m512 __W)
651{
652 return (__m512) __builtin_ia32_blendmps_512_mask ((__v16sf) __A,
653 (__v16sf) __W,
654 (__mmask16) __U);
655}
656
657static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
658_mm512_mask_blend_epi64(__mmask8 __U, __m512i __A, __m512i __W)
659{
660 return (__m512i) __builtin_ia32_blendmq_512_mask ((__v8di) __A,
661 (__v8di) __W,
662 (__mmask8) __U);
663}
664
665static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
666_mm512_mask_blend_epi32(__mmask16 __U, __m512i __A, __m512i __W)
667{
668 return (__m512i) __builtin_ia32_blendmd_512_mask ((__v16si) __A,
669 (__v16si) __W,
670 (__mmask16) __U);
671}
672
673/* Compare */
674
Craig Topper53565c62015-02-01 22:27:40 +0000675#define _mm512_cmp_round_ps_mask(A, B, P, R) __extension__ ({ \
676 (__mmask16)__builtin_ia32_cmpps512_mask((__v16sf)(__m512)(A), \
677 (__v16sf)(__m512)(B), \
678 (P), (__mmask16)-1, (R)); })
Adam Nemet0d5bb552014-07-28 17:14:40 +0000679
Craig Topper53565c62015-02-01 22:27:40 +0000680#define _mm512_mask_cmp_round_ps_mask(U, A, B, P, R) __extension__ ({ \
681 (__mmask16)__builtin_ia32_cmpps512_mask((__v16sf)(__m512)(A), \
682 (__v16sf)(__m512)(B), \
683 (P), (__mmask16)(U), (R)); })
684
685#define _mm512_cmp_ps_mask(A, B, P) \
686 _mm512_cmp_round_ps_mask((A), (B), (P), _MM_FROUND_CUR_DIRECTION)
687
688#define _mm512_mask_cmp_ps_mask(U, A, B, P) \
689 _mm512_mask_cmp_round_ps_mask((U), (A), (B), (P), _MM_FROUND_CUR_DIRECTION)
690
691#define _mm512_cmp_round_pd_mask(A, B, P, R) __extension__ ({ \
692 (__mmask8)__builtin_ia32_cmppd512_mask((__v8df)(__m512d)(A), \
693 (__v8df)(__m512d)(B), \
694 (P), (__mmask8)-1, (R)); })
695
696#define _mm512_mask_cmp_round_pd_mask(U, A, B, P, R) __extension__ ({ \
697 (__mmask8)__builtin_ia32_cmppd512_mask((__v8df)(__m512d)(A), \
698 (__v8df)(__m512d)(B), \
699 (P), (__mmask8)(U), (R)); })
700
701#define _mm512_cmp_pd_mask(A, B, P) \
702 _mm512_cmp_round_pd_mask((A), (B), (P), _MM_FROUND_CUR_DIRECTION)
703
704#define _mm512_mask_cmp_pd_mask(U, A, B, P) \
705 _mm512_mask_cmp_round_pd_mask((U), (A), (B), (P), _MM_FROUND_CUR_DIRECTION)
Adam Nemet0d5bb552014-07-28 17:14:40 +0000706
707/* Conversion */
708
709static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
710_mm512_cvttps_epu32(__m512 __A)
711{
712 return (__m512i) __builtin_ia32_cvttps2udq512_mask ((__v16sf) __A,
713 (__v16si)
714 _mm512_setzero_si512 (),
715 (__mmask16) -1,
716 _MM_FROUND_CUR_DIRECTION);
717}
718
Craig Topper72c7d512015-02-01 07:35:35 +0000719#define _mm512_cvt_roundepi32_ps(A, R) __extension__ ({ \
720 (__m512)__builtin_ia32_cvtdq2ps512_mask((__v16si)(A), \
721 (__v16sf)_mm512_setzero_ps(), \
722 (__mmask16)-1, (R)); })
Adam Nemet0d5bb552014-07-28 17:14:40 +0000723
Craig Topper72c7d512015-02-01 07:35:35 +0000724#define _mm512_cvt_roundepu32_ps(A, R) __extension__ ({ \
725 (__m512)__builtin_ia32_cvtudq2ps512_mask((__v16si)(A), \
726 (__v16sf)_mm512_setzero_ps(), \
727 (__mmask16)-1, (R)); })
Adam Nemet0d5bb552014-07-28 17:14:40 +0000728
729static __inline __m512d __attribute__ (( __always_inline__, __nodebug__))
730_mm512_cvtepi32_pd(__m256i __A)
731{
732 return (__m512d) __builtin_ia32_cvtdq2pd512_mask ((__v8si) __A,
733 (__v8df)
734 _mm512_setzero_pd (),
735 (__mmask8) -1);
736}
737
738static __inline __m512d __attribute__ (( __always_inline__, __nodebug__))
739_mm512_cvtepu32_pd(__m256i __A)
740{
741 return (__m512d) __builtin_ia32_cvtudq2pd512_mask ((__v8si) __A,
742 (__v8df)
743 _mm512_setzero_pd (),
744 (__mmask8) -1);
745}
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000746
Craig Topper72c7d512015-02-01 07:35:35 +0000747#define _mm512_cvt_roundpd_ps(A, R) __extension__ ({ \
748 (__m256)__builtin_ia32_cvtpd2ps512_mask((__v8df)(A), \
749 (__v8sf)_mm256_setzero_ps(), \
750 (__mmask8)-1, (R)); })
751
752#define _mm512_cvtps_ph(A, I) __extension__ ({ \
753 (__m256i)__builtin_ia32_vcvtps2ph512_mask((__v16sf)(A), (I), \
754 (__v16hi)_mm256_setzero_si256(), \
755 -1); })
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000756
757static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000758_mm512_cvtph_ps(__m256i __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000759{
760 return (__m512) __builtin_ia32_vcvtph2ps512_mask ((__v16hi) __A,
761 (__v16sf)
762 _mm512_setzero_ps (),
763 (__mmask16) -1,
764 _MM_FROUND_CUR_DIRECTION);
765}
766
767static __inline __m512i __attribute__((__always_inline__, __nodebug__))
768_mm512_cvttps_epi32(__m512 a)
769{
770 return (__m512i)
771 __builtin_ia32_cvttps2dq512_mask((__v16sf) a,
772 (__v16si) _mm512_setzero_si512 (),
773 (__mmask16) -1, _MM_FROUND_CUR_DIRECTION);
774}
775
776static __inline __m256i __attribute__((__always_inline__, __nodebug__))
777_mm512_cvttpd_epi32(__m512d a)
778{
779 return (__m256i)__builtin_ia32_cvttpd2dq512_mask((__v8df) a,
780 (__v8si)_mm256_setzero_si256(),
781 (__mmask8) -1,
782 _MM_FROUND_CUR_DIRECTION);
783}
784
Craig Topper72c7d512015-02-01 07:35:35 +0000785#define _mm512_cvtt_roundpd_epi32(A, R) __extension__ ({ \
786 (__m256i)__builtin_ia32_cvttpd2dq512_mask((__v8df)(A), \
787 (__v8si)_mm256_setzero_si256(), \
788 (__mmask8)-1, (R)); })
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000789
Craig Topper72c7d512015-02-01 07:35:35 +0000790#define _mm512_cvtt_roundps_epi32(A, R) __extension__ ({ \
791 (__m512i)__builtin_ia32_cvttps2dq512_mask((__v16sf)(A), \
792 (__v16si)_mm512_setzero_si512(), \
793 (__mmask16)-1, (R)); })
794
795#define _mm512_cvt_roundps_epi32(A, R) __extension__ ({ \
796 (__m512i)__builtin_ia32_cvtps2dq512_mask((__v16sf)(A), \
797 (__v16si)_mm512_setzero_si512(), \
798 (__mmask16)-1, (R)); })
799
800#define _mm512_cvt_roundpd_epi32(A, R) __extension__ ({ \
801 (__m256i)__builtin_ia32_cvtpd2dq512_mask((__v8df)(A), \
802 (__v8si)_mm256_setzero_si256(), \
803 (__mmask8)-1, (R)); })
804
805#define _mm512_cvt_roundps_epu32(A, R) __extension__ ({ \
806 (__m512i)__builtin_ia32_cvtps2udq512_mask((__v16sf)(A), \
807 (__v16si)_mm512_setzero_si512(), \
808 (__mmask16)-1, (R)); })
809
810#define _mm512_cvt_roundpd_epu32(A, R) __extension__ ({ \
811 (__m256i)__builtin_ia32_cvtpd2udq512_mask((__v8df)(A), \
812 (__v8si)_mm256_setzero_si256(), \
813 (__mmask8) -1, (R)); })
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000814
Adam Nemet63a951e2015-01-14 01:31:17 +0000815/* Unpack and Interleave */
816static __inline __m512d __attribute__((__always_inline__, __nodebug__))
817_mm512_unpackhi_pd(__m512d __a, __m512d __b)
818{
819 return __builtin_shufflevector(__a, __b, 1, 9, 1+2, 9+2, 1+4, 9+4, 1+6, 9+6);
820}
821
822static __inline __m512d __attribute__((__always_inline__, __nodebug__))
823_mm512_unpacklo_pd(__m512d __a, __m512d __b)
824{
825 return __builtin_shufflevector(__a, __b, 0, 8, 0+2, 8+2, 0+4, 8+4, 0+6, 8+6);
826}
827
828static __inline __m512 __attribute__((__always_inline__, __nodebug__))
829_mm512_unpackhi_ps(__m512 __a, __m512 __b)
830{
831 return __builtin_shufflevector(__a, __b,
832 2, 18, 3, 19,
833 2+4, 18+4, 3+4, 19+4,
834 2+8, 18+8, 3+8, 19+8,
835 2+12, 18+12, 3+12, 19+12);
836}
837
838static __inline __m512 __attribute__((__always_inline__, __nodebug__))
839_mm512_unpacklo_ps(__m512 __a, __m512 __b)
840{
841 return __builtin_shufflevector(__a, __b,
842 0, 16, 1, 17,
843 0+4, 16+4, 1+4, 17+4,
844 0+8, 16+8, 1+8, 17+8,
845 0+12, 16+12, 1+12, 17+12);
846}
847
Adam Nemet0d5bb552014-07-28 17:14:40 +0000848/* Bit Test */
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000849
850static __inline __mmask16 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000851_mm512_test_epi32_mask(__m512i __A, __m512i __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000852{
853 return (__mmask16) __builtin_ia32_ptestmd512 ((__v16si) __A,
854 (__v16si) __B,
855 (__mmask16) -1);
856}
857
858static __inline __mmask8 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000859_mm512_test_epi64_mask(__m512i __A, __m512i __B)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000860{
861 return (__mmask8) __builtin_ia32_ptestmq512 ((__v8di) __A,
862 (__v8di) __B,
863 (__mmask8) -1);
864}
865
Adam Nemet0d5bb552014-07-28 17:14:40 +0000866/* SIMD load ops */
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000867
868static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000869_mm512_maskz_loadu_epi32(__mmask16 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000870{
871 return (__m512i) __builtin_ia32_loaddqusi512_mask ((const __v16si *)__P,
872 (__v16si)
873 _mm512_setzero_si512 (),
874 (__mmask16) __U);
875}
876
877static __inline __m512i __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000878_mm512_maskz_loadu_epi64(__mmask8 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000879{
880 return (__m512i) __builtin_ia32_loaddqudi512_mask ((const __v8di *)__P,
881 (__v8di)
882 _mm512_setzero_si512 (),
883 (__mmask8) __U);
884}
885
886static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000887_mm512_maskz_loadu_ps(__mmask16 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000888{
889 return (__m512) __builtin_ia32_loadups512_mask ((const __v16sf *)__P,
890 (__v16sf)
891 _mm512_setzero_ps (),
892 (__mmask16) __U);
893}
894
895static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000896_mm512_maskz_loadu_pd(__mmask8 __U, void const *__P)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000897{
898 return (__m512d) __builtin_ia32_loadupd512_mask ((const __v8df *)__P,
899 (__v8df)
900 _mm512_setzero_pd (),
901 (__mmask8) __U);
902}
903
Adam Nemetc0cff242015-01-16 18:51:50 +0000904static __inline __m512 __attribute__ ((__always_inline__, __nodebug__))
905_mm512_maskz_load_ps(__mmask16 __U, void const *__P)
906{
907 return (__m512) __builtin_ia32_loadaps512_mask ((const __v16sf *)__P,
908 (__v16sf)
909 _mm512_setzero_ps (),
910 (__mmask16) __U);
911}
912
913static __inline __m512d __attribute__ ((__always_inline__, __nodebug__))
914_mm512_maskz_load_pd(__mmask8 __U, void const *__P)
915{
916 return (__m512d) __builtin_ia32_loadapd512_mask ((const __v8df *)__P,
917 (__v8df)
918 _mm512_setzero_pd (),
919 (__mmask8) __U);
920}
921
Adam Nemetda82bcc2014-07-31 04:00:39 +0000922static __inline __m512d __attribute__((__always_inline__, __nodebug__))
923_mm512_loadu_pd(double const *__p)
924{
925 struct __loadu_pd {
926 __m512d __v;
David Majnemer1cf22e62015-02-04 00:26:10 +0000927 } __attribute__((__packed__, __may_alias__));
Adam Nemetda82bcc2014-07-31 04:00:39 +0000928 return ((struct __loadu_pd*)__p)->__v;
929}
930
931static __inline __m512 __attribute__((__always_inline__, __nodebug__))
932_mm512_loadu_ps(float const *__p)
933{
934 struct __loadu_ps {
935 __m512 __v;
David Majnemer1cf22e62015-02-04 00:26:10 +0000936 } __attribute__((__packed__, __may_alias__));
Adam Nemetda82bcc2014-07-31 04:00:39 +0000937 return ((struct __loadu_ps*)__p)->__v;
938}
939
Adam Nemetc0cff242015-01-16 18:51:50 +0000940static __inline __m512 __attribute__((__always_inline__, __nodebug__))
941_mm512_load_ps(double const *__p)
942{
943 return (__m512) __builtin_ia32_loadaps512_mask ((const __v16sf *)__p,
944 (__v16sf)
945 _mm512_setzero_ps (),
946 (__mmask16) -1);
947}
948
949static __inline __m512d __attribute__((__always_inline__, __nodebug__))
950_mm512_load_pd(float const *__p)
951{
952 return (__m512d) __builtin_ia32_loadapd512_mask ((const __v8df *)__p,
953 (__v8df)
954 _mm512_setzero_pd (),
955 (__mmask8) -1);
956}
957
Adam Nemet0d5bb552014-07-28 17:14:40 +0000958/* SIMD store ops */
959
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000960static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000961_mm512_mask_storeu_epi64(void *__P, __mmask8 __U, __m512i __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000962{
963 __builtin_ia32_storedqudi512_mask ((__v8di *)__P, (__v8di) __A,
964 (__mmask8) __U);
965}
966
967static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000968_mm512_mask_storeu_epi32(void *__P, __mmask16 __U, __m512i __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000969{
970 __builtin_ia32_storedqusi512_mask ((__v16si *)__P, (__v16si) __A,
971 (__mmask16) __U);
972}
973
974static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000975_mm512_mask_storeu_pd(void *__P, __mmask8 __U, __m512d __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000976{
977 __builtin_ia32_storeupd512_mask ((__v8df *)__P, (__v8df) __A, (__mmask8) __U);
978}
979
980static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemetfce1ad02014-07-28 17:14:45 +0000981_mm512_storeu_pd(void *__P, __m512d __A)
982{
983 __builtin_ia32_storeupd512_mask((__v8df *)__P, (__v8df)__A, (__mmask8)-1);
984}
985
986static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemet9a3ea602014-07-28 17:14:38 +0000987_mm512_mask_storeu_ps(void *__P, __mmask16 __U, __m512 __A)
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +0000988{
989 __builtin_ia32_storeups512_mask ((__v16sf *)__P, (__v16sf) __A,
990 (__mmask16) __U);
991}
992
Adam Nemetfce1ad02014-07-28 17:14:45 +0000993static __inline void __attribute__ ((__always_inline__, __nodebug__))
994_mm512_storeu_ps(void *__P, __m512 __A)
995{
996 __builtin_ia32_storeups512_mask((__v16sf *)__P, (__v16sf)__A, (__mmask16)-1);
997}
998
999static __inline void __attribute__ ((__always_inline__, __nodebug__))
Adam Nemetc0cff242015-01-16 18:51:50 +00001000_mm512_mask_store_pd(void *__P, __mmask8 __U, __m512d __A)
Adam Nemetfce1ad02014-07-28 17:14:45 +00001001{
Adam Nemetc0cff242015-01-16 18:51:50 +00001002 __builtin_ia32_storeapd512_mask ((__v8df *)__P, (__v8df) __A, (__mmask8) __U);
Adam Nemetfce1ad02014-07-28 17:14:45 +00001003}
1004
1005static __inline void __attribute__ ((__always_inline__, __nodebug__))
1006_mm512_store_pd(void *__P, __m512d __A)
1007{
1008 *(__m512d*)__P = __A;
1009}
1010
Adam Nemetc0cff242015-01-16 18:51:50 +00001011static __inline void __attribute__ ((__always_inline__, __nodebug__))
1012_mm512_mask_store_ps(void *__P, __mmask16 __U, __m512 __A)
1013{
1014 __builtin_ia32_storeaps512_mask ((__v16sf *)__P, (__v16sf) __A,
1015 (__mmask16) __U);
1016}
1017
1018static __inline void __attribute__ ((__always_inline__, __nodebug__))
1019_mm512_store_ps(void *__P, __m512 __A)
1020{
1021 *(__m512*)__P = __A;
1022}
1023
Adam Nemet2db1d2f2014-07-30 16:51:27 +00001024/* Mask ops */
1025
1026static __inline __mmask16 __attribute__ ((__always_inline__, __nodebug__))
1027_mm512_knot(__mmask16 __M)
1028{
1029 return __builtin_ia32_knothi(__M);
1030}
1031
Robert Khasanovb9f3a912014-10-08 17:18:13 +00001032/* Integer compare */
1033
1034static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1035_mm512_cmpeq_epi32_mask(__m512i __a, __m512i __b) {
1036 return (__mmask16)__builtin_ia32_pcmpeqd512_mask((__v16si)__a, (__v16si)__b,
1037 (__mmask16)-1);
1038}
1039
1040static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1041_mm512_mask_cmpeq_epi32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1042 return (__mmask16)__builtin_ia32_pcmpeqd512_mask((__v16si)__a, (__v16si)__b,
1043 __u);
1044}
1045
Craig Topper4cac1c22015-01-25 23:30:07 +00001046static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1047_mm512_cmpeq_epu32_mask(__m512i __a, __m512i __b) {
1048 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 0,
1049 (__mmask16)-1);
1050}
1051
1052static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1053_mm512_mask_cmpeq_epu32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1054 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 0,
1055 __u);
1056}
1057
Robert Khasanovb9f3a912014-10-08 17:18:13 +00001058static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1059_mm512_mask_cmpeq_epi64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1060 return (__mmask8)__builtin_ia32_pcmpeqq512_mask((__v8di)__a, (__v8di)__b,
1061 __u);
1062}
1063
1064static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1065_mm512_cmpeq_epi64_mask(__m512i __a, __m512i __b) {
1066 return (__mmask8)__builtin_ia32_pcmpeqq512_mask((__v8di)__a, (__v8di)__b,
1067 (__mmask8)-1);
1068}
1069
Craig Topper4cac1c22015-01-25 23:30:07 +00001070static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1071_mm512_cmpeq_epu64_mask(__m512i __a, __m512i __b) {
1072 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 0,
1073 (__mmask8)-1);
1074}
1075
1076static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1077_mm512_mask_cmpeq_epu64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1078 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 0,
1079 __u);
1080}
1081
1082static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1083_mm512_cmpge_epi32_mask(__m512i __a, __m512i __b) {
1084 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 5,
1085 (__mmask16)-1);
1086}
1087
1088static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1089_mm512_mask_cmpge_epi32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1090 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 5,
1091 __u);
1092}
1093
1094static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1095_mm512_cmpge_epu32_mask(__m512i __a, __m512i __b) {
1096 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 5,
1097 (__mmask16)-1);
1098}
1099
1100static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1101_mm512_mask_cmpge_epu32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1102 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 5,
1103 __u);
1104}
1105
1106static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1107_mm512_cmpge_epi64_mask(__m512i __a, __m512i __b) {
1108 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 5,
1109 (__mmask8)-1);
1110}
1111
1112static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1113_mm512_mask_cmpge_epi64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1114 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 5,
1115 __u);
1116}
1117
1118static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1119_mm512_cmpge_epu64_mask(__m512i __a, __m512i __b) {
1120 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 5,
1121 (__mmask8)-1);
1122}
1123
1124static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1125_mm512_mask_cmpge_epu64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1126 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 5,
1127 __u);
1128}
1129
1130static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1131_mm512_cmpgt_epi32_mask(__m512i __a, __m512i __b) {
1132 return (__mmask16)__builtin_ia32_pcmpgtd512_mask((__v16si)__a, (__v16si)__b,
1133 (__mmask16)-1);
1134}
1135
1136static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1137_mm512_mask_cmpgt_epi32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1138 return (__mmask16)__builtin_ia32_pcmpgtd512_mask((__v16si)__a, (__v16si)__b,
1139 __u);
1140}
1141
1142static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1143_mm512_cmpgt_epu32_mask(__m512i __a, __m512i __b) {
1144 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 6,
1145 (__mmask16)-1);
1146}
1147
1148static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1149_mm512_mask_cmpgt_epu32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1150 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 6,
1151 __u);
1152}
1153
1154static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1155_mm512_mask_cmpgt_epi64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1156 return (__mmask8)__builtin_ia32_pcmpgtq512_mask((__v8di)__a, (__v8di)__b,
1157 __u);
1158}
1159
1160static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1161_mm512_cmpgt_epi64_mask(__m512i __a, __m512i __b) {
1162 return (__mmask8)__builtin_ia32_pcmpgtq512_mask((__v8di)__a, (__v8di)__b,
1163 (__mmask8)-1);
1164}
1165
1166static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1167_mm512_cmpgt_epu64_mask(__m512i __a, __m512i __b) {
1168 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 6,
1169 (__mmask8)-1);
1170}
1171
1172static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1173_mm512_mask_cmpgt_epu64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1174 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 6,
1175 __u);
1176}
1177
1178static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1179_mm512_cmple_epi32_mask(__m512i __a, __m512i __b) {
1180 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 2,
1181 (__mmask16)-1);
1182}
1183
1184static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1185_mm512_mask_cmple_epi32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1186 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 2,
1187 __u);
1188}
1189
1190static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1191_mm512_cmple_epu32_mask(__m512i __a, __m512i __b) {
1192 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 2,
1193 (__mmask16)-1);
1194}
1195
1196static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1197_mm512_mask_cmple_epu32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1198 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 2,
1199 __u);
1200}
1201
1202static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1203_mm512_cmple_epi64_mask(__m512i __a, __m512i __b) {
1204 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 2,
1205 (__mmask8)-1);
1206}
1207
1208static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1209_mm512_mask_cmple_epi64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1210 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 2,
1211 __u);
1212}
1213
1214static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1215_mm512_cmple_epu64_mask(__m512i __a, __m512i __b) {
1216 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 2,
1217 (__mmask8)-1);
1218}
1219
1220static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1221_mm512_mask_cmple_epu64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1222 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 2,
1223 __u);
1224}
1225
1226static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1227_mm512_cmplt_epi32_mask(__m512i __a, __m512i __b) {
1228 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 1,
1229 (__mmask16)-1);
1230}
1231
1232static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1233_mm512_mask_cmplt_epi32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1234 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 1,
1235 __u);
1236}
1237
1238static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1239_mm512_cmplt_epu32_mask(__m512i __a, __m512i __b) {
1240 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 1,
1241 (__mmask16)-1);
1242}
1243
1244static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1245_mm512_mask_cmplt_epu32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1246 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 1,
1247 __u);
1248}
1249
1250static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1251_mm512_cmplt_epi64_mask(__m512i __a, __m512i __b) {
1252 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 1,
1253 (__mmask8)-1);
1254}
1255
1256static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1257_mm512_mask_cmplt_epi64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1258 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 1,
1259 __u);
1260}
1261
1262static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1263_mm512_cmplt_epu64_mask(__m512i __a, __m512i __b) {
1264 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 1,
1265 (__mmask8)-1);
1266}
1267
1268static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1269_mm512_mask_cmplt_epu64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1270 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 1,
1271 __u);
1272}
1273
1274static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1275_mm512_cmpneq_epi32_mask(__m512i __a, __m512i __b) {
1276 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 4,
1277 (__mmask16)-1);
1278}
1279
1280static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1281_mm512_mask_cmpneq_epi32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1282 return (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, 4,
1283 __u);
1284}
1285
1286static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1287_mm512_cmpneq_epu32_mask(__m512i __a, __m512i __b) {
1288 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 4,
1289 (__mmask16)-1);
1290}
1291
1292static __inline__ __mmask16 __attribute__((__always_inline__, __nodebug__))
1293_mm512_mask_cmpneq_epu32_mask(__mmask16 __u, __m512i __a, __m512i __b) {
1294 return (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, 4,
1295 __u);
1296}
1297
1298static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1299_mm512_cmpneq_epi64_mask(__m512i __a, __m512i __b) {
1300 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 4,
1301 (__mmask8)-1);
1302}
1303
1304static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1305_mm512_mask_cmpneq_epi64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1306 return (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, 4,
1307 __u);
1308}
1309
1310static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1311_mm512_cmpneq_epu64_mask(__m512i __a, __m512i __b) {
1312 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 4,
1313 (__mmask8)-1);
1314}
1315
1316static __inline__ __mmask8 __attribute__((__always_inline__, __nodebug__))
1317_mm512_mask_cmpneq_epu64_mask(__mmask8 __u, __m512i __a, __m512i __b) {
1318 return (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, 4,
1319 __u);
1320}
1321
1322#define _mm512_cmp_epi32_mask(a, b, p) __extension__ ({ \
1323 __m512i __a = (a); \
1324 __m512i __b = (b); \
1325 (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, (p), \
1326 (__mmask16)-1); })
1327
1328#define _mm512_cmp_epu32_mask(a, b, p) __extension__ ({ \
1329 __m512i __a = (a); \
1330 __m512i __b = (b); \
1331 (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, (p), \
1332 (__mmask16)-1); })
1333
1334#define _mm512_cmp_epi64_mask(a, b, p) __extension__ ({ \
1335 __m512i __a = (a); \
1336 __m512i __b = (b); \
1337 (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, (p), \
1338 (__mmask8)-1); })
1339
1340#define _mm512_cmp_epu64_mask(a, b, p) __extension__ ({ \
1341 __m512i __a = (a); \
1342 __m512i __b = (b); \
1343 (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, (p), \
1344 (__mmask8)-1); })
1345
1346#define _mm512_mask_cmp_epi32_mask(m, a, b, p) __extension__ ({ \
1347 __m512i __a = (a); \
1348 __m512i __b = (b); \
1349 (__mmask16)__builtin_ia32_cmpd512_mask((__v16si)__a, (__v16si)__b, (p), \
1350 (__mmask16)(m)); })
1351
1352#define _mm512_mask_cmp_epu32_mask(m, a, b, p) __extension__ ({ \
1353 __m512i __a = (a); \
1354 __m512i __b = (b); \
1355 (__mmask16)__builtin_ia32_ucmpd512_mask((__v16si)__a, (__v16si)__b, (p), \
1356 (__mmask16)(m)); })
1357
1358#define _mm512_mask_cmp_epi64_mask(m, a, b, p) __extension__ ({ \
1359 __m512i __a = (a); \
1360 __m512i __b = (b); \
1361 (__mmask8)__builtin_ia32_cmpq512_mask((__v8di)__a, (__v8di)__b, (p), \
1362 (__mmask8)(m)); })
1363
1364#define _mm512_mask_cmp_epu64_mask(m, a, b, p) __extension__ ({ \
1365 __m512i __a = (a); \
1366 __m512i __b = (b); \
1367 (__mmask8)__builtin_ia32_ucmpq512_mask((__v8di)__a, (__v8di)__b, (p), \
1368 (__mmask8)(m)); })
Elena Demikhovskyfcc6df32014-07-22 11:31:39 +00001369#endif // __AVX512FINTRIN_H