1  /* { dg-require-effective-target scalar_all_fma } */
       2  
       3  #include "tree-vect.h"
       4  
       5  #define N (VECTOR_BITS * 11 / 64 + 3)
       6  
       7  #define DEF(INV)					\
       8    void __attribute__ ((noipa))				\
       9    f_##INV (double *restrict a, double *restrict b,	\
      10  	   double *restrict c, double *restrict d)	\
      11    {							\
      12      for (int i = 0; i < N; ++i)				\
      13        {							\
      14  	double mb = (INV & 1 ? -b[i] : b[i]);		\
      15  	double mc = c[i];				\
      16  	double md = (INV & 2 ? -d[i] : d[i]);		\
      17  	double fma = __builtin_fma (mb, mc, md);	\
      18  	a[i] = (INV & 4 ? -fma : fma);			\
      19        }							\
      20    }
      21  
      22  #define TEST(INV)					\
      23    {							\
      24      f_##INV (a, b, c, d);				\
      25      for (int i = 0; i < N; ++i)				\
      26        {							\
      27  	double mb = (INV & 1 ? -b[i] : b[i]);		\
      28  	double mc = c[i];				\
      29  	double md = (INV & 2 ? -d[i] : d[i]);		\
      30  	double fma = __builtin_fma (mb, mc, md);	\
      31  	double expected = (INV & 4 ? -fma : fma);	\
      32  	if (a[i] != expected)				\
      33  	  __builtin_abort ();				\
      34  	asm volatile ("" ::: "memory");			\
      35        }							\
      36    }
      37  
      38  #define FOR_EACH_INV(T)	\
      39    T (0) T (1) T (2) T (3) T (4) T (5) T (6) T (7)
      40  
      41  FOR_EACH_INV (DEF)
      42  
      43  int
      44  main (void)
      45  {
      46    double a[N], b[N], c[N], d[N];
      47    for (int i = 0; i < N; ++i)
      48      {
      49        b[i] = i % 17;
      50        c[i] = i % 9 + 11;
      51        d[i] = i % 13 + 14;
      52        asm volatile ("" ::: "memory");
      53      }
      54    FOR_EACH_INV (TEST)
      55    return 0;
      56  }
      57  
      58  /* { dg-final { scan-tree-dump-times "LOOP VECTORIZED" 8 "vect" { target vect_double } } } */