1 /* { dg-do run } */
2 /* { dg-options "-O2 -mavx512f -mavx512bw -mavx512vbmi2" } */
3 /* { dg-require-effective-target avx512f } */
4 /* { dg-require-effective-target avx512vbmi2 } */
5
6 #define AVX512F
7
8 #define AVX512VBMI2
9 #include "avx512f-helper.h"
10
11 #define SIZE (AVX512F_LEN / 32)
12
13 #include "avx512f-mask-type.h"
14
15 static void
CALC(int * r,int * dst,int * s1,int * s2)16 CALC (int *r, int *dst, int *s1, int *s2)
17 {
18 int i;
19 for (i = 0; i < SIZE; i++)
20 {
21 r[i] = (dst[i] >> (s2[i] & 31)) | (s1[i] << (32 - (s2[i] & 31)));
22 }
23 }
24
25 void
TEST(void)26 TEST (void)
27 {
28 int i;
29 UNION_TYPE (AVX512F_LEN, i_d) res1, res2, res3, src1, src2;
30 MASK_TYPE mask = MASK_VALUE;
31 int res_ref[SIZE];
32
33 for (i = 0; i < SIZE; i++)
34 {
35 src1.a[i] = 1 + i;
36 src2.a[i] = 2 + 2*i;
37 }
38
39 for (i = 0; i < SIZE; i++)
40 {
41 res1.a[i] = DEFAULT_VALUE;
42 res2.a[i] = DEFAULT_VALUE;
43 res3.a[i] = DEFAULT_VALUE;
44 }
45
46 CALC (res_ref, res1.a, src1.a, src2.a);
47
48 res1.x = INTRINSIC (_shrdv_epi32) (res1.x, src1.x, src2.x);
49 res2.x = INTRINSIC (_mask_shrdv_epi32) (res2.x, mask, src1.x, src2.x);
50 res3.x = INTRINSIC (_maskz_shrdv_epi32) (mask, res3.x, src1.x, src2.x);
51
52 if (UNION_CHECK (AVX512F_LEN, i_d) (res1, res_ref))
53 abort ();
54
55 MASK_MERGE (i_d) (res_ref, mask, SIZE);
56 if (UNION_CHECK (AVX512F_LEN, i_d) (res2, res_ref))
57 abort ();
58
59 MASK_ZERO (i_d) (res_ref, mask, SIZE);
60 if (UNION_CHECK (AVX512F_LEN, i_d) (res3, res_ref))
61 abort ();
62 }
63