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 *s2, MASK_TYPE mask)
      13  {
      14    int i;
      15    for (i = 0; i < SIZE; i++)
      16      {
      17        r[i] = (mask & (1 << i)) ? s2[i] : s1[i];
      18      }
      19  }
      20  
      21  void
      22  TEST (void)
      23  {
      24    int i, sign;
      25    UNION_TYPE (AVX512F_LEN, i_w) res1, src1, src2;
      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] = 15 + 3467 * i * sign;
      33        src2.a[i] = -2217 * i * sign;
      34        sign = sign * -1;
      35      }
      36  
      37    res1.x = INTRINSIC (_mask_blend_epi16) (mask, src1.x, src2.x);
      38  
      39    CALC (res_ref, src1.a, src2.a, mask);
      40  
      41    if (UNION_CHECK (AVX512F_LEN, i_w) (res1, res_ref))
      42      abort ();
      43  }