/* { dg-do run } */
/* { dg-require-effective-target avx512f } */
/* { dg-options "-O3 -mavx512f -mprefer-vector-width=512 -save-temps" } */

#include "fma-check.h"
void __attribute__((noipa))
check_fmaddsub (double * __restrict a, double *b, double *c, int n)
{
  for (int i = 0; i < n; ++i)
    {
      a[8*i + 0] = b[8*i + 0] * c[8*i + 0] + a[8*i + 0];
      a[8*i + 1] = b[8*i + 1] * c[8*i + 1] - a[8*i + 1];
      a[8*i + 2] = b[8*i + 2] * c[8*i + 2] + a[8*i + 2];
      a[8*i + 3] = b[8*i + 3] * c[8*i + 3] - a[8*i + 3];
      a[8*i + 4] = b[8*i + 4] * c[8*i + 4] + a[8*i + 4];
      a[8*i + 5] = b[8*i + 5] * c[8*i + 5] - a[8*i + 5];
      a[8*i + 6] = b[8*i + 6] * c[8*i + 6] + a[8*i + 6];
      a[8*i + 7] = b[8*i + 7] * c[8*i + 7] - a[8*i + 7];
    }
}

static void
fma_test (void)
{
  if (!__builtin_cpu_supports ("avx512f"))
    return;
  double a[8], b[8], c[8];
  for (int i = 0; i < 8; ++i)
    {
      a[i] = i;
      b[i] = 3*i;
      c[i] = 7*i;
    }
  check_fmaddsub (a, b, c, 1);
  const double d[8] = { 0., 20., 86., 186., 340., 520., 762., 1022.};
  for (int i = 0; i < 8; ++i)
    if (a[i] != d[i])
      __builtin_abort ();
}

/* { dg-final { scan-assembler {(?n)fmsubadd...pd[ \t].*%zmm[0-9]} } } */