slp-perm-8.c

   1 /* { dg-require-effective-target vect_int } */
   2
   3 #include <stdarg.h>
   4 #include "tree-vect.h"
   5
   6 #define N 200
   7
   8 void foo (unsigned char *__restrict__ pInput, unsigned char *__restrict__ pOutput)
   9 {
  10   unsigned char i, a, b, c;
  11
  12   for (i = 0; i < N / 3; i++)
  13     {
  14        a = *pInput++;
  15        b = *pInput++;
  16        c = *pInput++;
  17
  18        *pOutput++ = a + b + c + 3;
  19        *pOutput++ = a + b + c + 12;
  20        *pOutput++ = a + b + c + 1;
  21     }
  22 }
  23
  24 int main (int argc, const char* argv[])
  25 {
  26   unsigned char input[N], output[N], i;
  27   unsigned char check_results[N];
  28
  29   check_vect ();
  30
  31   for (i = 0; i < N; i++)
  32     {
  33       input[i] = i;
  34       output[i] = 0;
  35       if (input[i] > 256)
  36         abort ();
  37     }
  38
  39   for (i = 0; i < N / 3; i++)
  40     {
  41       check_results[3*i] = 9 * i + 6;
  42       check_results[3*i+1] = 9 * i + 15;
  43       check_results[3*i+2] = 9 * i + 4;
  44     }
  45
  46   foo (input, output);
  47
  48   for (i = 0; i < N - (N % 3); i++)
  49      if (output[i] != check_results[i])
  50        abort ();
  51
  52   return 0;
  53 }
  54
  55 /* { dg-final { scan-tree-dump-times "vectorized 1 loops" 2 "vect" { target vect_perm_byte } } } */
  56 /* { dg-final { scan-tree-dump-times "vectorizing stmts using SLP" 1 "vect" { target vect_perm_byte } } } */
  57 /* { dg-final { cleanup-tree-dump "vect" } } */
  58