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