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 (unsigned short *r, unsigned short *s1, unsigned 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;
      25    UNION_TYPE (AVX512F_LEN, i_w) res1, res2, res3, src1;
      26    MASK_TYPE mask = MASK_VALUE;
      27    unsigned short res_ref[SIZE];
      28  
      29    for (i = 0; i < SIZE; i++)
      30      {
      31        src1.a[i] = 2 + 7 * i % 291;
      32      }
      33  
      34    for (i = 0; i < SIZE; i++)
      35      res2.a[i] = DEFAULT_VALUE;
      36  
      37    res1.x = INTRINSIC (_srli_epi16) (src1.x, 5);
      38    res2.x = INTRINSIC (_mask_srli_epi16) (res2.x, mask, src1.x, 5);
      39    res3.x = INTRINSIC (_maskz_srli_epi16) (mask, src1.x, 5);
      40  
      41    CALC (res_ref, src1.a, 5);
      42  
      43    if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
      44      abort ();
      45  
      46    MASK_MERGE (i_w) (res_ref, mask, SIZE);
      47    if (UNION_CHECK (AVX512F_LEN, i_w) (res2, res_ref))
      48      abort ();
      49  
      50    MASK_ZERO (i_w) (res_ref, mask, SIZE);
      51    if (UNION_CHECK (AVX512F_LEN, i_w) (res3, res_ref))
      52      abort ();
      53  
      54    for (i = 0; i < SIZE; i++)
      55      res2.a[i] = DEFAULT_VALUE;
      56  
      57    res1.x = INTRINSIC (_srli_epi16) (src1.x, 17);
      58    res2.x = INTRINSIC (_mask_srli_epi16) (res2.x, mask, src1.x, 17);
      59    res3.x = INTRINSIC (_maskz_srli_epi16) (mask, src1.x, 17);
      60  
      61    CALC (res_ref, src1.a, 17);
      62  
      63    if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
      64      abort ();
      65  
      66    MASK_MERGE (i_w) (res_ref, mask, SIZE);
      67    if (UNION_CHECK (AVX512F_LEN, i_w) (res2, res_ref))
      68      abort ();
      69  
      70    MASK_ZERO (i_w) (res_ref, mask, SIZE);
      71    if (UNION_CHECK (AVX512F_LEN, i_w) (res3, res_ref))
      72      abort ();
      73  }