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
8 #define SIZE (128 / 32)
9 #include "avx512f-mask-type.h"
10
11 static void
compute_sqrtss(float * s1,float * s2,float * r)12 compute_sqrtss (float *s1, float *s2, float *r)
13 {
14 r[0] = sqrt(s2[0]);
15 int i;
16 for (i = 1; i < SIZE; i++)
17 {
18 r[i] = s1[i];
19 }
20 }
21
22 static void
avx512f_test(void)23 avx512f_test (void)
24 {
25 union128 res1, res2, res3;
26 union128 s1, s2;
27 float res_ref[SIZE];
28 MASK_TYPE mask = MASK_VALUE;
29 int i;
30
31 for (i = 0; i < SIZE; i++)
32 {
33 s1.a[i] = 11.5 * (i + 1);
34 s2.a[i] = 10.5 * (i + 1);
35 res_ref[i] = 9.5 * (i + 1);
36 res1.a[i] = DEFAULT_VALUE;
37 res2.a[i] = DEFAULT_VALUE;
38 res3.a[i] = DEFAULT_VALUE;
39 }
40
41 res1.x = _mm_sqrt_round_ss (s1.x, s2.x,
42 _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC);
43 res2.x = _mm_mask_sqrt_round_ss (s1.x, mask, s1.x, s2.x,
44 _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC);
45 res3.x = _mm_maskz_sqrt_round_ss (mask, s1.x, s2.x,
46 _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC);
47
48 compute_sqrtss (s1.a, s2.a, res_ref);
49
50 if (check_union128 (res1, res_ref))
51 abort ();
52
53 MASK_MERGE () (res_ref, mask, 1);
54
55 if (check_union128 (res2, res_ref))
56 abort ();
57
58 MASK_ZERO () (res_ref, mask, 1);
59
60 if (check_union128 (res3, res_ref))
61 abort ();
62 }
63
64