master
  1/*===---- avx512vpopcntdqintrin.h - AVX512VPOPCNTDQ 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                                                                         \
 12    "Never use <avx512vpopcntdqvlintrin.h> directly; include <immintrin.h> instead."
 13#endif
 14
 15#ifndef __AVX512VPOPCNTDQVLINTRIN_H
 16#define __AVX512VPOPCNTDQVLINTRIN_H
 17
 18/* Define the default attributes for the functions in this file. */
 19#define __DEFAULT_FN_ATTRS128                                                  \
 20  __attribute__((__always_inline__, __nodebug__,                               \
 21                 __target__("avx512vpopcntdq,avx512vl,no-evex512"),            \
 22                 __min_vector_width__(128)))
 23#define __DEFAULT_FN_ATTRS256                                                  \
 24  __attribute__((__always_inline__, __nodebug__,                               \
 25                 __target__("avx512vpopcntdq,avx512vl,no-evex512"),            \
 26                 __min_vector_width__(256)))
 27
 28#if defined(__cplusplus) && (__cplusplus >= 201103L)
 29#define __DEFAULT_FN_ATTRS128_CONSTEXPR __DEFAULT_FN_ATTRS128 constexpr
 30#define __DEFAULT_FN_ATTRS256_CONSTEXPR __DEFAULT_FN_ATTRS256 constexpr
 31#else
 32#define __DEFAULT_FN_ATTRS128_CONSTEXPR __DEFAULT_FN_ATTRS128
 33#define __DEFAULT_FN_ATTRS256_CONSTEXPR __DEFAULT_FN_ATTRS256
 34#endif
 35
 36static __inline__ __m128i __DEFAULT_FN_ATTRS128_CONSTEXPR
 37_mm_popcnt_epi64(__m128i __A) {
 38  return (__m128i)__builtin_elementwise_popcount((__v2du)__A);
 39}
 40
 41static __inline__ __m128i __DEFAULT_FN_ATTRS128
 42_mm_mask_popcnt_epi64(__m128i __W, __mmask8 __U, __m128i __A) {
 43  return (__m128i)__builtin_ia32_selectq_128(
 44      (__mmask8)__U, (__v2di)_mm_popcnt_epi64(__A), (__v2di)__W);
 45}
 46
 47static __inline__ __m128i __DEFAULT_FN_ATTRS128
 48_mm_maskz_popcnt_epi64(__mmask8 __U, __m128i __A) {
 49  return _mm_mask_popcnt_epi64((__m128i)_mm_setzero_si128(), __U, __A);
 50}
 51
 52static __inline__ __m128i __DEFAULT_FN_ATTRS128_CONSTEXPR
 53_mm_popcnt_epi32(__m128i __A) {
 54  return (__m128i)__builtin_elementwise_popcount((__v4su)__A);
 55}
 56
 57static __inline__ __m128i __DEFAULT_FN_ATTRS128
 58_mm_mask_popcnt_epi32(__m128i __W, __mmask8 __U, __m128i __A) {
 59  return (__m128i)__builtin_ia32_selectd_128(
 60      (__mmask8)__U, (__v4si)_mm_popcnt_epi32(__A), (__v4si)__W);
 61}
 62
 63static __inline__ __m128i __DEFAULT_FN_ATTRS128
 64_mm_maskz_popcnt_epi32(__mmask8 __U, __m128i __A) {
 65  return _mm_mask_popcnt_epi32((__m128i)_mm_setzero_si128(), __U, __A);
 66}
 67
 68static __inline__ __m256i __DEFAULT_FN_ATTRS256_CONSTEXPR
 69_mm256_popcnt_epi64(__m256i __A) {
 70  return (__m256i)__builtin_elementwise_popcount((__v4du)__A);
 71}
 72
 73static __inline__ __m256i __DEFAULT_FN_ATTRS256
 74_mm256_mask_popcnt_epi64(__m256i __W, __mmask8 __U, __m256i __A) {
 75  return (__m256i)__builtin_ia32_selectq_256(
 76      (__mmask8)__U, (__v4di)_mm256_popcnt_epi64(__A), (__v4di)__W);
 77}
 78
 79static __inline__ __m256i __DEFAULT_FN_ATTRS256
 80_mm256_maskz_popcnt_epi64(__mmask8 __U, __m256i __A) {
 81  return _mm256_mask_popcnt_epi64((__m256i)_mm256_setzero_si256(), __U, __A);
 82}
 83
 84static __inline__ __m256i __DEFAULT_FN_ATTRS256_CONSTEXPR
 85_mm256_popcnt_epi32(__m256i __A) {
 86  return (__m256i)__builtin_elementwise_popcount((__v8su)__A);
 87}
 88
 89static __inline__ __m256i __DEFAULT_FN_ATTRS256
 90_mm256_mask_popcnt_epi32(__m256i __W, __mmask8 __U, __m256i __A) {
 91  return (__m256i)__builtin_ia32_selectd_256(
 92      (__mmask8)__U, (__v8si)_mm256_popcnt_epi32(__A), (__v8si)__W);
 93}
 94
 95static __inline__ __m256i __DEFAULT_FN_ATTRS256
 96_mm256_maskz_popcnt_epi32(__mmask8 __U, __m256i __A) {
 97  return _mm256_mask_popcnt_epi32((__m256i)_mm256_setzero_si256(), __U, __A);
 98}
 99
100#undef __DEFAULT_FN_ATTRS128
101#undef __DEFAULT_FN_ATTRS256
102
103#endif