1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize -fdump-tree-vect-details --save-temps" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define SHRACC(TYPE,SHIFT)			\
       7  void __attribute__ ((noinline, noclone))	\
       8  f_##TYPE##_##SHIFT				\
       9    (TYPE *restrict a, TYPE *restrict b, int n)	\
      10  {						\
      11    for (int i = 0; i < n; i++)			\
      12      a[i] += b[i] >> (SHIFT);			\
      13  }
      14  
      15  SHRACC (int8_t, 5);
      16  SHRACC (int16_t, 14);
      17  SHRACC (int32_t, 19);
      18  SHRACC (int64_t, 27);
      19  
      20  SHRACC (uint8_t, 2);
      21  SHRACC (uint16_t, 6);
      22  SHRACC (uint32_t, 24);
      23  SHRACC (uint64_t, 53);
      24  
      25  /* { dg-final { scan-tree-dump-times "vectorized 1 loops in function" 8 "vect" } } */
      26  
      27  /* { dg-final { scan-assembler-not {\tasr\tz[0-9]+\.[bhsd]} } } */
      28  /* { dg-final { scan-assembler-not {\tlsr\tz[0-9]+\.[bhsd]} } } */
      29  /* { dg-final { scan-assembler-not {\tadd\tz[0-9]+\.[bhsd]} } } */
      30  
      31  /* { dg-final { scan-assembler-times {\tssra\tz[0-9]+\.b, z[0-9]+\.b, #5\n} 1 } } */
      32  /* { dg-final { scan-assembler-times {\tssra\tz[0-9]+\.h, z[0-9]+\.h, #14\n} 1 } } */
      33  /* { dg-final { scan-assembler-times {\tssra\tz[0-9]+\.s, z[0-9]+\.s, #19\n} 1 } } */
      34  /* { dg-final { scan-assembler-times {\tssra\tz[0-9]+\.d, z[0-9]+\.d, #27\n} 1 } } */
      35  
      36  /* { dg-final { scan-assembler-times {\tusra\tz[0-9]+\.b, z[0-9]+\.b, #2\n} 1 } } */
      37  /* { dg-final { scan-assembler-times {\tusra\tz[0-9]+\.h, z[0-9]+\.h, #6\n} 1 } } */
      38  /* { dg-final { scan-assembler-times {\tusra\tz[0-9]+\.s, z[0-9]+\.s, #24\n} 1 } } */
      39  /* { dg-final { scan-assembler-times {\tusra\tz[0-9]+\.d, z[0-9]+\.d, #53\n} 1 } } */