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

/* { dg-do run } */
/* { dg-options "-mavx2 -O2" } */
/* { dg-require-effective-target avx2 } */

#include "avx2-check.h"
#include <string.h>

#define NUM 10

static void
init_permd (int *src1, int *src2, int seed)
{
  int i, sign = 1;

  for (i = 0; i < 8; i++)
    {
      src1[i] = (i + seed) * (i + seed) * sign;
      src2[i] = (i + seed + 20) * sign;
      sign = -sign;
    }
}

static void
calc_permd (int *src1, int *src2, int *dst)
{
  int i;
  unsigned temp;

  memcpy (dst, src1, 32);
  for (i = 0; i < 8; i++)
    {
      temp = src2[i];
      dst[i] = src1[temp & 7];
    }
}

static void
avx2_test (void)
{
  union256i_d src1, src2, dst;
  int dst_ref[8];
  int i;

  for (i = 0; i < NUM; i++)
    {
      init_permd (src1.a, src2.a, i);

      dst.x = _mm256_permutevar8x32_epi32 (src1.x, src2.x);
      calc_permd (src1.a, src2.a, dst_ref);

      if (check_union256i_d (dst, dst_ref))
	abort ();
    }
}