Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | /*===------------- avx512vbmi2intrin.h - VBMI2 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 <avx512vbmi2intrin.h> directly; include <immintrin.h> instead." |
||
| 12 | #endif |
||
| 13 | |||
| 14 | #ifndef __AVX512VBMI2INTRIN_H |
||
| 15 | #define __AVX512VBMI2INTRIN_H |
||
| 16 | |||
| 17 | /* Define the default attributes for the functions in this file. */ |
||
| 18 | #define __DEFAULT_FN_ATTRS __attribute__((__always_inline__, __nodebug__, __target__("avx512vbmi2"), __min_vector_width__(512))) |
||
| 19 | |||
| 20 | |||
| 21 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 22 | _mm512_mask_compress_epi16(__m512i __S, __mmask32 __U, __m512i __D) |
||
| 23 | { |
||
| 24 | return (__m512i) __builtin_ia32_compresshi512_mask ((__v32hi) __D, |
||
| 25 | (__v32hi) __S, |
||
| 26 | __U); |
||
| 27 | } |
||
| 28 | |||
| 29 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 30 | _mm512_maskz_compress_epi16(__mmask32 __U, __m512i __D) |
||
| 31 | { |
||
| 32 | return (__m512i) __builtin_ia32_compresshi512_mask ((__v32hi) __D, |
||
| 33 | (__v32hi) _mm512_setzero_si512(), |
||
| 34 | __U); |
||
| 35 | } |
||
| 36 | |||
| 37 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 38 | _mm512_mask_compress_epi8(__m512i __S, __mmask64 __U, __m512i __D) |
||
| 39 | { |
||
| 40 | return (__m512i) __builtin_ia32_compressqi512_mask ((__v64qi) __D, |
||
| 41 | (__v64qi) __S, |
||
| 42 | __U); |
||
| 43 | } |
||
| 44 | |||
| 45 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 46 | _mm512_maskz_compress_epi8(__mmask64 __U, __m512i __D) |
||
| 47 | { |
||
| 48 | return (__m512i) __builtin_ia32_compressqi512_mask ((__v64qi) __D, |
||
| 49 | (__v64qi) _mm512_setzero_si512(), |
||
| 50 | __U); |
||
| 51 | } |
||
| 52 | |||
| 53 | static __inline__ void __DEFAULT_FN_ATTRS |
||
| 54 | _mm512_mask_compressstoreu_epi16(void *__P, __mmask32 __U, __m512i __D) |
||
| 55 | { |
||
| 56 | __builtin_ia32_compressstorehi512_mask ((__v32hi *) __P, (__v32hi) __D, |
||
| 57 | __U); |
||
| 58 | } |
||
| 59 | |||
| 60 | static __inline__ void __DEFAULT_FN_ATTRS |
||
| 61 | _mm512_mask_compressstoreu_epi8(void *__P, __mmask64 __U, __m512i __D) |
||
| 62 | { |
||
| 63 | __builtin_ia32_compressstoreqi512_mask ((__v64qi *) __P, (__v64qi) __D, |
||
| 64 | __U); |
||
| 65 | } |
||
| 66 | |||
| 67 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 68 | _mm512_mask_expand_epi16(__m512i __S, __mmask32 __U, __m512i __D) |
||
| 69 | { |
||
| 70 | return (__m512i) __builtin_ia32_expandhi512_mask ((__v32hi) __D, |
||
| 71 | (__v32hi) __S, |
||
| 72 | __U); |
||
| 73 | } |
||
| 74 | |||
| 75 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 76 | _mm512_maskz_expand_epi16(__mmask32 __U, __m512i __D) |
||
| 77 | { |
||
| 78 | return (__m512i) __builtin_ia32_expandhi512_mask ((__v32hi) __D, |
||
| 79 | (__v32hi) _mm512_setzero_si512(), |
||
| 80 | __U); |
||
| 81 | } |
||
| 82 | |||
| 83 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 84 | _mm512_mask_expand_epi8(__m512i __S, __mmask64 __U, __m512i __D) |
||
| 85 | { |
||
| 86 | return (__m512i) __builtin_ia32_expandqi512_mask ((__v64qi) __D, |
||
| 87 | (__v64qi) __S, |
||
| 88 | __U); |
||
| 89 | } |
||
| 90 | |||
| 91 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 92 | _mm512_maskz_expand_epi8(__mmask64 __U, __m512i __D) |
||
| 93 | { |
||
| 94 | return (__m512i) __builtin_ia32_expandqi512_mask ((__v64qi) __D, |
||
| 95 | (__v64qi) _mm512_setzero_si512(), |
||
| 96 | __U); |
||
| 97 | } |
||
| 98 | |||
| 99 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 100 | _mm512_mask_expandloadu_epi16(__m512i __S, __mmask32 __U, void const *__P) |
||
| 101 | { |
||
| 102 | return (__m512i) __builtin_ia32_expandloadhi512_mask ((const __v32hi *)__P, |
||
| 103 | (__v32hi) __S, |
||
| 104 | __U); |
||
| 105 | } |
||
| 106 | |||
| 107 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 108 | _mm512_maskz_expandloadu_epi16(__mmask32 __U, void const *__P) |
||
| 109 | { |
||
| 110 | return (__m512i) __builtin_ia32_expandloadhi512_mask ((const __v32hi *)__P, |
||
| 111 | (__v32hi) _mm512_setzero_si512(), |
||
| 112 | __U); |
||
| 113 | } |
||
| 114 | |||
| 115 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 116 | _mm512_mask_expandloadu_epi8(__m512i __S, __mmask64 __U, void const *__P) |
||
| 117 | { |
||
| 118 | return (__m512i) __builtin_ia32_expandloadqi512_mask ((const __v64qi *)__P, |
||
| 119 | (__v64qi) __S, |
||
| 120 | __U); |
||
| 121 | } |
||
| 122 | |||
| 123 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 124 | _mm512_maskz_expandloadu_epi8(__mmask64 __U, void const *__P) |
||
| 125 | { |
||
| 126 | return (__m512i) __builtin_ia32_expandloadqi512_mask ((const __v64qi *)__P, |
||
| 127 | (__v64qi) _mm512_setzero_si512(), |
||
| 128 | __U); |
||
| 129 | } |
||
| 130 | |||
| 131 | #define _mm512_shldi_epi64(A, B, I) \ |
||
| 132 | ((__m512i)__builtin_ia32_vpshldq512((__v8di)(__m512i)(A), \ |
||
| 133 | (__v8di)(__m512i)(B), (int)(I))) |
||
| 134 | |||
| 135 | #define _mm512_mask_shldi_epi64(S, U, A, B, I) \ |
||
| 136 | ((__m512i)__builtin_ia32_selectq_512((__mmask8)(U), \ |
||
| 137 | (__v8di)_mm512_shldi_epi64((A), (B), (I)), \ |
||
| 138 | (__v8di)(__m512i)(S))) |
||
| 139 | |||
| 140 | #define _mm512_maskz_shldi_epi64(U, A, B, I) \ |
||
| 141 | ((__m512i)__builtin_ia32_selectq_512((__mmask8)(U), \ |
||
| 142 | (__v8di)_mm512_shldi_epi64((A), (B), (I)), \ |
||
| 143 | (__v8di)_mm512_setzero_si512())) |
||
| 144 | |||
| 145 | #define _mm512_shldi_epi32(A, B, I) \ |
||
| 146 | ((__m512i)__builtin_ia32_vpshldd512((__v16si)(__m512i)(A), \ |
||
| 147 | (__v16si)(__m512i)(B), (int)(I))) |
||
| 148 | |||
| 149 | #define _mm512_mask_shldi_epi32(S, U, A, B, I) \ |
||
| 150 | ((__m512i)__builtin_ia32_selectd_512((__mmask16)(U), \ |
||
| 151 | (__v16si)_mm512_shldi_epi32((A), (B), (I)), \ |
||
| 152 | (__v16si)(__m512i)(S))) |
||
| 153 | |||
| 154 | #define _mm512_maskz_shldi_epi32(U, A, B, I) \ |
||
| 155 | ((__m512i)__builtin_ia32_selectd_512((__mmask16)(U), \ |
||
| 156 | (__v16si)_mm512_shldi_epi32((A), (B), (I)), \ |
||
| 157 | (__v16si)_mm512_setzero_si512())) |
||
| 158 | |||
| 159 | #define _mm512_shldi_epi16(A, B, I) \ |
||
| 160 | ((__m512i)__builtin_ia32_vpshldw512((__v32hi)(__m512i)(A), \ |
||
| 161 | (__v32hi)(__m512i)(B), (int)(I))) |
||
| 162 | |||
| 163 | #define _mm512_mask_shldi_epi16(S, U, A, B, I) \ |
||
| 164 | ((__m512i)__builtin_ia32_selectw_512((__mmask32)(U), \ |
||
| 165 | (__v32hi)_mm512_shldi_epi16((A), (B), (I)), \ |
||
| 166 | (__v32hi)(__m512i)(S))) |
||
| 167 | |||
| 168 | #define _mm512_maskz_shldi_epi16(U, A, B, I) \ |
||
| 169 | ((__m512i)__builtin_ia32_selectw_512((__mmask32)(U), \ |
||
| 170 | (__v32hi)_mm512_shldi_epi16((A), (B), (I)), \ |
||
| 171 | (__v32hi)_mm512_setzero_si512())) |
||
| 172 | |||
| 173 | #define _mm512_shrdi_epi64(A, B, I) \ |
||
| 174 | ((__m512i)__builtin_ia32_vpshrdq512((__v8di)(__m512i)(A), \ |
||
| 175 | (__v8di)(__m512i)(B), (int)(I))) |
||
| 176 | |||
| 177 | #define _mm512_mask_shrdi_epi64(S, U, A, B, I) \ |
||
| 178 | ((__m512i)__builtin_ia32_selectq_512((__mmask8)(U), \ |
||
| 179 | (__v8di)_mm512_shrdi_epi64((A), (B), (I)), \ |
||
| 180 | (__v8di)(__m512i)(S))) |
||
| 181 | |||
| 182 | #define _mm512_maskz_shrdi_epi64(U, A, B, I) \ |
||
| 183 | ((__m512i)__builtin_ia32_selectq_512((__mmask8)(U), \ |
||
| 184 | (__v8di)_mm512_shrdi_epi64((A), (B), (I)), \ |
||
| 185 | (__v8di)_mm512_setzero_si512())) |
||
| 186 | |||
| 187 | #define _mm512_shrdi_epi32(A, B, I) \ |
||
| 188 | ((__m512i)__builtin_ia32_vpshrdd512((__v16si)(__m512i)(A), \ |
||
| 189 | (__v16si)(__m512i)(B), (int)(I))) |
||
| 190 | |||
| 191 | #define _mm512_mask_shrdi_epi32(S, U, A, B, I) \ |
||
| 192 | ((__m512i)__builtin_ia32_selectd_512((__mmask16)(U), \ |
||
| 193 | (__v16si)_mm512_shrdi_epi32((A), (B), (I)), \ |
||
| 194 | (__v16si)(__m512i)(S))) |
||
| 195 | |||
| 196 | #define _mm512_maskz_shrdi_epi32(U, A, B, I) \ |
||
| 197 | ((__m512i)__builtin_ia32_selectd_512((__mmask16)(U), \ |
||
| 198 | (__v16si)_mm512_shrdi_epi32((A), (B), (I)), \ |
||
| 199 | (__v16si)_mm512_setzero_si512())) |
||
| 200 | |||
| 201 | #define _mm512_shrdi_epi16(A, B, I) \ |
||
| 202 | ((__m512i)__builtin_ia32_vpshrdw512((__v32hi)(__m512i)(A), \ |
||
| 203 | (__v32hi)(__m512i)(B), (int)(I))) |
||
| 204 | |||
| 205 | #define _mm512_mask_shrdi_epi16(S, U, A, B, I) \ |
||
| 206 | ((__m512i)__builtin_ia32_selectw_512((__mmask32)(U), \ |
||
| 207 | (__v32hi)_mm512_shrdi_epi16((A), (B), (I)), \ |
||
| 208 | (__v32hi)(__m512i)(S))) |
||
| 209 | |||
| 210 | #define _mm512_maskz_shrdi_epi16(U, A, B, I) \ |
||
| 211 | ((__m512i)__builtin_ia32_selectw_512((__mmask32)(U), \ |
||
| 212 | (__v32hi)_mm512_shrdi_epi16((A), (B), (I)), \ |
||
| 213 | (__v32hi)_mm512_setzero_si512())) |
||
| 214 | |||
| 215 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 216 | _mm512_shldv_epi64(__m512i __A, __m512i __B, __m512i __C) |
||
| 217 | { |
||
| 218 | return (__m512i)__builtin_ia32_vpshldvq512((__v8di)__A, (__v8di)__B, |
||
| 219 | (__v8di)__C); |
||
| 220 | } |
||
| 221 | |||
| 222 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 223 | _mm512_mask_shldv_epi64(__m512i __A, __mmask8 __U, __m512i __B, __m512i __C) |
||
| 224 | { |
||
| 225 | return (__m512i)__builtin_ia32_selectq_512(__U, |
||
| 226 | (__v8di)_mm512_shldv_epi64(__A, __B, __C), |
||
| 227 | (__v8di)__A); |
||
| 228 | } |
||
| 229 | |||
| 230 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 231 | _mm512_maskz_shldv_epi64(__mmask8 __U, __m512i __A, __m512i __B, __m512i __C) |
||
| 232 | { |
||
| 233 | return (__m512i)__builtin_ia32_selectq_512(__U, |
||
| 234 | (__v8di)_mm512_shldv_epi64(__A, __B, __C), |
||
| 235 | (__v8di)_mm512_setzero_si512()); |
||
| 236 | } |
||
| 237 | |||
| 238 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 239 | _mm512_shldv_epi32(__m512i __A, __m512i __B, __m512i __C) |
||
| 240 | { |
||
| 241 | return (__m512i)__builtin_ia32_vpshldvd512((__v16si)__A, (__v16si)__B, |
||
| 242 | (__v16si)__C); |
||
| 243 | } |
||
| 244 | |||
| 245 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 246 | _mm512_mask_shldv_epi32(__m512i __A, __mmask16 __U, __m512i __B, __m512i __C) |
||
| 247 | { |
||
| 248 | return (__m512i)__builtin_ia32_selectd_512(__U, |
||
| 249 | (__v16si)_mm512_shldv_epi32(__A, __B, __C), |
||
| 250 | (__v16si)__A); |
||
| 251 | } |
||
| 252 | |||
| 253 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 254 | _mm512_maskz_shldv_epi32(__mmask16 __U, __m512i __A, __m512i __B, __m512i __C) |
||
| 255 | { |
||
| 256 | return (__m512i)__builtin_ia32_selectd_512(__U, |
||
| 257 | (__v16si)_mm512_shldv_epi32(__A, __B, __C), |
||
| 258 | (__v16si)_mm512_setzero_si512()); |
||
| 259 | } |
||
| 260 | |||
| 261 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 262 | _mm512_shldv_epi16(__m512i __A, __m512i __B, __m512i __C) |
||
| 263 | { |
||
| 264 | return (__m512i)__builtin_ia32_vpshldvw512((__v32hi)__A, (__v32hi)__B, |
||
| 265 | (__v32hi)__C); |
||
| 266 | } |
||
| 267 | |||
| 268 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 269 | _mm512_mask_shldv_epi16(__m512i __A, __mmask32 __U, __m512i __B, __m512i __C) |
||
| 270 | { |
||
| 271 | return (__m512i)__builtin_ia32_selectw_512(__U, |
||
| 272 | (__v32hi)_mm512_shldv_epi16(__A, __B, __C), |
||
| 273 | (__v32hi)__A); |
||
| 274 | } |
||
| 275 | |||
| 276 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 277 | _mm512_maskz_shldv_epi16(__mmask32 __U, __m512i __A, __m512i __B, __m512i __C) |
||
| 278 | { |
||
| 279 | return (__m512i)__builtin_ia32_selectw_512(__U, |
||
| 280 | (__v32hi)_mm512_shldv_epi16(__A, __B, __C), |
||
| 281 | (__v32hi)_mm512_setzero_si512()); |
||
| 282 | } |
||
| 283 | |||
| 284 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 285 | _mm512_shrdv_epi64(__m512i __A, __m512i __B, __m512i __C) |
||
| 286 | { |
||
| 287 | return (__m512i)__builtin_ia32_vpshrdvq512((__v8di)__A, (__v8di)__B, |
||
| 288 | (__v8di)__C); |
||
| 289 | } |
||
| 290 | |||
| 291 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 292 | _mm512_mask_shrdv_epi64(__m512i __A, __mmask8 __U, __m512i __B, __m512i __C) |
||
| 293 | { |
||
| 294 | return (__m512i)__builtin_ia32_selectq_512(__U, |
||
| 295 | (__v8di)_mm512_shrdv_epi64(__A, __B, __C), |
||
| 296 | (__v8di)__A); |
||
| 297 | } |
||
| 298 | |||
| 299 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 300 | _mm512_maskz_shrdv_epi64(__mmask8 __U, __m512i __A, __m512i __B, __m512i __C) |
||
| 301 | { |
||
| 302 | return (__m512i)__builtin_ia32_selectq_512(__U, |
||
| 303 | (__v8di)_mm512_shrdv_epi64(__A, __B, __C), |
||
| 304 | (__v8di)_mm512_setzero_si512()); |
||
| 305 | } |
||
| 306 | |||
| 307 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 308 | _mm512_shrdv_epi32(__m512i __A, __m512i __B, __m512i __C) |
||
| 309 | { |
||
| 310 | return (__m512i)__builtin_ia32_vpshrdvd512((__v16si)__A, (__v16si)__B, |
||
| 311 | (__v16si)__C); |
||
| 312 | } |
||
| 313 | |||
| 314 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 315 | _mm512_mask_shrdv_epi32(__m512i __A, __mmask16 __U, __m512i __B, __m512i __C) |
||
| 316 | { |
||
| 317 | return (__m512i) __builtin_ia32_selectd_512(__U, |
||
| 318 | (__v16si)_mm512_shrdv_epi32(__A, __B, __C), |
||
| 319 | (__v16si)__A); |
||
| 320 | } |
||
| 321 | |||
| 322 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 323 | _mm512_maskz_shrdv_epi32(__mmask16 __U, __m512i __A, __m512i __B, __m512i __C) |
||
| 324 | { |
||
| 325 | return (__m512i) __builtin_ia32_selectd_512(__U, |
||
| 326 | (__v16si)_mm512_shrdv_epi32(__A, __B, __C), |
||
| 327 | (__v16si)_mm512_setzero_si512()); |
||
| 328 | } |
||
| 329 | |||
| 330 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 331 | _mm512_shrdv_epi16(__m512i __A, __m512i __B, __m512i __C) |
||
| 332 | { |
||
| 333 | return (__m512i)__builtin_ia32_vpshrdvw512((__v32hi)__A, (__v32hi)__B, |
||
| 334 | (__v32hi)__C); |
||
| 335 | } |
||
| 336 | |||
| 337 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 338 | _mm512_mask_shrdv_epi16(__m512i __A, __mmask32 __U, __m512i __B, __m512i __C) |
||
| 339 | { |
||
| 340 | return (__m512i)__builtin_ia32_selectw_512(__U, |
||
| 341 | (__v32hi)_mm512_shrdv_epi16(__A, __B, __C), |
||
| 342 | (__v32hi)__A); |
||
| 343 | } |
||
| 344 | |||
| 345 | static __inline__ __m512i __DEFAULT_FN_ATTRS |
||
| 346 | _mm512_maskz_shrdv_epi16(__mmask32 __U, __m512i __A, __m512i __B, __m512i __C) |
||
| 347 | { |
||
| 348 | return (__m512i)__builtin_ia32_selectw_512(__U, |
||
| 349 | (__v32hi)_mm512_shrdv_epi16(__A, __B, __C), |
||
| 350 | (__v32hi)_mm512_setzero_si512()); |
||
| 351 | } |
||
| 352 | |||
| 353 | |||
| 354 | #undef __DEFAULT_FN_ATTRS |
||
| 355 | |||
| 356 | #endif |
||
| 357 |