diff --git a/clang/include/clang/Basic/BuiltinsX86.def b/clang/include/clang/Basic/BuiltinsX86.def index 8020f06edc3b..4a58c91dcd1f 100644 --- a/clang/include/clang/Basic/BuiltinsX86.def +++ b/clang/include/clang/Basic/BuiltinsX86.def @@ -1058,6 +1058,10 @@ TARGET_BUILTIN(__builtin_ia32_vpermt2varps512_mask, "V16fV16iV16fV16fUs", "", "a TARGET_BUILTIN(__builtin_ia32_vpermt2varpd512_mask, "V8dV8LLiV8dV8dUc", "", "avx512f") TARGET_BUILTIN(__builtin_ia32_alignq512_mask, "V8LLiV8LLiV8LLiIiV8LLiUc", "", "avx512f") TARGET_BUILTIN(__builtin_ia32_alignd512_mask, "V16iV16iV16iIiV16iUs", "", "avx512f") +TARGET_BUILTIN(__builtin_ia32_alignd128_mask, "V4iV4iV4iIiV4iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_alignd256_mask, "V8iV8iV8iIiV8iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_alignq128_mask, "V2LLiV2LLiV2LLiIiV2LLiUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_alignq256_mask, "V4LLiV4LLiV4LLiIiV4LLiUc","","avx512vl") TARGET_BUILTIN(__builtin_ia32_extractf64x4_mask, "V4dV8dIiV4dUc", "", "avx512f") TARGET_BUILTIN(__builtin_ia32_extractf32x4_mask, "V4fV16fIiV4fUc", "", "avx512f") @@ -2207,6 +2211,10 @@ TARGET_BUILTIN(__builtin_ia32_movntdq512, "vV8LLi*V8LLi","","avx512f") TARGET_BUILTIN(__builtin_ia32_movntdqa512, "V8LLiV8LLi*","","avx512f") TARGET_BUILTIN(__builtin_ia32_movntpd512, "vd*V8d","","avx512f") TARGET_BUILTIN(__builtin_ia32_movntps512, "vf*V16f","","avx512f") +TARGET_BUILTIN(__builtin_ia32_palignr512_mask, "V64cV64cV64ciV64cULLi","","avx512bw") +TARGET_BUILTIN(__builtin_ia32_palignr128_mask, "V16cV16cV16ciV16cUs","","avx512bw,avx512vl") +TARGET_BUILTIN(__builtin_ia32_palignr256_mask, "V32cV32cV32ciV32cUi","","avx512bw,avx512vl") + #undef BUILTIN #undef TARGET_BUILTIN diff --git a/clang/lib/Headers/avx512bwintrin.h b/clang/lib/Headers/avx512bwintrin.h index e0307cdcf867..4f451df3f869 100644 --- a/clang/lib/Headers/avx512bwintrin.h +++ b/clang/lib/Headers/avx512bwintrin.h @@ -2168,6 +2168,29 @@ _mm512_mask_permutexvar_epi16 (__m512i __W, __mmask32 __M, __m512i __A, (__mmask32) __M); } +#define _mm512_alignr_epi8( __A, __B, __N) __extension__ ({\ +__builtin_ia32_palignr512_mask ((__v8di) __A,\ + (__v8di) __B ,__N * 8,\ + (__v8di) _mm512_undefined_pd (),\ + (__mmask64) -1);\ +}) + +#define _mm512_mask_alignr_epi8( __W, __U, __A, __B, __N) __extension__({\ +__builtin_ia32_palignr512_mask ((__v8di) __A,\ + (__v8di) __B,\ + __N * 8,\ + (__v8di) __W,\ + (__mmask64) __U);\ +}) + +#define _mm512_maskz_alignr_epi8( __U, __A, __B, __N) __extension__({\ +__builtin_ia32_palignr512_mask ((__v8di) __A,\ + (__v8di) __B,\ + __N * 8,\ + (__v8di) _mm512_setzero_si512 (),\ + (__mmask64) __U);\ +}) + #undef __DEFAULT_FN_ATTRS #endif diff --git a/clang/lib/Headers/avx512fintrin.h b/clang/lib/Headers/avx512fintrin.h index 38d2ccb52af3..844025836a48 100644 --- a/clang/lib/Headers/avx512fintrin.h +++ b/clang/lib/Headers/avx512fintrin.h @@ -2550,12 +2550,40 @@ _mm512_permutex2var_ps(__m512 __A, __m512i __I, __m512 __B) (I), (__v8di)_mm512_setzero_si512(), \ (__mmask8)-1); }) +#define _mm512_mask_alignr_epi64( __W, __U, __A, __B, __imm) __extension__({\ + (__m512i)__builtin_ia32_alignq512_mask ((__v8di) __A,\ + (__v8di) __B, __imm,\ + (__v8di) __W,\ + (__mmask8) __U);\ +}) + +#define _mm512_maskz_alignr_epi64( __U, __A, __B, __imm) __extension__({\ + (__m512i)__builtin_ia32_alignq512_mask ((__v8di) __A,\ + (__v8di) __B, __imm,\ + (__v8di) _mm512_setzero_si512 (),\ + (__mmask8) __U);\ +}) + #define _mm512_alignr_epi32(A, B, I) __extension__ ({ \ - (__m512i)__builtin_ia32_alignd512_mask((__v16si)(__m512i)(A), \ + (__m512i)__builtin_ia32_alignd512_mask((__v16si)(__m512i)(A), \ (__v16si)(__m512i)(B), \ (I), (__v16si)_mm512_setzero_si512(), \ - (__mmask16)-1); }) + (__mmask16)-1);\ +}) + +#define _mm512_mask_alignr_epi32( __W, __U, __A, __B, __imm) __extension__ ({\ + (__m512i) __builtin_ia32_alignd512_mask((__v16si) __A,\ + (__v16si) __B, __imm,\ + (__v16si) __W,\ + (__mmask16) __U);\ +}) +#define _mm512_maskz_alignr_epi32( __U, __A, __B, __imm) __extension__({\ + (__m512i) __builtin_ia32_alignd512_mask ((__v16si) __A,\ + (__v16si) __B, __imm,\ + (__v16si) _mm512_setzero_si512 (),\ + (__mmask16) __U);\ +}) /* Vector Extract */ #define _mm512_extractf64x4_pd(A, I) __extension__ ({ \ diff --git a/clang/lib/Headers/avx512vlbwintrin.h b/clang/lib/Headers/avx512vlbwintrin.h index 361df9391163..bee20aa183e7 100644 --- a/clang/lib/Headers/avx512vlbwintrin.h +++ b/clang/lib/Headers/avx512vlbwintrin.h @@ -3358,6 +3358,40 @@ _mm256_mask_permutexvar_epi16 (__m256i __W, __mmask16 __M, __m256i __A, (__mmask16) __M); } +#define _mm_mask_alignr_epi8( __W, __U, __A, __B, __N) __extension__ ({ \ +__builtin_ia32_palignr128_mask ((__v2di)( __A),\ + (__v2di)( __B),\ + ( __N) * 8,\ + (__v2di)( __W),\ + (__mmask16)( __U));\ +}) + +#define _mm_maskz_alignr_epi8( __U, __A, __B, __N) __extension__ ({ \ +__builtin_ia32_palignr128_mask ((__v2di)( __A),\ + (__v2di)( __B),\ + ( __N) * 8,\ + (__v2di)\ + _mm_setzero_si128 (),\ + (__mmask16)( __U));\ +}) + +#define _mm256_mask_alignr_epi8( __W, __U, __A, __B, __N) __extension__ ({ \ +__builtin_ia32_palignr256_mask ((__v4di)( __A),\ + (__v4di)( __B),\ + ( __N) * 8,\ + (__v4di)( __W),\ + (__mmask32)( __U));\ +}) + +#define _mm256_maskz_alignr_epi8( __U, __A, __B, __N) __extension__ ({ \ +__builtin_ia32_palignr256_mask ((__v4di)( __A),\ + (__v4di)( __B),\ + ( __N) * 8,\ + (__v4di)\ + _mm256_setzero_si256 (),\ + (__mmask32)( __U));\ +}) + #undef __DEFAULT_FN_ATTRS #endif /* __AVX512VLBWINTRIN_H */ diff --git a/clang/lib/Headers/avx512vlintrin.h b/clang/lib/Headers/avx512vlintrin.h index 77d98b887ff5..60c2fbec8ec3 100644 --- a/clang/lib/Headers/avx512vlintrin.h +++ b/clang/lib/Headers/avx512vlintrin.h @@ -9209,6 +9209,90 @@ _mm256_permutexvar_epi32 (__m256i __X, __m256i __Y) (__mmask8) -1); } +#define _mm_alignr_epi32( __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignd128_mask ((__v4si)( __A),\ + (__v4si)( __B),( __imm),\ + (__v4si) _mm_undefined_si128 (),\ + (__mmask8) -1);\ +}) + +#define _mm_mask_alignr_epi32( __W, __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignd128_mask ((__v4si)( __A),\ + (__v4si)( __B),( __imm),\ + (__v4si)( __W),\ + (__mmask8)( __U));\ +}) + +#define _mm_maskz_alignr_epi32( __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignd128_mask ((__v4si)( __A),\ + (__v4si)( __B),( __imm),\ + (__v4si) _mm_setzero_si128 (),\ + (__mmask8)( __U));\ +}) + +#define _mm256_alignr_epi32( __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignd256_mask ((__v8si)( __A),\ + (__v8si)( __B),( __imm),\ + (__v8si) _mm256_undefined_si256 (),\ + (__mmask8) -1);\ +}) + +#define _mm256_mask_alignr_epi32( __W, __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignd256_mask ((__v8si)( __A),\ + (__v8si)( __B),( __imm),\ + (__v8si)( __W),\ + (__mmask8)( __U));\ +}) + +#define _mm256_maskz_alignr_epi32( __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignd256_mask ((__v8si)( __A),\ + (__v8si)( __B),( __imm),\ + (__v8si) _mm256_setzero_si256 (),\ + (__mmask8)( __U));\ +}) + +#define _mm_alignr_epi64( __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignq128_mask ((__v2di)( __A),\ + (__v2di)( __B),( __imm),\ + (__v2di) _mm_setzero_di (),\ + (__mmask8) -1);\ +}) + +#define _mm_mask_alignr_epi64( __W, __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignq128_mask ((__v2di)( __A),\ + (__v2di)( __B),( __imm),\ + (__v2di)( __W),\ + (__mmask8)( __U));\ +}) + +#define _mm_maskz_alignr_epi64( __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignq128_mask ((__v2di)( __A),\ + (__v2di)( __B),( __imm),\ + (__v2di) _mm_setzero_di (),\ + (__mmask8)( __U));\ +}) + +#define _mm256_alignr_epi64( __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignq256_mask ((__v4di)( __A),\ + (__v4di)( __B),( __imm),\ + (__v4di) _mm256_undefined_pd (),\ + (__mmask8) -1);\ +}) + +#define _mm256_mask_alignr_epi64( __W, __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignq256_mask ((__v4di)( __A),\ + (__v4di)( __B),( __imm),\ + (__v4di)( __W),\ + (__mmask8)( __U));\ +}) + +#define _mm256_maskz_alignr_epi64( __U, __A, __B, __imm) __extension__ ({ \ +__builtin_ia32_alignq256_mask ((__v4di)( __A),\ + (__v4di)( __B),( __imm),\ + (__v4di) _mm256_setzero_si256 (),\ + (__mmask8)( __U));\ +}) + #undef __DEFAULT_FN_ATTRS #undef __DEFAULT_FN_ATTRS_BOTH diff --git a/clang/test/CodeGen/avx512bw-builtins.c b/clang/test/CodeGen/avx512bw-builtins.c index 3023e6060044..8925cbadc064 100644 --- a/clang/test/CodeGen/avx512bw-builtins.c +++ b/clang/test/CodeGen/avx512bw-builtins.c @@ -1487,3 +1487,23 @@ __m512i test_mm512_mask_permutexvar_epi16(__m512i __W, __mmask32 __M, __m512i __ // CHECK: @llvm.x86.avx512.mask.permvar.hi.512 return _mm512_mask_permutexvar_epi16(__W, __M, __A, __B); } +__m512i test_mm512_alignr_epi8(__m512i __A,__m512i __B){ + // CHECK-LABEL: @test_mm512_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.512 + return _mm512_alignr_epi8(__A, __B, 2); +} + +__m512i test_mm512_mask_alignr_epi8(__m512i __W, __mmask64 __U, __m512i __A,__m512i __B){ + // CHECK-LABEL: @test_mm512_mask_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.512 + return _mm512_mask_alignr_epi8(__W, __U, __A, __B, 2); +} + +__m512i test_mm512_maskz_alignr_epi8(__mmask64 __U, __m512i __A,__m512i __B){ + // CHECK-LABEL: @test_mm512_maskz_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.512 + return _mm512_maskz_alignr_epi8(__U, __A, __B, 2); +} + + + diff --git a/clang/test/CodeGen/avx512f-builtins.c b/clang/test/CodeGen/avx512f-builtins.c index da3bf07668cc..1f048d36b158 100644 --- a/clang/test/CodeGen/avx512f-builtins.c +++ b/clang/test/CodeGen/avx512f-builtins.c @@ -180,6 +180,20 @@ __m512i test_mm512_alignr_epi32(__m512i a, __m512i b) return _mm512_alignr_epi32(a, b, 2); } +__m512i test_mm512_mask_alignr_epi32(__m512i w, __mmask16 u, __m512i a, __m512i b) +{ + // CHECK-LABEL: @test_mm512_mask_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.512 + return _mm512_mask_alignr_epi32(w, u, a, b, 2); +} + +__m512i test_mm512_maskz_alignr_epi32( __mmask16 u, __m512i a, __m512i b) +{ + // CHECK-LABEL: @test_mm512_maskz_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.512 + return _mm512_maskz_alignr_epi32(u, a, b, 2); +} + __m512i test_mm512_alignr_epi64(__m512i a, __m512i b) { // CHECK-LABEL: @test_mm512_alignr_epi64 @@ -187,6 +201,20 @@ __m512i test_mm512_alignr_epi64(__m512i a, __m512i b) return _mm512_alignr_epi64(a, b, 2); } +__m512i test_mm512_mask_alignr_epi64(__m512i w, __mmask8 u, __m512i a, __m512i b) +{ + // CHECK-LABEL: @test_mm512_mask_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.512 + return _mm512_mask_alignr_epi64(w, u, a, b, 2); +} + +__m512i test_mm512_maskz_alignr_epi64( __mmask8 u, __m512i a, __m512i b) +{ + // CHECK-LABEL: @test_mm512_maskz_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.512 + return _mm512_maskz_alignr_epi64(u, a, b, 2); +} + __m512d test_mm512_broadcastsd_pd(__m128d a) { // CHECK-LABEL: @test_mm512_broadcastsd_pd diff --git a/clang/test/CodeGen/avx512vl-builtins.c b/clang/test/CodeGen/avx512vl-builtins.c index 9ba949e72963..aea65bcb3ede 100644 --- a/clang/test/CodeGen/avx512vl-builtins.c +++ b/clang/test/CodeGen/avx512vl-builtins.c @@ -6461,3 +6461,75 @@ __m256i test_mm256_mask_permutexvar_epi32(__m256i __W, __mmask8 __M, __m256i __X // CHECK: @llvm.x86.avx512.mask.permvar.si.256 return _mm256_mask_permutexvar_epi32(__W, __M, __X, __Y); } + +__m128i test_mm_alignr_epi32(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.128 + return _mm_alignr_epi32(__A, __B, 1); +} + +__m128i test_mm_mask_alignr_epi32(__m128i __W, __mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.128 + return _mm_mask_alignr_epi32(__W, __U, __A, __B, 1); +} + +__m128i test_mm_maskz_alignr_epi32(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_maskz_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.128 + return _mm_maskz_alignr_epi32(__U, __A, __B, 1); +} + +__m256i test_mm256_alignr_epi32(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.256 + return _mm256_alignr_epi32(__A, __B, 1); +} + +__m256i test_mm256_mask_alignr_epi32(__m256i __W, __mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.256 + return _mm256_mask_alignr_epi32(__W, __U, __A, __B, 1); +} + +__m256i test_mm256_maskz_alignr_epi32(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_maskz_alignr_epi32 + // CHECK: @llvm.x86.avx512.mask.valign.d.256 + return _mm256_maskz_alignr_epi32(__U, __A, __B, 1); +} + +__m128i test_mm_alignr_epi64(__m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.128 + return _mm_alignr_epi64(__A, __B, 1); +} + +__m128i test_mm_mask_alignr_epi64(__m128i __W, __mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.128 + return _mm_mask_alignr_epi64(__W, __U, __A, __B, 1); +} + +__m128i test_mm_maskz_alignr_epi64(__mmask8 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_maskz_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.128 + return _mm_maskz_alignr_epi64(__U, __A, __B, 1); +} + +__m256i test_mm256_alignr_epi64(__m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.256 + return _mm256_alignr_epi64(__A, __B, 1); +} + +__m256i test_mm256_mask_alignr_epi64(__m256i __W, __mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.256 + return _mm256_mask_alignr_epi64(__W, __U, __A, __B, 1); +} + +__m256i test_mm256_maskz_alignr_epi64(__mmask8 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_maskz_alignr_epi64 + // CHECK: @llvm.x86.avx512.mask.valign.q.256 + return _mm256_maskz_alignr_epi64(__U, __A, __B, 1); +} diff --git a/clang/test/CodeGen/avx512vlbw-builtins.c b/clang/test/CodeGen/avx512vlbw-builtins.c index f05b32d2fe6f..f72363d8e9ca 100644 --- a/clang/test/CodeGen/avx512vlbw-builtins.c +++ b/clang/test/CodeGen/avx512vlbw-builtins.c @@ -2316,3 +2316,27 @@ __m256i test_mm256_mask_permutexvar_epi16(__m256i __W, __mmask16 __M, __m256i __ // CHECK: @llvm.x86.avx512.mask.permvar.hi.256 return _mm256_mask_permutexvar_epi16(__W, __M, __A, __B); } +__m128i test_mm_mask_alignr_epi8(__m128i __W, __mmask16 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_mask_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.128 + return _mm_mask_alignr_epi8(__W, __U, __A, __B, 2); +} + +__m128i test_mm_maskz_alignr_epi8(__mmask16 __U, __m128i __A, __m128i __B) { + // CHECK-LABEL: @test_mm_maskz_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.128 + return _mm_maskz_alignr_epi8(__U, __A, __B, 2); +} + +__m256i test_mm256_mask_alignr_epi8(__m256i __W, __mmask32 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_mask_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.256 + return _mm256_mask_alignr_epi8(__W, __U, __A, __B, 2); +} + +__m256i test_mm256_maskz_alignr_epi8(__mmask32 __U, __m256i __A, __m256i __B) { + // CHECK-LABEL: @test_mm256_maskz_alignr_epi8 + // CHECK: @llvm.x86.avx512.mask.palignr.256 + return _mm256_maskz_alignr_epi8(__U, __A, __B, 2); +} +