1 /* { dg-do run } */
2 /* { dg-options "-O2 -mavx512f" } */
3 /* { dg-require-effective-target avx512f } */
4
5 #define AVX512F
6
7 #include "avx512f-helper.h"
8
9 #define SIZE (AVX512F_LEN / 32)
10 #include "avx512f-mask-type.h"
11
12 static void
CALC(unsigned int * r,unsigned int * s1,int count)13 CALC (unsigned int *r, unsigned int *s1, int count)
14 {
15 int i;
16 for (i = 0; i < SIZE; i++)
17 {
18 r[i] = count < 32 ? (s1[i] >> count) : 0;
19 }
20 }
21
22 void
TEST(void)23 TEST (void)
24 {
25 int i;
26 UNION_TYPE (AVX512F_LEN, i_ud) res1, res2, res3, src1;
27 MASK_TYPE mask = MASK_VALUE;
28 unsigned int res_ref[SIZE];
29
30 for (i = 0; i < SIZE; i++)
31 {
32 src1.a[i] = 2 + 7 * i % 291;
33 }
34
35 for (i = 0; i < SIZE; i++)
36 res2.a[i] = DEFAULT_VALUE;
37
38 res1.x = INTRINSIC (_srli_epi32) (src1.x, 3);
39 res2.x =
40 INTRINSIC (_mask_srli_epi32) (res2.x, mask, src1.x, 3);
41 res3.x = INTRINSIC (_maskz_srli_epi32) (mask, src1.x, 3);
42
43 CALC (res_ref, src1.a, 3);
44
45 if (UNION_CHECK (AVX512F_LEN, i_ud) (res1, res_ref))
46 abort ();
47
48 MASK_MERGE (i_ud) (res_ref, mask, SIZE);
49 if (UNION_CHECK (AVX512F_LEN, i_ud) (res2, res_ref))
50 abort ();
51
52 MASK_ZERO (i_ud) (res_ref, mask, SIZE);
53 if (UNION_CHECK (AVX512F_LEN, i_ud) (res3, res_ref))
54 abort ();
55
56
57 for (i = 0; i < SIZE; i++)
58 res2.a[i] = DEFAULT_VALUE;
59
60 res1.x = INTRINSIC (_srli_epi32) (src1.x, 33);
61 res2.x =
62 INTRINSIC (_mask_srli_epi32) (res2.x, mask, src1.x, 33);
63 res3.x = INTRINSIC (_maskz_srli_epi32) (mask, src1.x, 33);
64
65 CALC (res_ref, src1.a, 33);
66
67 if (UNION_CHECK (AVX512F_LEN, i_ud) (res1, res_ref))
68 abort ();
69
70 MASK_MERGE (i_ud) (res_ref, mask, SIZE);
71 if (UNION_CHECK (AVX512F_LEN, i_ud) (res2, res_ref))
72 abort ();
73
74 MASK_ZERO (i_ud) (res_ref, mask, SIZE);
75 if (UNION_CHECK (AVX512F_LEN, i_ud) (res3, res_ref))
76 abort ();
77 }
78