gcc/testsuite/gcc.target/i386/avx512f-vshufpd-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 / 64)
  10 #include "avx512f-mask-type.h"
  11
  12 #if AVX512F_LEN > 256
  13   #define IMM_VAR 120
  14 #elif AVX512F_LEN > 128
  15   #define IMM_VAR 5
  16 #else
  17   #define IMM_VAR 1
  18 #endif
  19
  20 void static
  21 CALC (double *e, UNION_TYPE (AVX512F_LEN, d) s1,
  22       UNION_TYPE (AVX512F_LEN, d) s2, int imm)
  23 {
  24   e[0] = (imm & (1 << 0)) ? s1.a[1] : s1.a[0];
  25   e[1] = (imm & (1 << 1)) ? s2.a[1] : s2.a[0];
  26 #if AVX512F_LEN > 128
  27   e[2] = (imm & (1 << 2)) ? s1.a[3] : s1.a[2];
  28   e[3] = (imm & (1 << 3)) ? s2.a[3] : s2.a[2];
  29 #if AVX512F_LEN > 256
  30   e[4] = (imm & (1 << 4)) ? s1.a[5] : s1.a[4];
  31   e[5] = (imm & (1 << 5)) ? s2.a[5] : s2.a[4];
  32   e[6] = (imm & (1 << 6)) ? s1.a[7] : s1.a[6];
  33   e[7] = (imm & (1 << 7)) ? s2.a[7] : s2.a[6];
  34 #endif
  35 #endif
  36 }
  37
  38 void
  39 TEST (void)
  40 {
  41   UNION_TYPE (AVX512F_LEN, d) u1, u2, u3, s1, s2;
  42   double e[SIZE];
  43   MASK_TYPE mask = MASK_VALUE;
  44   int i;
  45
  46   for (i = 0; i < SIZE; i++)
  47     {
  48       s1.a[i] = 2134.3343 * i + 54846.4641;
  49       s2.a[i] = 856.43576 * i + 1124.209;
  50       u1.a[i] = DEFAULT_VALUE;
  51       u2.a[i] = DEFAULT_VALUE;
  52       u3.a[i] = DEFAULT_VALUE;
  53     }
  54
  55   u1.x = INTRINSIC (_shuffle_pd) (s1.x, s2.x, IMM_VAR);
  56   u2.x = INTRINSIC (_mask_shuffle_pd) (u2.x, mask, s1.x, s2.x, IMM_VAR);
  57   u3.x = INTRINSIC (_maskz_shuffle_pd) (mask, s1.x, s2.x, IMM_VAR);
  58   CALC (e, s1, s2, IMM_VAR);
  59
  60   if (UNION_CHECK (AVX512F_LEN, d) (u1, e))
  61     abort ();
  62
  63   MASK_MERGE (d) (e, mask, SIZE);
  64   if (UNION_CHECK (AVX512F_LEN, d) (u2, e))
  65     abort ();
  66
  67   MASK_ZERO (d) (e, mask, SIZE);
  68   if (UNION_CHECK (AVX512F_LEN, d) (u3, e))
  69     abort ();
  70 }
  71
  72 #undef IMM_VAR