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#define SIZE (AVX512F_LEN / 64)
10#include "avx512f-mask-type.h"
11
12static void
13CALC (double *s1, double *s2, double *s3, double *r)
14{
15  int i;
16
17  for (i = 0; i < SIZE; i++)
18    {
19      r[i] = s1[i] * s2[i] + s3[i];
20    }
21}
22
23void
24TEST (void)
25{
26  UNION_TYPE (AVX512F_LEN, d) s1, s2, s3, res1, res2, res3, res4;
27  MASK_TYPE mask = MASK_VALUE;
28  double res_ref1[SIZE];
29  double res_ref2[SIZE];
30  int i, sign = 1;
31
32  for (i = 0; i < SIZE; i++)
33    {
34      s1.a[i] = DEFAULT_VALUE;
35      s2.a[i] = 56.78 * (i + 1) * sign;
36      s3.a[i] = 90.12 * (i + 2) * sign;
37      sign = -sign;
38    }
39
40#if AVX512F_LEN == 512
41  res1.x = INTRINSIC (_fmadd_pd) (s1.x, s2.x, s3.x);
42#endif
43  res2.x = INTRINSIC (_mask_fmadd_pd) (s1.x, mask, s2.x, s3.x);
44  res3.x = INTRINSIC (_mask3_fmadd_pd) (s2.x, s3.x, s1.x, mask);
45  res4.x = INTRINSIC (_maskz_fmadd_pd) (mask, s1.x, s2.x, s3.x);
46
47  CALC (s1.a, s2.a, s3.a, res_ref1);
48  CALC (s2.a, s3.a, s1.a, res_ref2);
49
50#if AVX512F_LEN == 512
51  if (UNION_ROUGH_CHECK (AVX512F_LEN, d) (res1, res_ref1, 0.0001))
52    abort ();
53#endif
54
55  MASK_MERGE (d) (res_ref1, mask, SIZE);
56  if (UNION_ROUGH_CHECK (AVX512F_LEN, d) (res2, res_ref1, 0.0001))
57    abort ();
58
59  MASK_MERGE (d) (res_ref2, mask, SIZE);
60  if (UNION_ROUGH_CHECK (AVX512F_LEN, d) (res3, res_ref2, 0.0001))
61    abort ();
62
63  MASK_ZERO (d) (res_ref1, mask, SIZE);
64  if (UNION_ROUGH_CHECK (AVX512F_LEN, d) (res4, res_ref1, 0.0001))
65    abort ();
66}
67