/ lib / numpy / distutils / checks / cpu_avx512_icl.c
cpu_avx512_icl.c
 1  #if defined(DETECT_FEATURES) && defined(__INTEL_COMPILER)
 2      /*
 3       * Unlike GCC and CLANG, Intel Compiler exposes all supported intrinsics,
 4       * whether or not the build options for those features are specified.
 5       * Therefore, we must test #definitions of CPU features when option native/host
 6       * is enabled via `--cpu-baseline` or through env var `CFLAGS` otherwise
 7       * the test will be broken and leads to enable all possible features.
 8       */
 9      #if !defined(__AVX512VPOPCNTDQ__) || !defined(__AVX512BITALG__) || !defined(__AVX512VPOPCNTDQ__)
10          #error "HOST/ARCH doesn't support IceLake AVX512 features"
11      #endif
12  #endif
13  
14  #include <immintrin.h>
15  
16  int main(int argc, char **argv)
17  {
18      __m512i a = _mm512_loadu_si512((const __m512i*)argv[argc-1]);
19      /* VBMI2 */
20      a = _mm512_shrdv_epi64(a, a, _mm512_setzero_si512());
21      /* BITLAG */
22      a = _mm512_popcnt_epi8(a);
23      /* VPOPCNTDQ */
24      a = _mm512_popcnt_epi64(a);
25      return _mm_cvtsi128_si32(_mm512_castsi512_si128(a));
26  }