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  void
      12  CALC (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  
      21  void
      22  TEST (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  }