gcc/testsuite/gcc.target/i386/avx512f-vshufps-2.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512f" } */
   3 /* { dg-require-effective-target avx512f } */
   4
   5 #define AVX512F
   6
   7 #include "avx512f-helper.h"
   8
   9 #define SIZE (AVX512F_LEN / 32)
  10 #include "avx512f-mask-type.h"
  11
  12 void
  13 CALC (float *e, UNION_TYPE (AVX512F_LEN,) s1, UNION_TYPE (AVX512F_LEN,) s2,
  14       int imm)
  15 {
  16   e[0] = s1.a[(imm >> 0) & 0x3];
  17   e[1] = s1.a[(imm >> 2) & 0x3];
  18   e[2] = s2.a[(imm >> 4) & 0x3];
  19   e[3] = s2.a[(imm >> 6) & 0x3];
  20 #if AVX512F_LEN > 128
  21   e[4] = s1.a[4 + ((imm >> 0) & 0x3)];
  22   e[5] = s1.a[4 + ((imm >> 2) & 0x3)];
  23   e[6] = s2.a[4 + ((imm >> 4) & 0x3)];
  24   e[7] = s2.a[4 + ((imm >> 6) & 0x3)];
  25 #if AVX512F_LEN > 256
  26   e[8] = s1.a[8 + ((imm >> 0) & 0x3)];
  27   e[9] = s1.a[8 + ((imm >> 2) & 0x3)];
  28   e[10] = s2.a[8 + ((imm >> 4) & 0x3)];
  29   e[11] = s2.a[8 + ((imm >> 6) & 0x3)];
  30   e[12] = s1.a[12 + ((imm >> 0) & 0x3)];
  31   e[13] = s1.a[12 + ((imm >> 2) & 0x3)];
  32   e[14] = s2.a[12 + ((imm >> 4) & 0x3)];
  33   e[15] = s2.a[12 + ((imm >> 6) & 0x3)];
  34 #endif
  35 #endif
  36 }
  37
  38 void
  39 TEST (void)
  40 {
  41   UNION_TYPE (AVX512F_LEN,) u1, u2, u3, s1, s2;
  42   float e[SIZE];
  43   int i, sign;
  44   MASK_TYPE mask = MASK_VALUE;
  45
  46   sign = -1;
  47   for (i = 0; i < SIZE; i++)
  48     {
  49       s1.a[i] = 1.5 + 34.67 * i * sign;
  50       s2.a[i] = -22.17 * i * sign;
  51       u1.a[i] = DEFAULT_VALUE;
  52       u2.a[i] = DEFAULT_VALUE;
  53       u3.a[i] = DEFAULT_VALUE;
  54       sign = sign * -1;
  55     }
  56
  57
  58   u1.x = INTRINSIC (_shuffle_ps) (s1.x, s2.x, 203);
  59   u2.x = INTRINSIC (_mask_shuffle_ps) (u2.x, mask, s1.x, s2.x, 203);
  60   u3.x = INTRINSIC (_maskz_shuffle_ps) (mask, s1.x, s2.x, 203);
  61
  62   CALC (e, s1, s2, 203);
  63
  64   if (UNION_CHECK (AVX512F_LEN,) (u1, e))
  65     abort ();
  66
  67   MASK_MERGE ()(e, mask, SIZE);
  68   if (UNION_CHECK (AVX512F_LEN,) (u2, e))
  69     abort ();
  70
  71   MASK_ZERO ()(e, mask, SIZE);
  72   if (UNION_CHECK (AVX512F_LEN,) (u3, e))
  73     abort ();
  74 }