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  #ifndef CTRL
      13  #define CTRL 129
      14  #endif
      15  
      16  #ifndef SELECT4_DEFINED
      17  #define SELECT4_DEFINED
      18  static int
      19  select4 (int i, unsigned ctrl)
      20  {
      21    int res;
      22    switch (i % 4)
      23      {
      24      case 0:
      25        res = (CTRL & 0x03);
      26        break;
      27      case 1:
      28        res = ((CTRL & 0x0c) >> 2);
      29        break;
      30      case 2:
      31        res = ((CTRL & 0x30) >> 4);
      32        break;
      33      case 3:
      34        res = ((CTRL & 0xc0) >> 6);
      35        break;
      36      }
      37    return res;
      38  }
      39  #endif
      40  
      41  static void
      42  CALC (float *s, float *r)
      43  {
      44    int i;
      45  
      46    for (i = 0; i < SIZE; i++)
      47      {
      48        r[i] = s[(4 * (i / 4)) + select4 (i, CTRL)];
      49      }
      50  }
      51  
      52  void
      53  TEST (void)
      54  {
      55    UNION_TYPE (AVX512F_LEN,) s1, res1, res2, res3;
      56    MASK_TYPE mask = MASK_VALUE;
      57    float res_ref[SIZE];
      58    int i;
      59  
      60    for (i = 0; i < SIZE; i++)
      61      {
      62        s1.a[i] = i + 10.;
      63        res2.a[i] = DEFAULT_VALUE;
      64      }
      65  
      66    res1.x = INTRINSIC (_permute_ps) (s1.x, CTRL);
      67    res2.x = INTRINSIC (_mask_permute_ps) (res2.x, mask, s1.x, CTRL);
      68    res3.x = INTRINSIC (_maskz_permute_ps) (mask, s1.x, CTRL);
      69  
      70    CALC (s1.a, res_ref);
      71  
      72    if (UNION_CHECK (AVX512F_LEN,) (res1, res_ref))
      73      abort ();
      74  
      75    MASK_MERGE ()(res_ref, mask, SIZE);
      76    if (UNION_CHECK (AVX512F_LEN,) (res2, res_ref))
      77      abort ();
      78  
      79    MASK_ZERO ()(res_ref, mask, SIZE);
      80    if (UNION_CHECK (AVX512F_LEN,) (res3, res_ref))
      81      abort ();
      82  }