gcc/testsuite/gcc.target/i386/avx2-vpermd-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 10
   9
  10 static void
  11 init_permd (int *src1, int *src2, int seed)
  12 {
  13   int i, sign = 1;
  14
  15   for (i = 0; i < 8; i++)
  16     {
  17       src1[i] = (i + seed) * (i + seed) * sign;
  18       src2[i] = (i + seed + 20) * sign;
  19       sign = -sign;
  20     }
  21 }
  22
  23 static void
  24 calc_permd (int *src1, int *src2, int *dst)
  25 {
  26   int i;
  27   unsigned temp;
  28
  29   memcpy (dst, src1, 32);
  30   for (i = 0; i < 8; i++)
  31     {
  32       temp = src2[i];
  33       dst[i] = src1[temp & 7];
  34     }
  35 }
  36
  37 static void
  38 avx2_test (void)
  39 {
  40   union256i_d src1, src2, dst;
  41   int dst_ref[8];
  42   int i;
  43
  44   for (i = 0; i < NUM; i++)
  45     {
  46       init_permd (src1.a, src2.a, i);
  47
  48       dst.x = _mm256_permutevar8x32_epi32 (src1.x, src2.x);
  49       calc_permd (src1.a, src2.a, dst_ref);
  50
  51       if (check_union256i_d (dst, dst_ref))
  52         abort ();
  53     }
  54 }