Index: include/clang/Basic/BuiltinsX86.def =================================================================== --- include/clang/Basic/BuiltinsX86.def +++ include/clang/Basic/BuiltinsX86.def @@ -2185,6 +2185,9 @@ TARGET_BUILTIN(__builtin_ia32_kunpckhi, "UsUsUs","","avx512f") TARGET_BUILTIN(__builtin_ia32_kxnorhi, "UsUsUs","","avx512f") TARGET_BUILTIN(__builtin_ia32_kxorhi, "UsUsUs","","avx512f") +TARGET_BUILTIN(__builtin_ia32_pshufd512_mask, "V16iV16iCsV16iUc","","avx512f") +TARGET_BUILTIN(__builtin_ia32_pshufd256_mask, "V8iV8iCsV8iUc","","avx512vl") +TARGET_BUILTIN(__builtin_ia32_pshufd128_mask, "V4iV4iCsV4iUc","","avx512vl") #undef BUILTIN #undef TARGET_BUILTIN Index: lib/Headers/avx512fintrin.h =================================================================== --- lib/Headers/avx512fintrin.h +++ lib/Headers/avx512fintrin.h @@ -7324,6 +7324,27 @@ return (__mmask16) __builtin_ia32_kxorhi ((__mmask16) __A, (__mmask16) __B); } +#define _mm512_shuffle_epi32( __A, __I) __extension__ ({ \ +__builtin_ia32_pshufd512_mask ((__v16si)( __A),\ + ( __I),\ + (__v16si) _mm512_undefined_epi32 (),\ + (__mmask16) -1);\ +}) + +#define _mm512_mask_shuffle_epi32( __W, __U, __A, __I) __extension__ ({ \ +__builtin_ia32_pshufd512_mask ((__v16si)( __A),\ + ( __I),\ + (__v16si)( __W),\ + (__mmask16)( __U));\ +}) + +#define _mm512_maskz_shuffle_epi32( __U, __A, __I) __extension__ ({ \ +__builtin_ia32_pshufd512_mask ((__v16si)( __A),\ + ( __I),\ + (__v16si) _mm512_setzero_si512 (),\ + (__mmask16)( __U));\ +}) + #undef __DEFAULT_FN_ATTRS #endif // __AVX512FINTRIN_H Index: lib/Headers/avx512vlintrin.h =================================================================== --- lib/Headers/avx512vlintrin.h +++ lib/Headers/avx512vlintrin.h @@ -9209,6 +9209,29 @@ (__mmask8) -1); } +#define _mm256_mask_shuffle_epi32( __W, __U, __A, __I) __extension__({\ +__builtin_ia32_pshufd256_mask((__v8si) (__A), (__I),\ + (__v8si) (__W), (__mmask8) __U);\ +}) + +#define _mm256_maskz_shuffle_epi32( __U, __A, __I) __extension__({\ +__builtin_ia32_pshufd256_mask((__v8si) (__A), (__I),\ + (__v8si) _mm256_setzero_si256 (),\ + (__mmask8) (__U));\ +}) + +#define _mm_mask_shuffle_epi32( __W, __U, __A, __I) __extension__({\ + __builtin_ia32_pshufd128_mask ((__v4si) (__A), (__I),\ + (__v4si) (__W), (__mmask8) __U);\ +}) + +#define _mm_maskz_shuffle_epi32( __U, __A, __I) __extension__({\ + __builtin_ia32_pshufd128_mask ((__v4si) (__A), (__I),\ + (__v4si)\ + _mm_setzero_si128 (),\ + (__mmask8) (__U));\ +}) + #undef __DEFAULT_FN_ATTRS #undef __DEFAULT_FN_ATTRS_BOTH Index: test/CodeGen/avx512f-builtins.c =================================================================== --- test/CodeGen/avx512f-builtins.c +++ test/CodeGen/avx512f-builtins.c @@ -5053,3 +5053,22 @@ // CHECK: @llvm.x86.avx512.kxor.w return _mm512_kxor(__A, __B); } + +__m512i test_mm512_shuffle_epi32(__m512i __A) { + // CHECK-LABEL: @test_mm512_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.512 + return _mm512_shuffle_epi32(__A, 1); +} + +__m512i test_mm512_mask_shuffle_epi32(__m512i __W, __mmask16 __U, __m512i __A) { + // CHECK-LABEL: @test_mm512_mask_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.512 + return _mm512_mask_shuffle_epi32(__W, __U, __A, 1); +} + +__m512i test_mm512_maskz_shuffle_epi32(__mmask16 __U, __m512i __A) { + // CHECK-LABEL: @test_mm512_maskz_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.512 + return _mm512_maskz_shuffle_epi32(__U, __A, 1); +} + Index: test/CodeGen/avx512vl-builtins.c =================================================================== --- test/CodeGen/avx512vl-builtins.c +++ test/CodeGen/avx512vl-builtins.c @@ -6461,3 +6461,27 @@ // CHECK: @llvm.x86.avx512.mask.permvar.si.256 return _mm256_mask_permutexvar_epi32(__W, __M, __X, __Y); } + +__m128i test_mm_mask_shuffle_epi32(__m128i __W, __mmask8 __U, __m128i __A) { + // CHECK-LABEL: @test_mm_mask_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.128 + return _mm_mask_shuffle_epi32(__W, __U, __A, 1); +} + +__m128i test_mm_maskz_shuffle_epi32(__mmask8 __U, __m128i __A) { + // CHECK-LABEL: @test_mm_maskz_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.128 + return _mm_maskz_shuffle_epi32(__U, __A, 2); +} + +__m256i test_mm256_mask_shuffle_epi32(__m256i __W, __mmask8 __U, __m256i __A) { + // CHECK-LABEL: @test_mm256_mask_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.256 + return _mm256_mask_shuffle_epi32(__W, __U, __A, 2); +} + +__m256i test_mm256_maskz_shuffle_epi32(__mmask8 __U, __m256i __A) { + // CHECK-LABEL: @test_mm256_maskz_shuffle_epi32 + // CHECK: @llvm.x86.avx512.mask.pshuf.d.256 + return _mm256_maskz_shuffle_epi32(__U, __A, 2); +} \ No newline at end of file