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

   1 /* { dg-do run } */
   2 /* { dg-options "-mavx512f -O2" } */
   3 /* { dg-require-effective-target avx512f } */
   4
   5 #include <math.h>
   6 #include "avx512f-check.h"
   7 #include "avx512f-helper.h"
   8
   9 static void
  10 compute_vrsqrt14sd (double *s1, double *s2, double *r)
  11 {
  12   r[0] = 1.0 / sqrt (s2[0]);
  13   r[1] = s1[1];
  14 }
  15
  16 static void
  17 avx512f_test (void)
  18 {
  19   union128d s1, s2, res1, res2, res3;
  20   __mmask8 m = 0;
  21   double res_ref[2];
  22
  23   s1.x = _mm_set_pd (-3.0, 111.111);
  24   s2.x = _mm_set_pd (222.222, 4.0);
  25
  26   res1.x = _mm_rsqrt14_sd (s1.x, s2.x);
  27
  28   compute_vrsqrt14sd (s1.a, s2.a, res_ref);
  29
  30   if (check_fp_union128d (res1, res_ref))
  31     abort ();
  32
  33   res2.x = _mm_set_pd (-4.0, DEFAULT_VALUE);
  34   res2.x = _mm_mask_rsqrt14_sd(res2.x, m, s1.x, s2.x);
  35
  36   MASK_MERGE (d) (res_ref, m, 1);
  37   if (checkVd (res2.a, res_ref, 2))
  38     abort();
  39
  40   res3.x = _mm_maskz_rsqrt14_sd(m, s1.x, s2.x);
  41
  42   MASK_ZERO (d) (res_ref, m, 1);
  43   if (checkVd (res3.a, res_ref, 2))
  44     abort();
  45 }