replica
/
hanchenye-llvm-project
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Clang][BuiltIn][AVX512] Adding intrinsics fot align{d|q} and palignr instruction set 

Differential Revision: http://reviews.llvm.org/D19588

llvm-svn: 267876
This commit is contained in:
Michael Zuckerman 2016-04-28 12:47:30 +00:00
parent 947dbe3aae
commit 533e065bdc
9 changed files with 323 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clang/include/clang/Basic/BuiltinsX86.def
View File

@ -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

23
clang/lib/Headers/avx512bwintrin.h
View File

@ -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

32
clang/lib/Headers/avx512fintrin.h
View File

@ -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__ ({ \

34
clang/lib/Headers/avx512vlbwintrin.h
View File

@ -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 */

84
clang/lib/Headers/avx512vlintrin.h
View File

@ -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

20
clang/test/CodeGen/avx512bw-builtins.c
View File

@ -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);
}

28
clang/test/CodeGen/avx512f-builtins.c
View File

@ -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

72
clang/test/CodeGen/avx512vl-builtins.c
View File

@ -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);
}

24
clang/test/CodeGen/avx512vlbw-builtins.c
View File

@ -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);
}