/ lib / numpy / distutils / checks / extra_avx512bw_mask.c
extra_avx512bw_mask.c
 1  #include <immintrin.h>
 2  /**
 3   * Test BW mask operations due to:
 4   *  - MSVC has supported it since vs2019 see,
 5   *    https://developercommunity.visualstudio.com/content/problem/518298/missing-avx512bw-mask-intrinsics.html
 6   *  - Clang >= v8.0
 7   *  - GCC >= v7.1
 8   */
 9  int main(void)
10  {
11      __mmask64 m64 = _mm512_cmpeq_epi8_mask(_mm512_set1_epi8((char)1), _mm512_set1_epi8((char)1));
12      m64 = _kor_mask64(m64, m64);
13      m64 = _kxor_mask64(m64, m64);
14      m64 = _cvtu64_mask64(_cvtmask64_u64(m64));
15      m64 = _mm512_kunpackd(m64, m64);
16      m64 = (__mmask64)_mm512_kunpackw((__mmask32)m64, (__mmask32)m64);
17      return (int)_cvtmask64_u64(m64);
18  }