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  #include <math.h>
      10  #define SIZE (AVX512F_LEN / 32)
      11  #include "avx512f-mask-type.h"
      12  
      13  static void
      14  CALC (float *s, float *r)
      15  {
      16    int i;
      17  
      18    for (i = 0; i < SIZE; i++)
      19      {
      20        r[i] = 1.0 / sqrt(s[i]);
      21      }
      22  }
      23  
      24  void
      25  TEST (void)
      26  {
      27    UNION_TYPE (AVX512F_LEN,) s, res1, res2, res3;
      28    MASK_TYPE mask = MASK_VALUE;
      29    float res_ref[SIZE];
      30    int i;
      31  
      32    for (i = 0; i < SIZE; i++)
      33      {
      34        s.a[i] = 123.456 * (i + 2000);
      35        res2.a[i] = DEFAULT_VALUE;
      36      }
      37  
      38    res1.x = INTRINSIC (_rsqrt14_ps) (s.x);
      39    res2.x = INTRINSIC (_mask_rsqrt14_ps) (res2.x, mask, s.x);
      40    res3.x = INTRINSIC (_maskz_rsqrt14_ps) (mask, s.x);
      41  
      42    CALC (s.a, res_ref);
      43  
      44    if (UNION_ROUGH_CHECK (AVX512F_LEN,) (res1, res_ref, 0.0001))
      45      abort ();
      46  
      47    MASK_MERGE () (res_ref, mask, SIZE);
      48    if (UNION_ROUGH_CHECK (AVX512F_LEN,) (res2, res_ref, 0.0001))
      49      abort ();
      50  
      51    MASK_ZERO () (res_ref, mask, SIZE);
      52    if (UNION_ROUGH_CHECK (AVX512F_LEN,) (res3, res_ref, 0.0001))
      53      abort ();
      54  }