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
11void
12CALC (short *r, short *s1, short count)
13{
14  int i;
15  for (i = 0; i < SIZE; i++)
16    {
17      r[i] = count < 16 ? (s1[i] << count) : 0;
18    }
19}
20
21void
22TEST (void)
23{
24  int i, sign;
25  UNION_TYPE (AVX512F_LEN, i_w) res1, res2, res3, src1;
26  MASK_TYPE mask = MASK_VALUE;
27  short res_ref[SIZE];
28
29  sign = -1;
30  for (i = 0; i < SIZE; i++)
31    {
32      src1.a[i] = 2 + sign * 7 * i % 291;
33      sign = sign * -1;
34    }
35
36  for (i = 0; i < SIZE; i++)
37    res2.a[i] = DEFAULT_VALUE;
38
39  res1.x = INTRINSIC (_slli_epi16) (src1.x, 5);
40  res2.x = INTRINSIC (_mask_slli_epi16) (res2.x, mask, src1.x, 5);
41  res3.x = INTRINSIC (_maskz_slli_epi16) (mask, src1.x, 5);
42
43  CALC (res_ref, src1.a, 5);
44
45  if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
46    abort ();
47
48  MASK_MERGE (i_w) (res_ref, mask, SIZE);
49  if (UNION_CHECK (AVX512F_LEN, i_w) (res2, res_ref))
50    abort ();
51
52  MASK_ZERO (i_w) (res_ref, mask, SIZE);
53  if (UNION_CHECK (AVX512F_LEN, i_w) (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 (_slli_epi16) (src1.x, 17);
61  res2.x = INTRINSIC (_mask_slli_epi16) (res2.x, mask, src1.x, 17);
62  res3.x = INTRINSIC (_maskz_slli_epi16) (mask, src1.x, 17);
63
64  CALC (res_ref, src1.a, 17);
65
66  if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
67    abort ();
68
69  MASK_MERGE (i_w) (res_ref, mask, SIZE);
70  if (UNION_CHECK (AVX512F_LEN, i_w) (res2, res_ref))
71    abort ();
72
73  MASK_ZERO (i_w) (res_ref, mask, SIZE);
74  if (UNION_CHECK (AVX512F_LEN, i_w) (res3, res_ref))
75    abort ();
76}
77