1  /* { dg-do run } */
       2  /* { dg-options "-mavx2 -O2" } */
       3  /* { dg-require-effective-target avx2 } */
       4  
       5  #include "avx2-check.h"
       6  #include <string.h>
       7  
       8  #define NUM 20
       9  
      10  #undef MASK
      11  #define MASK 0xfe
      12  
      13  static void
      14  init_pblendw (short *src1, short *src2, int seed)
      15  {
      16    int i, sign = 1;
      17  
      18    for (i = 0; i < 16; i++)
      19      {
      20        src1[i] = (i + seed) * (i + seed) * sign;
      21        src2[i] = (i + seed + 20) * sign;
      22        sign = -sign;
      23      }
      24  }
      25  
      26  static void
      27  calc_pblendw (short *src1, short *src2, unsigned int mask, short *dst)
      28  {
      29    int i;
      30  
      31    memcpy (dst, src1, 32);
      32    for (i = 0; i < 16; i++)
      33      if (mask & (1 << (i % 8)))
      34        dst[i] = src2[i];
      35  }
      36  
      37  static void
      38  avx2_test (void)
      39  {
      40    union256i_w src1, src2, dst;
      41    short dst_ref[16];
      42    int i;
      43  
      44    for (i = 0; i < NUM; i++)
      45      {
      46        init_pblendw (src1.a, src2.a, i);
      47  
      48        dst.x = _mm256_blend_epi16 (src1.x, src2.x, MASK);
      49        calc_pblendw (src1.a, src2.a, MASK, dst_ref);
      50  
      51        if (check_union256i_w (dst, dst_ref))
      52  	abort ();
      53      }
      54  }