1/* { dg-do run } */
2/* { dg-require-effective-target avx512er } */
3/* { dg-options "-O2 -mavx512er" } */
4
5#include "avx512er-check.h"
6#include "avx512f-mask-type.h"
7#include "avx512f-helper.h"
8#include <math.h>
9
10void static
11compute_vrsqrt28ps (float *s, float *r)
12{
13  int i;
14  for (i = 0; i < 16; i++)
15    r[i] = 1.0 / sqrt (s[i]);
16}
17
18void static
19avx512er_test (void)
20{
21  union512 src, res1, res2, res3;
22  __mmask16 mask = MASK_VALUE;
23  float res_ref[16];
24  int i;
25
26  for (i = 0; i < 16; i++)
27    {
28      src.a[i] = 179.345 - 6.5645 * i;
29      res2.a[i] = DEFAULT_VALUE;
30    }
31
32  res1.x = _mm512_rsqrt28_ps (src.x);
33  res2.x = _mm512_mask_rsqrt28_ps (res2.x, mask, src.x);
34  res3.x = _mm512_maskz_rsqrt28_ps (mask, src.x);
35
36  compute_vrsqrt28ps (src.a, res_ref);
37
38  if (check_rough_union512 (res1, res_ref, 0.0001))
39    abort ();
40
41  MASK_MERGE ()(res_ref, mask, 16);
42  if (check_rough_union512 (res2, res_ref, 0.0001))
43    abort ();
44
45  MASK_ZERO ()(res_ref, mask, 16);
46  if (check_rough_union512 (res3, res_ref, 0.0001))
47    abort ();
48}
49