1  /* { dg-options "-O2 -ftree-vectorize" } */
       2  
       3  #include <stdint.h>
       4  
       5  #define TEST_LOOP(TYPE1, TYPE2, SHIFT)					\
       6    void									\
       7    f_##TYPE1##_##TYPE2 (TYPE2 *restrict dst, TYPE1 *restrict src1,	\
       8  		       TYPE1 *restrict src2, int n)			\
       9    {									\
      10      for (int i = 0; i < n; ++i)						\
      11        dst[i] = (TYPE1) (src1[i] + src2[i]) >> SHIFT;			\
      12    }
      13  
      14  #define TEST_ALL(T) \
      15    T (uint16_t, uint8_t, 2) \
      16    T (uint32_t, uint8_t, 18) \
      17    T (uint64_t, uint8_t, 34) \
      18    T (uint32_t, uint16_t, 3) \
      19    T (uint64_t, uint16_t, 19) \
      20    T (uint64_t, uint32_t, 4)
      21  
      22  TEST_ALL (TEST_LOOP)
      23  
      24  /* { dg-final { scan-assembler-times {\tld1h\tz[0-9]+\.h,} 2 } } */
      25  /* { dg-final { scan-assembler-times {\tld1w\tz[0-9]+\.s,} 4 } } */
      26  /* { dg-final { scan-assembler-times {\tld1d\tz[0-9]+\.d,} 6 } } */
      27  
      28  /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.h, z[0-9]+\.h, z[0-9]+\.h\n} 1 } } */
      29  /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.s, z[0-9]+\.s, z[0-9]+\.s\n} 2 } } */
      30  /* { dg-final { scan-assembler-times {\tadd\tz[0-9]+\.d, z[0-9]+\.d, z[0-9]+\.d\n} 3 } } */
      31  
      32  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.h, z[0-9]+\.h, #2\n} 1 } } */
      33  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.s, z[0-9]+\.s, #18\n} 1 } } */
      34  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.d, z[0-9]+\.d, #34\n} 1 } } */
      35  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.s, z[0-9]+\.s, #3\n} 1 } } */
      36  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.d, z[0-9]+\.d, #19\n} 1 } } */
      37  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.d, z[0-9]+\.d, #4\n} 1 } } */
      38  
      39  /* { dg-final { scan-assembler-times {\tst1b\tz[0-9]+\.h,} 1 } } */
      40  /* { dg-final { scan-assembler-times {\tst1b\tz[0-9]+\.s,} 1 } } */
      41  /* { dg-final { scan-assembler-times {\tst1b\tz[0-9]+\.d,} 1 } } */
      42  /* { dg-final { scan-assembler-times {\tst1h\tz[0-9]+\.s,} 1 } } */
      43  /* { dg-final { scan-assembler-times {\tst1h\tz[0-9]+\.d,} 1 } } */
      44  /* { dg-final { scan-assembler-times {\tst1w\tz[0-9]+\.d,} 1 } } */