1/*===------------- avx512ifmavlintrin.h - IFMA intrinsics ------------------===
2 *
3 *
4 * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5 * See https://llvm.org/LICENSE.txt for license information.
6 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7 *
8 *===-----------------------------------------------------------------------===
9 */
10#ifndef __IMMINTRIN_H
11#error "Never use <avx512ifmavlintrin.h> directly; include <immintrin.h> instead."
12#endif
13
14#ifndef __IFMAVLINTRIN_H
15#define __IFMAVLINTRIN_H
16
17/* Define the default attributes for the functions in this file. */
18#define __DEFAULT_FN_ATTRS128 __attribute__((__always_inline__, __nodebug__, __target__("avx512ifma,avx512vl"), __min_vector_width__(128)))
19#define __DEFAULT_FN_ATTRS256 __attribute__((__always_inline__, __nodebug__, __target__("avx512ifma,avx512vl"), __min_vector_width__(256)))
20
21#define _mm_madd52hi_epu64(X, Y, Z)                                            \
22  ((__m128i)__builtin_ia32_vpmadd52huq128((__v2di)(X), (__v2di)(Y),            \
23                                          (__v2di)(Z)))
24
25#define _mm256_madd52hi_epu64(X, Y, Z)                                         \
26  ((__m256i)__builtin_ia32_vpmadd52huq256((__v4di)(X), (__v4di)(Y),            \
27                                          (__v4di)(Z)))
28
29#define _mm_madd52lo_epu64(X, Y, Z)                                            \
30  ((__m128i)__builtin_ia32_vpmadd52luq128((__v2di)(X), (__v2di)(Y),            \
31                                          (__v2di)(Z)))
32
33#define _mm256_madd52lo_epu64(X, Y, Z)                                         \
34  ((__m256i)__builtin_ia32_vpmadd52luq256((__v4di)(X), (__v4di)(Y),            \
35                                          (__v4di)(Z)))
36
37static __inline__ __m128i __DEFAULT_FN_ATTRS128
38_mm_mask_madd52hi_epu64 (__m128i __W, __mmask8 __M, __m128i __X, __m128i __Y)
39{
40  return (__m128i)__builtin_ia32_selectq_128(__M,
41                                      (__v2di)_mm_madd52hi_epu64(__W, __X, __Y),
42                                      (__v2di)__W);
43}
44
45static __inline__ __m128i __DEFAULT_FN_ATTRS128
46_mm_maskz_madd52hi_epu64 (__mmask8 __M, __m128i __X, __m128i __Y, __m128i __Z)
47{
48  return (__m128i)__builtin_ia32_selectq_128(__M,
49                                      (__v2di)_mm_madd52hi_epu64(__X, __Y, __Z),
50                                      (__v2di)_mm_setzero_si128());
51}
52
53static __inline__ __m256i __DEFAULT_FN_ATTRS256
54_mm256_mask_madd52hi_epu64 (__m256i __W, __mmask8 __M, __m256i __X, __m256i __Y)
55{
56  return (__m256i)__builtin_ia32_selectq_256(__M,
57                                   (__v4di)_mm256_madd52hi_epu64(__W, __X, __Y),
58                                   (__v4di)__W);
59}
60
61static __inline__ __m256i __DEFAULT_FN_ATTRS256
62_mm256_maskz_madd52hi_epu64 (__mmask8 __M, __m256i __X, __m256i __Y, __m256i __Z)
63{
64  return (__m256i)__builtin_ia32_selectq_256(__M,
65                                   (__v4di)_mm256_madd52hi_epu64(__X, __Y, __Z),
66                                   (__v4di)_mm256_setzero_si256());
67}
68
69static __inline__ __m128i __DEFAULT_FN_ATTRS128
70_mm_mask_madd52lo_epu64 (__m128i __W, __mmask8 __M, __m128i __X, __m128i __Y)
71{
72  return (__m128i)__builtin_ia32_selectq_128(__M,
73                                      (__v2di)_mm_madd52lo_epu64(__W, __X, __Y),
74                                      (__v2di)__W);
75}
76
77static __inline__ __m128i __DEFAULT_FN_ATTRS128
78_mm_maskz_madd52lo_epu64 (__mmask8 __M, __m128i __X, __m128i __Y, __m128i __Z)
79{
80  return (__m128i)__builtin_ia32_selectq_128(__M,
81                                      (__v2di)_mm_madd52lo_epu64(__X, __Y, __Z),
82                                      (__v2di)_mm_setzero_si128());
83}
84
85static __inline__ __m256i __DEFAULT_FN_ATTRS256
86_mm256_mask_madd52lo_epu64 (__m256i __W, __mmask8 __M, __m256i __X, __m256i __Y)
87{
88  return (__m256i)__builtin_ia32_selectq_256(__M,
89                                   (__v4di)_mm256_madd52lo_epu64(__W, __X, __Y),
90                                   (__v4di)__W);
91}
92
93static __inline__ __m256i __DEFAULT_FN_ATTRS256
94_mm256_maskz_madd52lo_epu64 (__mmask8 __M, __m256i __X, __m256i __Y, __m256i __Z)
95{
96  return (__m256i)__builtin_ia32_selectq_256(__M,
97                                   (__v4di)_mm256_madd52lo_epu64(__X, __Y, __Z),
98                                   (__v4di)_mm256_setzero_si256());
99}
100
101
102#undef __DEFAULT_FN_ATTRS128
103#undef __DEFAULT_FN_ATTRS256
104
105#endif
106