gcc/testsuite/gcc.dg/vect/vect-cond-arith-7.c

   1 /* { dg-require-effective-target scalar_all_fma } */
   2 /* { dg-additional-options "-fdump-tree-optimized -ffp-contract=fast" } */
   3
   4 #include "tree-vect.h"
   5
   6 #define N (VECTOR_BITS * 11 / 64 + 3)
   7
   8 #define DEF(INV)                                        \
   9   void __attribute__ ((noipa))                          \
  10   f_##INV (double *restrict a, double *restrict b,      \
  11            double *restrict c, double *restrict d)      \
  12   {                                                     \
  13     for (int i = 0; i < N; ++i)                         \
  14       {                                                 \
  15         double mb = (INV & 1 ? -b[i] : b[i]);           \
  16         double mc = c[i];                               \
  17         double md = (INV & 2 ? -d[i] : d[i]);           \
  18         a[i] = b[i] < 10 ? mb * mc + md : 10.0;         \
  19       }                                                 \
  20   }
  21
  22 #define TEST(INV)                                       \
  23   {                                                     \
  24     f_##INV (a, b, c, d);                               \
  25     _Pragma("GCC novector")                             \
  26     for (int i = 0; i < N; ++i)                         \
  27       {                                                 \
  28         double mb = (INV & 1 ? -b[i] : b[i]);           \
  29         double mc = c[i];                               \
  30         double md = (INV & 2 ? -d[i] : d[i]);           \
  31         double fma = __builtin_fma (mb, mc, md);        \
  32         if (a[i] != (i % 17 < 10 ? fma : 10.0))         \
  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)
  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 { = \.COND_FMA } 1 "optimized" { target vect_double_cond_arith } } } */
  59 /* { dg-final { scan-tree-dump-times { = \.COND_FMS } 1 "optimized" { target vect_double_cond_arith } } } */
  60 /* { dg-final { scan-tree-dump-times { = \.COND_FNMA } 1 "optimized" { target vect_double_cond_arith } } } */
  61 /* { dg-final { scan-tree-dump-times { = \.COND_FNMS } 1 "optimized" { target vect_double_cond_arith } } } */