gcc/testsuite/gcc.dg/vect/vect-avg-11.c

   1 /* { dg-require-effective-target vect_int } */
   2
   3 #include "tree-vect.h"
   4
   5 #define N 50
   6
   7 #ifndef SIGNEDNESS
   8 #define SIGNEDNESS unsigned
   9 #endif
  10 #ifndef BIAS
  11 #define BIAS 0
  12 #endif
  13
  14 void __attribute__ ((noipa))
  15 f (SIGNEDNESS char *restrict a, SIGNEDNESS char *restrict b,
  16    SIGNEDNESS char *restrict c)
  17 {
  18   for (__INTPTR_TYPE__ i = 0; i < N; ++i)
  19     {
  20       int tmp = b[i];
  21       tmp ^= 0x55;
  22       tmp += BIAS;
  23       tmp += c[i];
  24       tmp >>= 1;
  25       tmp |= 0x40;
  26       a[i] = tmp;
  27     }
  28 }
  29
  30 #define BASE1 ((SIGNEDNESS int) -1 < 0 ? -126 : 4)
  31 #define BASE2 ((SIGNEDNESS int) -1 < 0 ? -101 : 26)
  32
  33 int
  34 main (void)
  35 {
  36   check_vect ();
  37
  38   SIGNEDNESS char a[N], b[N], c[N];
  39   for (int i = 0; i < N; ++i)
  40     {
  41       b[i] = BASE1 + i * 5;
  42       c[i] = BASE2 + i * 4;
  43       asm volatile ("" ::: "memory");
  44     }
  45   f (a, b, c);
  46   for (int i = 0; i < N; ++i)
  47     if (a[i] != (((((BASE1 + i * 5) ^ 0x55)
  48                    + (BASE2 + i * 4)
  49                    + BIAS) >> 1) | 0x40))
  50       __builtin_abort ();
  51   return 0;
  52 }
  53
  54 /* { dg-final { scan-tree-dump "vect_recog_average_pattern: detected" "vect" } } */
  55 /* { dg-final { scan-tree-dump {\.AVG_FLOOR} "vect" { target vect_avg_qi } } } */
  56 /* { dg-final { scan-tree-dump-not {vector\([^\n]*short} "vect" { target vect_avg_qi } } } */
  57 /* { dg-final { scan-tree-dump-times "vectorized 1 loop" 1 "vect" { target vect_avg_qi } } } */