gcc/testsuite/gcc.dg/vect/vect-reduc-chain-1.c

   1 /* Disabling epilogues until we find a better way to deal with scans.  */
   2 /* { dg-additional-options "--param vect-epilogues-nomask=0" } */
   3 /* { dg-require-effective-target vect_int } */
   4 /* { dg-require-effective-target arm_v8_2a_dotprod_neon_hw { target { aarch64*-*-* || arm*-*-* } } } */
   5 /* { dg-add-options arm_v8_2a_dotprod_neon }  */
   6
   7 #include "tree-vect.h"
   8
   9 #define N 50
  10
  11 #ifndef SIGNEDNESS_1
  12 #define SIGNEDNESS_1 signed
  13 #define SIGNEDNESS_2 signed
  14 #endif
  15
  16 SIGNEDNESS_1 int __attribute__ ((noipa))
  17 f (SIGNEDNESS_1 int res,
  18    SIGNEDNESS_2 char *restrict a,
  19    SIGNEDNESS_2 char *restrict b,
  20    SIGNEDNESS_2 char *restrict c,
  21    SIGNEDNESS_2 char *restrict d,
  22    SIGNEDNESS_1 int *restrict e)
  23 {
  24   for (int i = 0; i < N; ++i)
  25     {
  26       res += a[i] * b[i];
  27       res += c[i] * d[i];
  28       res += e[i];
  29     }
  30   return res;
  31 }
  32
  33 #define BASE ((SIGNEDNESS_2 int) -1 < 0 ? -126 : 4)
  34 #define OFFSET 20
  35
  36 int
  37 main (void)
  38 {
  39   check_vect ();
  40
  41   SIGNEDNESS_2 char a[N], b[N];
  42   SIGNEDNESS_2 char c[N], d[N];
  43   SIGNEDNESS_1 int e[N];
  44   int expected = 0x12345;
  45
  46   #pragma GCC novector
  47   for (int i = 0; i < N; ++i)
  48     {
  49       a[i] = BASE + i * 5;
  50       b[i] = BASE + OFFSET + i * 4;
  51       c[i] = BASE + i * 2;
  52       d[i] = BASE + OFFSET + i * 3;
  53       e[i] = i;
  54       expected += a[i] * b[i];
  55       expected += c[i] * d[i];
  56       expected += e[i];
  57     }
  58
  59   if (f (0x12345, a, b, c, d, e) != expected)
  60     __builtin_abort ();
  61 }
  62
  63 /* { dg-final { scan-tree-dump "vect_recog_dot_prod_pattern: detected" "vect" } } */
  64 /* { dg-final { scan-tree-dump-times "vectorizing statement: \\S+ = DOT_PROD_EXPR" 2 "vect" { target vect_sdot_qi } } } */