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