1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize -fdump-tree-vect-details --save-temps" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define AVERAGE(TYPE, BIGGER, RND)					  \
       7  TYPE __attribute__ ((noinline, noclone))				  \
       8  avg_##TYPE##_##RND (TYPE *restrict x, TYPE *restrict y, TYPE *restrict z, \
       9  		    int n)						  \
      10  {									  \
      11    for (int i = 0; i < n; i++)						  \
      12      {									  \
      13        z[i] = ((BIGGER)x[i] + y[i] + RND) >> 1;				  \
      14      }									  \
      15  }
      16  
      17  AVERAGE (int8_t, int64_t, 0)
      18  AVERAGE (int16_t, int64_t, 0)
      19  AVERAGE (int32_t, int64_t, 0)
      20  AVERAGE (uint8_t, uint64_t, 0)
      21  AVERAGE (uint16_t, uint64_t, 0)
      22  AVERAGE (uint32_t, uint64_t, 0)
      23  AVERAGE (int8_t, int64_t, 1)
      24  AVERAGE (int16_t, int64_t, 1)
      25  AVERAGE (int32_t, int64_t, 1)
      26  AVERAGE (uint8_t, uint64_t, 1)
      27  AVERAGE (uint16_t, uint64_t, 1)
      28  AVERAGE (uint32_t, uint64_t, 1)
      29  
      30  /* { dg-final { scan-tree-dump-times "vectorized 1 loops in function" 12 "vect" } } */
      31  
      32  /* { dg-final { scan-assembler-times {\tshadd\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, z[0-9]+\.b\n} 1 } } */
      33  /* { dg-final { scan-assembler-times {\tshadd\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, z[0-9]+\.h\n} 1 } } */
      34  /* { dg-final { scan-assembler-times {\tshadd\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, z[0-9]+\.s\n} 1 } } */
      35  
      36  /* { dg-final { scan-assembler-times {\tuhadd\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, z[0-9]+\.b\n} 1 } } */
      37  /* { dg-final { scan-assembler-times {\tuhadd\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, z[0-9]+\.h\n} 1 } } */
      38  /* { dg-final { scan-assembler-times {\tuhadd\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, z[0-9]+\.s\n} 1 } } */
      39  
      40  /* { dg-final { scan-assembler-times {\tsrhadd\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, z[0-9]+\.b\n} 1 } } */
      41  /* { dg-final { scan-assembler-times {\tsrhadd\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, z[0-9]+\.h\n} 1 } } */
      42  /* { dg-final { scan-assembler-times {\tsrhadd\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, z[0-9]+\.s\n} 1 } } */
      43  
      44  /* { dg-final { scan-assembler-times {\turhadd\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, z[0-9]+\.b\n} 1 } } */
      45  /* { dg-final { scan-assembler-times {\turhadd\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, z[0-9]+\.h\n} 1 } } */
      46  /* { dg-final { scan-assembler-times {\turhadd\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, z[0-9]+\.s\n} 1 } } */