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