avx512bitalgintrin.h 3.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. /*===------------- avx512bitalgintrin.h - BITALG intrinsics ------------------===
  2. *
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a copy
  5. * of this software and associated documentation files (the "Software"), to deal
  6. * in the Software without restriction, including without limitation the rights
  7. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. * copies of the Software, and to permit persons to whom the Software is
  9. * furnished to do so, subject to the following conditions:
  10. *
  11. * The above copyright notice and this permission notice shall be included in
  12. * all copies or substantial portions of the Software.
  13. *
  14. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20. * THE SOFTWARE.
  21. *
  22. *===-----------------------------------------------------------------------===
  23. */
  24. #ifndef __IMMINTRIN_H
  25. #error "Never use <avx512bitalgintrin.h> directly; include <immintrin.h> instead."
  26. #endif
  27. #ifndef __AVX512BITALGINTRIN_H
  28. #define __AVX512BITALGINTRIN_H
  29. /* Define the default attributes for the functions in this file. */
  30. #define __DEFAULT_FN_ATTRS __attribute__((__always_inline__, __nodebug__, __target__("avx512bitalg")))
  31. static __inline__ __m512i __DEFAULT_FN_ATTRS
  32. _mm512_popcnt_epi16(__m512i __A)
  33. {
  34. return (__m512i) __builtin_ia32_vpopcntw_512((__v32hi) __A);
  35. }
  36. static __inline__ __m512i __DEFAULT_FN_ATTRS
  37. _mm512_mask_popcnt_epi16(__m512i __A, __mmask32 __U, __m512i __B)
  38. {
  39. return (__m512i) __builtin_ia32_selectw_512((__mmask32) __U,
  40. (__v32hi) _mm512_popcnt_epi16(__B),
  41. (__v32hi) __A);
  42. }
  43. static __inline__ __m512i __DEFAULT_FN_ATTRS
  44. _mm512_maskz_popcnt_epi16(__mmask32 __U, __m512i __B)
  45. {
  46. return _mm512_mask_popcnt_epi16((__m512i) _mm512_setzero_hi(),
  47. __U,
  48. __B);
  49. }
  50. static __inline__ __m512i __DEFAULT_FN_ATTRS
  51. _mm512_popcnt_epi8(__m512i __A)
  52. {
  53. return (__m512i) __builtin_ia32_vpopcntb_512((__v64qi) __A);
  54. }
  55. static __inline__ __m512i __DEFAULT_FN_ATTRS
  56. _mm512_mask_popcnt_epi8(__m512i __A, __mmask64 __U, __m512i __B)
  57. {
  58. return (__m512i) __builtin_ia32_selectb_512((__mmask64) __U,
  59. (__v64qi) _mm512_popcnt_epi8(__B),
  60. (__v64qi) __A);
  61. }
  62. static __inline__ __m512i __DEFAULT_FN_ATTRS
  63. _mm512_maskz_popcnt_epi8(__mmask64 __U, __m512i __B)
  64. {
  65. return _mm512_mask_popcnt_epi8((__m512i) _mm512_setzero_qi(),
  66. __U,
  67. __B);
  68. }
  69. static __inline__ __mmask64 __DEFAULT_FN_ATTRS
  70. _mm512_mask_bitshuffle_epi64_mask(__mmask64 __U, __m512i __A, __m512i __B)
  71. {
  72. return (__mmask64) __builtin_ia32_vpshufbitqmb512_mask((__v64qi) __A,
  73. (__v64qi) __B,
  74. __U);
  75. }
  76. static __inline__ __mmask64 __DEFAULT_FN_ATTRS
  77. _mm512_bitshuffle_epi64_mask(__m512i __A, __m512i __B)
  78. {
  79. return _mm512_mask_bitshuffle_epi64_mask((__mmask64) -1,
  80. __A,
  81. __B);
  82. }
  83. #undef __DEFAULT_FN_ATTRS
  84. #endif