gcc/testsuite/gcc.target/i386/avx2-vpblendw-2.c

   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 }