gcc/testsuite/gcc.dg/vect/vect-strided-u16-i3.c

   1 #include <stdarg.h>
   2 #include "tree-vect.h"
   3
   4 #define N 128
   5
   6 typedef struct {
   7    unsigned short a;
   8    unsigned short b;
   9    unsigned short c;
  10 } s;
  11
  12 #define A(I) (I)
  13 #define B(I) ((I) * 2)
  14 #define C(I) ((unsigned short) ~((I) ^ 0x18))
  15
  16 void __attribute__ ((noinline))
  17 check1 (s *res)
  18 {
  19   int i;
  20
  21 #pragma GCC novector
  22   for (i = 0; i < N; i++)
  23     if (res[i].a != C (i)
  24         || res[i].b != A (i)
  25         || res[i].c != B (i))
  26       abort ();
  27 }
  28
  29 void __attribute__ ((noinline))
  30 check2 (unsigned short *res)
  31 {
  32   int i;
  33
  34 #pragma GCC novector
  35   for (i = 0; i < N; i++)
  36     if (res[i] != (unsigned short) (A (i) + B (i) + C (i)))
  37       abort ();
  38 }
  39
  40 void __attribute__ ((noinline))
  41 check3 (s *res)
  42 {
  43   int i;
  44
  45 #pragma GCC novector
  46   for (i = 0; i < N; i++)
  47     if (res[i].a != i
  48         || res[i].b != i
  49         || res[i].c != i)
  50       abort ();
  51 }
  52
  53 void __attribute__ ((noinline))
  54 check4 (unsigned short *res)
  55 {
  56   int i;
  57
  58 #pragma GCC novector
  59   for (i = 0; i < N; i++)
  60     if (res[i] != (unsigned short) (A (i) + B (i)))
  61       abort ();
  62 }
  63
  64 void __attribute__ ((noinline))
  65 main1 (s *arr)
  66 {
  67   int i;
  68   s *ptr = arr;
  69   s res1[N];
  70   unsigned short res2[N];
  71
  72   for (i = 0; i < N; i++)
  73     {
  74       res1[i].a = arr[i].c;
  75       res1[i].b = arr[i].a;
  76       res1[i].c = arr[i].b;
  77     }
  78   check1 (res1);
  79
  80   for (i = 0; i < N; i++)
  81     res2[i] = arr[i].a + arr[i].b + arr[i].c;
  82   check2 (res2);
  83
  84   for (i = 0; i < N; i++)
  85     {
  86       res1[i].a = i;
  87       res1[i].b = i;
  88       res1[i].c = i;
  89     }
  90   check3 (res1);
  91
  92   for (i = 0; i < N; i++)
  93     res2[i] = arr[i].a + arr[i].b;
  94   check4 (res2);
  95 }
  96
  97 int main (void)
  98 {
  99   int i;
 100   s arr[N];
 101
 102   check_vect ();
 103
 104   for (i = 0; i < N; i++)
 105     {
 106       arr[i].a = A (i);
 107       arr[i].b = B (i);
 108       arr[i].c = C (i);
 109     }
 110   main1 (arr);
 111
 112   return 0;
 113 }
 114
 115 /* { dg-final { scan-tree-dump-times "vectorized 4 loops" 1 "vect"  { target vect_strided3 } } } */