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,unsigned int * s2)13 CALC (unsigned int *r, unsigned int *s1, unsigned int *s2)
14 {
15 int i;
16 for (i = 0; i < SIZE; i++)
17 {
18 r[i] = s2[i] < 32 ? (s1[i] >> s2[i]) : 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, src2;
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 src2.a[i] = 1 + 17 * i % 71;
34 }
35
36 for (i = 0; i < SIZE; i++)
37 res2.a[i] = DEFAULT_VALUE;
38
39 res1.x = INTRINSIC (_srlv_epi32) (src1.x, src2.x);
40 res2.x = INTRINSIC (_mask_srlv_epi32) (res2.x, mask, src1.x, src2.x);
41 res3.x = INTRINSIC (_maskz_srlv_epi32) (mask, src1.x, src2.x);
42
43 CALC (res_ref, src1.a, src2.a);
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