Index: include/clang/Basic/BuiltinsX86.def =================================================================== --- include/clang/Basic/BuiltinsX86.def +++ include/clang/Basic/BuiltinsX86.def @@ -1794,6 +1794,28 @@ TARGET_BUILTIN(__builtin_ia32_loadupd256_mask, "V4dV4d*V4dUc","","avx512vl") TARGET_BUILTIN(__builtin_ia32_loadups128_mask, "V4fV4f*V4fUc","","avx512vl") TARGET_BUILTIN(__builtin_ia32_loadups256_mask, "V8fV8f*V8fUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmb512, "ULLiV64cV64cULLi","","avx512bw") +TARGET_BUILTIN(__builtin_ia32_ptestmw512, "UiV32sV32sUi","","avx512bw") +TARGET_BUILTIN(__builtin_ia32_ptestnmb512, "ULLiV64cV64cULLi","","avx512bw") +TARGET_BUILTIN(__builtin_ia32_ptestnmw512, "UiV32sV32sUi","","avx512bw") +TARGET_BUILTIN(__builtin_ia32_ptestmb128, "UsV16cV16cUs","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmb256, "UiV32cV32cUi","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmw128, "UcV8sV8sUc","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmw256, "UsV16sV16sUs","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmb128, "UsV16cV16cUs","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmb256, "UiV32cV32cUi","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmw128, "UcV8sV8sUc","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmw256, "UsV16sV16sUs","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmd128, "UcV4iV4iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmd256, "UcV8iV8iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmq128, "UcV2LLiV2LLiUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestmq256, "UcV4LLiV4LLiUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmd128, "UcV4iV4iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmd256, "UcV8iV8iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmq128, "UcV2LLiV2LLiUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmq256, "UcV4LLiV4LLiUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_ptestnmd512, "UsV16iV16iUs","","avx512f") +TARGET_BUILTIN(__builtin_ia32_ptestnmq512, "UcV8LLiV8LLiUc","","avx512f") #undef BUILTIN #undef TARGET_BUILTIN Index: lib/Headers/avx512bwintrin.h =================================================================== --- lib/Headers/avx512bwintrin.h +++ lib/Headers/avx512bwintrin.h @@ -1981,6 +1981,67 @@ _mm512_setzero_hi (), (__mmask64) __U); } + +static __inline__ __mmask64 __DEFAULT_FN_ATTRS +_mm512_test_epi8_mask (__m512i __A, __m512i __B) +{ + return (__mmask64) __builtin_ia32_ptestmb512 ((__v64qi) __A, + (__v64qi) __B, + (__mmask64) -1); +} + +static __inline__ __mmask64 __DEFAULT_FN_ATTRS +_mm512_mask_test_epi8_mask (__mmask64 __U, __m512i __A, __m512i __B) +{ + return (__mmask64) __builtin_ia32_ptestmb512 ((__v64qi) __A, + (__v64qi) __B, __U); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm512_test_epi16_mask (__m512i __A, __m512i __B) +{ + return (__mmask32) __builtin_ia32_ptestmw512 ((__v32hi) __A, + (__v32hi) __B, + (__mmask32) -1); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm512_mask_test_epi16_mask (__mmask32 __U, __m512i __A, __m512i __B) +{ + return (__mmask32) __builtin_ia32_ptestmw512 ((__v32hi) __A, + (__v32hi) __B, __U); +} + +static __inline__ __mmask64 __DEFAULT_FN_ATTRS +_mm512_testn_epi8_mask (__m512i __A, __m512i __B) +{ + return (__mmask64) __builtin_ia32_ptestnmb512 ((__v64qi) __A, + (__v64qi) __B, + (__mmask64) -1); +} + +static __inline__ __mmask64 __DEFAULT_FN_ATTRS +_mm512_mask_testn_epi8_mask (__mmask64 __U, __m512i __A, __m512i __B) +{ + return (__mmask64) __builtin_ia32_ptestnmb512 ((__v64qi) __A, + (__v64qi) __B, __U); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm512_testn_epi16_mask (__m512i __A, __m512i __B) +{ + return (__mmask32) __builtin_ia32_ptestnmw512 ((__v32hi) __A, + (__v32hi) __B, + (__mmask32) -1); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm512_mask_testn_epi16_mask (__mmask32 __U, __m512i __A, __m512i __B) +{ + return (__mmask32) __builtin_ia32_ptestnmw512 ((__v32hi) __A, + (__v32hi) __B, __U); +} + #undef __DEFAULT_FN_ATTRS #endif Index: lib/Headers/avx512fintrin.h =================================================================== --- lib/Headers/avx512fintrin.h +++ lib/Headers/avx512fintrin.h @@ -3974,6 +3974,65 @@ __builtin_ia32_vcomiss ((__v4sf) (__A), (__v4sf) (__B), ( __P), ( __R));\ }) +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm512_testn_epi32_mask (__m512i __A, __m512i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmd512 ((__v16si) __A, + (__v16si) __B, + (__mmask16) -1); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm512_mask_testn_epi32_mask (__mmask16 __U, __m512i __A, __m512i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmd512 ((__v16si) __A, + (__v16si) __B, __U); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm512_testn_epi32_mask (__m512i __A, __m512i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmd512 ((__v16si) __A, + (__v16si) __B, + (__mmask16) -1); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm512_mask_testn_epi32_mask (__mmask16 __U, __m512i __A, __m512i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmd512 ((__v16si) __A, + (__v16si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm512_testn_epi64_mask (__m512i __A, __m512i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq512 ((__v8di) __A, + (__v8di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm512_mask_testn_epi64_mask (__mmask8 __U, __m512i __A, __m512i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq512 ((__v8di) __A, + (__v8di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm512_testn_epi64_mask (__m512i __A, __m512i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq512 ((__v8di) __A, + (__v8di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm512_mask_testn_epi64_mask (__mmask8 __U, __m512i __A, __m512i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq512 ((__v8di) __A, + (__v8di) __B, __U); +} #undef __DEFAULT_FN_ATTRS Index: lib/Headers/avx512vlbwintrin.h =================================================================== --- lib/Headers/avx512vlbwintrin.h +++ lib/Headers/avx512vlbwintrin.h @@ -3020,6 +3020,127 @@ _mm256_setzero_si256 (), (__mmask32) __U); } + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm_test_epi8_mask (__m128i __A, __m128i __B) +{ + return (__mmask16) __builtin_ia32_ptestmb128 ((__v16qi) __A, + (__v16qi) __B, + (__mmask16) -1); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm_mask_test_epi8_mask (__mmask16 __U, __m128i __A, __m128i __B) +{ + return (__mmask16) __builtin_ia32_ptestmb128 ((__v16qi) __A, + (__v16qi) __B, __U); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm256_test_epi8_mask (__m256i __A, __m256i __B) +{ + return (__mmask32) __builtin_ia32_ptestmb256 ((__v32qi) __A, + (__v32qi) __B, + (__mmask32) -1); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm256_mask_test_epi8_mask (__mmask32 __U, __m256i __A, __m256i __B) +{ + return (__mmask32) __builtin_ia32_ptestmb256 ((__v32qi) __A, + (__v32qi) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_test_epi16_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmw128 ((__v8hi) __A, + (__v8hi) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_test_epi16_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmw128 ((__v8hi) __A, + (__v8hi) __B, __U); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm256_test_epi16_mask (__m256i __A, __m256i __B) +{ + return (__mmask16) __builtin_ia32_ptestmw256 ((__v16hi) __A, + (__v16hi) __B, + (__mmask16) -1); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm256_mask_test_epi16_mask (__mmask16 __U, __m256i __A, __m256i __B) +{ + return (__mmask16) __builtin_ia32_ptestmw256 ((__v16hi) __A, + (__v16hi) __B, __U); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm_testn_epi8_mask (__m128i __A, __m128i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmb128 ((__v16qi) __A, + (__v16qi) __B, + (__mmask16) -1); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm_mask_testn_epi8_mask (__mmask16 __U, __m128i __A, __m128i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmb128 ((__v16qi) __A, + (__v16qi) __B, __U); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm256_testn_epi8_mask (__m256i __A, __m256i __B) +{ + return (__mmask32) __builtin_ia32_ptestnmb256 ((__v32qi) __A, + (__v32qi) __B, + (__mmask32) -1); +} + +static __inline__ __mmask32 __DEFAULT_FN_ATTRS +_mm256_mask_testn_epi8_mask (__mmask32 __U, __m256i __A, __m256i __B) +{ + return (__mmask32) __builtin_ia32_ptestnmb256 ((__v32qi) __A, + (__v32qi) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_testn_epi16_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmw128 ((__v8hi) __A, + (__v8hi) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_testn_epi16_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmw128 ((__v8hi) __A, + (__v8hi) __B, __U); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm256_testn_epi16_mask (__m256i __A, __m256i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmw256 ((__v16hi) __A, + (__v16hi) __B, + (__mmask16) -1); +} + +static __inline__ __mmask16 __DEFAULT_FN_ATTRS +_mm256_mask_testn_epi16_mask (__mmask16 __U, __m256i __A, __m256i __B) +{ + return (__mmask16) __builtin_ia32_ptestnmw256 ((__v16hi) __A, + (__v16hi) __B, __U); +} + #undef __DEFAULT_FN_ATTRS #endif /* __AVX512VLBWINTRIN_H */ Index: lib/Headers/avx512vlintrin.h =================================================================== --- lib/Headers/avx512vlintrin.h +++ lib/Headers/avx512vlintrin.h @@ -6360,6 +6360,247 @@ _mm256_setzero_ps (), (__mmask8) __U); } + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_test_epi32_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd128 ((__v4si) __A, + (__v4si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_test_epi32_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd128 ((__v4si) __A, + (__v4si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_test_epi32_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd128 ((__v4si) __A, + (__v4si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_test_epi32_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd128 ((__v4si) __A, + (__v4si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_test_epi32_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd256 ((__v8si) __A, + (__v8si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_test_epi32_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd256 ((__v8si) __A, + (__v8si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_test_epi32_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd256 ((__v8si) __A, + (__v8si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_test_epi32_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmd256 ((__v8si) __A, + (__v8si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_test_epi64_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq128 ((__v2di) __A, + (__v2di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_test_epi64_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq128 ((__v2di) __A, + (__v2di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_test_epi64_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq128 ((__v2di) __A, + (__v2di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_test_epi64_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq128 ((__v2di) __A, + (__v2di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_test_epi64_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq256 ((__v4di) __A, + (__v4di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_test_epi64_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq256 ((__v4di) __A, + (__v4di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_test_epi64_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq256 ((__v4di) __A, + (__v4di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_test_epi64_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestmq256 ((__v4di) __A, + (__v4di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_testn_epi32_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd128 ((__v4si) __A, + (__v4si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_testn_epi32_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd128 ((__v4si) __A, + (__v4si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_testn_epi32_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd128 ((__v4si) __A, + (__v4si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_testn_epi32_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd128 ((__v4si) __A, + (__v4si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_testn_epi32_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd256 ((__v8si) __A, + (__v8si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_testn_epi32_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd256 ((__v8si) __A, + (__v8si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_testn_epi32_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd256 ((__v8si) __A, + (__v8si) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_testn_epi32_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmd256 ((__v8si) __A, + (__v8si) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_testn_epi64_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq128 ((__v2di) __A, + (__v2di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_testn_epi64_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq128 ((__v2di) __A, + (__v2di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_testn_epi64_mask (__m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq128 ((__v2di) __A, + (__v2di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm_mask_testn_epi64_mask (__mmask8 __U, __m128i __A, __m128i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq128 ((__v2di) __A, + (__v2di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_testn_epi64_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq256 ((__v4di) __A, + (__v4di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_testn_epi64_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq256 ((__v4di) __A, + (__v4di) __B, __U); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_testn_epi64_mask (__m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq256 ((__v4di) __A, + (__v4di) __B, + (__mmask8) -1); +} + +static __inline__ __mmask8 __DEFAULT_FN_ATTRS +_mm256_mask_testn_epi64_mask (__mmask8 __U, __m256i __A, __m256i __B) +{ + return (__mmask8) __builtin_ia32_ptestnmq256 ((__v4di) __A, + (__v4di) __B, __U); +} + #undef __DEFAULT_FN_ATTRS #undef __DEFAULT_FN_ATTRS_BOTH Index: test/CodeGen/avx512bw-builtins.c =================================================================== --- test/CodeGen/avx512bw-builtins.c +++ test/CodeGen/avx512bw-builtins.c @@ -1346,5 +1346,51 @@ // CHECK: @llvm.x86.avx512.mask.loadu.b.512 return _mm512_maskz_loadu_epi8(__U, __P); } +__mmask64 test_mm512_test_epi8_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_test_epi8_mask + // CHECK: @llvm.x86.avx512.ptestm.b.512 + return _mm512_test_epi8_mask(__A, __B); +} + +__mmask64 test_mm512_mask_test_epi8_mask(__mmask64 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_test_epi8_mask + // CHECK: @llvm.x86.avx512.ptestm.b.512 + return _mm512_mask_test_epi8_mask(__U, __A, __B); +} + +__mmask32 test_mm512_test_epi16_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_test_epi16_mask + // CHECK: @llvm.x86.avx512.ptestm.w. + return _mm512_test_epi16_mask(__A, __B); +} + +__mmask32 test_mm512_mask_test_epi16_mask(__mmask32 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_test_epi16_mask + // CHECK: @llvm.x86.avx512.ptestm.w. + return _mm512_mask_test_epi16_mask(__U, __A, __B); +} +__mmask64 test_mm512_testn_epi8_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_testn_epi8_mask + // CHECK: @llvm.x86.avx512.ptestnm.b. + return _mm512_testn_epi8_mask(__A, __B); +} + +__mmask64 test_mm512_mask_testn_epi8_mask(__mmask64 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_testn_epi8_mask + // CHECK: @llvm.x86.avx512.ptestnm.b. + return _mm512_mask_testn_epi8_mask(__U, __A, __B); +} + +__mmask32 test_mm512_testn_epi16_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_testn_epi16_mask + // CHECK: @llvm.x86.avx512.ptestnm.w. + return _mm512_testn_epi16_mask(__A, __B); +} + +__mmask32 test_mm512_mask_testn_epi16_mask(__mmask32 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_testn_epi16_mask + // CHECK: @llvm.x86.avx512.ptestnm.w. + return _mm512_mask_testn_epi16_mask(__U, __A, __B); +} Index: test/CodeGen/avx512f-builtins.c =================================================================== --- test/CodeGen/avx512f-builtins.c +++ test/CodeGen/avx512f-builtins.c @@ -2576,5 +2576,53 @@ return _mm512_kmov(__A); } +__mmask16 test_mm512_testn_epi32_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.512 + return _mm512_testn_epi32_mask(__A, __B); +} + +__mmask16 test_mm512_mask_testn_epi32_mask(__mmask16 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.512 + return _mm512_mask_testn_epi32_mask(__U, __A, __B); +} + +__mmask16 test_mm512_testn_epi32_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.512 + return _mm512_testn_epi32_mask(__A, __B); +} + +__mmask16 test_mm512_mask_testn_epi32_mask(__mmask16 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.512 + return _mm512_mask_testn_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm512_testn_epi64_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.512 + return _mm512_testn_epi64_mask(__A, __B); +} + +__mmask8 test_mm512_mask_testn_epi64_mask(__mmask8 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.512 + return _mm512_mask_testn_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm512_testn_epi64_mask(__m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.512 + return _mm512_testn_epi64_mask(__A, __B); +} + +__mmask8 test_mm512_mask_testn_epi64_mask(__mmask8 __U, __m512i __A, __m512i __B) { + // CHECK-LABEL: @test_mm512_mask_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.512 + return _mm512_mask_testn_epi64_mask(__U, __A, __B); +} + Index: test/CodeGen/avx512vl-builtins.c =================================================================== --- test/CodeGen/avx512vl-builtins.c +++ test/CodeGen/avx512vl-builtins.c @@ -4296,3 +4296,195 @@ return _mm256_maskz_loadu_ps(__U, __P); } +__mmask8 test_mm_test_epi32_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.128 + return _mm_test_epi32_mask(__A, __B); +} + +__mmask8 test_mm_mask_test_epi32_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.128 + return _mm_mask_test_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm_test_epi32_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.128 + return _mm_test_epi32_mask(__A, __B); +} + +__mmask8 test_mm_mask_test_epi32_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.128 + return _mm_mask_test_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm256_test_epi32_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.256 + return _mm256_test_epi32_mask(__A, __B); +} + +__mmask8 test_mm256_mask_test_epi32_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.256 + return _mm256_mask_test_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm256_test_epi32_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.256 + return _mm256_test_epi32_mask(__A, __B); +} + +__mmask8 test_mm256_mask_test_epi32_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_test_epi32_mask + // CHECK: @llvm.x86.avx512.ptestm.d.256 + return _mm256_mask_test_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm_test_epi64_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.128 + return _mm_test_epi64_mask(__A, __B); +} + +__mmask8 test_mm_mask_test_epi64_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.128 + return _mm_mask_test_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm_test_epi64_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.128 + return _mm_test_epi64_mask(__A, __B); +} + +__mmask8 test_mm_mask_test_epi64_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.128 + return _mm_mask_test_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm256_test_epi64_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.256 + return _mm256_test_epi64_mask(__A, __B); +} + +__mmask8 test_mm256_mask_test_epi64_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.256 + return _mm256_mask_test_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm256_test_epi64_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.256 + return _mm256_test_epi64_mask(__A, __B); +} + +__mmask8 test_mm256_mask_test_epi64_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_test_epi64_mask + // CHECK: @llvm.x86.avx512.ptestm.q.256 + return _mm256_mask_test_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm_testn_epi32_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.128 + return _mm_testn_epi32_mask(__A, __B); +} + +__mmask8 test_mm_mask_testn_epi32_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.128 + return _mm_mask_testn_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm_testn_epi32_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.128 + return _mm_testn_epi32_mask(__A, __B); +} + +__mmask8 test_mm_mask_testn_epi32_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.128 + return _mm_mask_testn_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm256_testn_epi32_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.256 + return _mm256_testn_epi32_mask(__A, __B); +} + +__mmask8 test_mm256_mask_testn_epi32_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.256 + return _mm256_mask_testn_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm256_testn_epi32_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.256 + return _mm256_testn_epi32_mask(__A, __B); +} + +__mmask8 test_mm256_mask_testn_epi32_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_testn_epi32_mask + // CHECK: @llvm.x86.avx512.ptestnm.d.256 + return _mm256_mask_testn_epi32_mask(__U, __A, __B); +} + +__mmask8 test_mm_testn_epi64_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.128 + return _mm_testn_epi64_mask(__A, __B); +} + +__mmask8 test_mm_mask_testn_epi64_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.128 + return _mm_mask_testn_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm_testn_epi64_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.128 + return _mm_testn_epi64_mask(__A, __B); +} + +__mmask8 test_mm_mask_testn_epi64_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.128 + return _mm_mask_testn_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm256_testn_epi64_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.256 + return _mm256_testn_epi64_mask(__A, __B); +} + +__mmask8 test_mm256_mask_testn_epi64_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.256 + return _mm256_mask_testn_epi64_mask(__U, __A, __B); +} + +__mmask8 test_mm256_testn_epi64_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.256 + return _mm256_testn_epi64_mask(__A, __B); +} + +__mmask8 test_mm256_mask_testn_epi64_mask(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_testn_epi64_mask + // CHECK: @llvm.x86.avx512.ptestnm.q.256 + return _mm256_mask_testn_epi64_mask(__U, __A, __B); +} + Index: test/CodeGen/avx512vlbw-builtins.c =================================================================== --- test/CodeGen/avx512vlbw-builtins.c +++ test/CodeGen/avx512vlbw-builtins.c @@ -2053,3 +2053,99 @@ return _mm256_maskz_loadu_epi8(__U, __P); } +__mmask16 test_mm_test_epi8_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_test_epi8_mask + // CHECK: @llvm.x86.avx512.ptestm.b.128 + return _mm_test_epi8_mask(__A, __B); +} + +__mmask16 test_mm_mask_test_epi8_mask(__mmask16 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_test_epi8_mask + // CHECK: @llvm.x86.avx512.ptestm.b.128 + return _mm_mask_test_epi8_mask(__U, __A, __B); +} + +__mmask32 test_mm256_test_epi8_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_test_epi8_mask + // CHECK: @llvm.x86.avx512.ptestm.b.256 + return _mm256_test_epi8_mask(__A, __B); +} + +__mmask32 test_mm256_mask_test_epi8_mask(__mmask32 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_test_epi8_mask + // CHECK: @llvm.x86.avx512.ptestm.b.256 + return _mm256_mask_test_epi8_mask(__U, __A, __B); +} + +__mmask8 test_mm_test_epi16_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_test_epi16_mask + // CHECK: @llvm.x86.avx512.ptestm.w.128 + return _mm_test_epi16_mask(__A, __B); +} + +__mmask8 test_mm_mask_test_epi16_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_test_epi16_mask + // CHECK: @llvm.x86.avx512.ptestm.w.128 + return _mm_mask_test_epi16_mask(__U, __A, __B); +} + +__mmask16 test_mm256_test_epi16_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_test_epi16_mask + // CHECK: @llvm.x86.avx512.ptestm.w.256 + return _mm256_test_epi16_mask(__A, __B); +} + +__mmask16 test_mm256_mask_test_epi16_mask(__mmask16 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_test_epi16_mask + // CHECK: @llvm.x86.avx512.ptestm.w.256 + return _mm256_mask_test_epi16_mask(__U, __A, __B); +} + +__mmask16 test_mm_testn_epi8_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_testn_epi8_mask + // CHECK: @llvm.x86.avx512.ptestnm.b.128 + return _mm_testn_epi8_mask(__A, __B); +} + +__mmask16 test_mm_mask_testn_epi8_mask(__mmask16 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_testn_epi8_mask + // CHECK: @llvm.x86.avx512.ptestnm.b.128 + return _mm_mask_testn_epi8_mask(__U, __A, __B); +} + +__mmask32 test_mm256_testn_epi8_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_testn_epi8_mask + // CHECK: @llvm.x86.avx512.ptestnm.b.256 + return _mm256_testn_epi8_mask(__A, __B); +} + +__mmask32 test_mm256_mask_testn_epi8_mask(__mmask32 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_testn_epi8_mask + // CHECK: @llvm.x86.avx512.ptestnm.b.256 + return _mm256_mask_testn_epi8_mask(__U, __A, __B); +} + +__mmask8 test_mm_testn_epi16_mask(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_testn_epi16_mask + // CHECK: @llvm.x86.avx512.ptestnm.w.128 + return _mm_testn_epi16_mask(__A, __B); +} + +__mmask8 test_mm_mask_testn_epi16_mask(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_testn_epi16_mask + // CHECK: @llvm.x86.avx512.ptestnm.w.128 + return _mm_mask_testn_epi16_mask(__U, __A, __B); +} + +__mmask16 test_mm256_testn_epi16_mask(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_testn_epi16_mask + // CHECK: @llvm.x86.avx512.ptestnm.w.256 + return _mm256_testn_epi16_mask(__A, __B); +} + +__mmask16 test_mm256_mask_testn_epi16_mask(__mmask16 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_testn_epi16_mask + // CHECK: @llvm.x86.avx512.ptestnm.w.256 + return _mm256_mask_testn_epi16_mask(__U, __A, __B); +} +