gcc/testsuite/gcc.dg/vect/slp-perm-5.c

   1 /* { dg-require-effective-target vect_int } */
   2
   3 #include <stdarg.h>
   4 #include <stdio.h>
   5 #include "tree-vect.h"
   6
   7 #define M00 100
   8 #define M10 216
   9 #define M20 23
  10 #define M01 1322
  11 #define M11 13
  12 #define M21 27271
  13 #define M02 74
  14 #define M12 191
  15 #define M22 500
  16
  17 #define K00 405
  18 #define K10 112
  19 #define K01 4322
  20 #define K11 135
  21
  22 #define N 16
  23
  24 void foo (int *__restrict__ pInput, int *__restrict__ pOutput,
  25           int *__restrict__ pInput2, int *__restrict__ pOutput2)
  26 {
  27   int i, a, b, c, d, e;
  28
  29   for (i = 0; i < N / 3; i++)
  30     {
  31        a = *pInput++;
  32        b = *pInput++;
  33        c = *pInput++;
  34
  35        d = *pInput2++;
  36        e = *pInput2++;
  37
  38        *pOutput++ = M00 * a + M01 * b + M02 * c;
  39        *pOutput++ = M10 * a + M11 * b + M12 * c;
  40        *pOutput++ = M20 * a + M21 * b + M22 * c;
  41
  42        *pOutput2++ = K00 * d + K01 * e;
  43        *pOutput2++ = K10 * d + K11 * e;
  44     }
  45 }
  46
  47 int main (int argc, const char* argv[])
  48 {
  49   int input[N], output[N], i;
  50   int check_results[N] = {1470, 395, 28271, 5958, 1655, 111653, 10446, 2915, 195035, 14934, 4175, 278417, 19422, 5435, 361799, 0};
  51   int input2[N], output2[N];
  52   int check_results2[N] = {4322, 135, 13776, 629, 23230, 1123, 32684, 1617, 42138, 2111, 0, 0, 0, 0, 0, 0};
  53
  54   check_vect ();
  55
  56   check_vect ();
  57
  58   for (i = 0; i < N; i++)
  59     {
  60       input[i] = i%256;
  61       input2[i] = i%256;
  62       output[i] = 0;
  63       output2[i] = 0;
  64       if (input[i] > 256)
  65         abort ();
  66     }
  67
  68   foo (input, output, input2, output2);
  69
  70   for (i = 0; i < N; i++)
  71      if (output[i] != check_results[i] || output2[i] != check_results2[i])
  72        abort ();
  73
  74   return 0;
  75 }
  76
  77 /* { dg-final { scan-tree-dump-times "vectorized 1 loops" 1 "vect"  { target vect_perm } } } */
  78 /* { dg-final { scan-tree-dump-times "vectorizing stmts using SLP" 2 "vect" { target vect_perm } } } */
  79 /* { dg-final { cleanup-tree-dump "vect" } } */
  80
  81