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

/* { dg-do run } */
/* { dg-options "-mavx512f -O2" } */
/* { dg-require-effective-target avx512f } */

#include <math.h>
#include "avx512f-check.h"
#include "avx512f-helper.h"

static void
compute_vrsqrt14sd (double *s1, double *s2, double *r)
{
  r[0] = 1.0 / sqrt (s2[0]);
  r[1] = s1[1];
}

static void
avx512f_test (void)
{
  union128d s1, s2, res1, res2, res3;
  __mmask8 m = 0;
  double res_ref[2];

  s1.x = _mm_set_pd (-3.0, 111.111);
  s2.x = _mm_set_pd (222.222, 4.0);

  res1.x = _mm_rsqrt14_sd (s1.x, s2.x);

  compute_vrsqrt14sd (s1.a, s2.a, res_ref);

  if (check_fp_union128d (res1, res_ref))
    abort ();

  res2.x = _mm_set_pd (-4.0, DEFAULT_VALUE);
  res2.x = _mm_mask_rsqrt14_sd(res2.x, m, s1.x, s2.x);
 
  MASK_MERGE (d) (res_ref, m, 1);
  if (checkVd (res2.a, res_ref, 2))
    abort();

  res3.x = _mm_maskz_rsqrt14_sd(m, s1.x, s2.x);
  
  MASK_ZERO (d) (res_ref, m, 1);
  if (checkVd (res3.a, res_ref, 2))
    abort();
}