1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define DEF_LOOP(TYPE1, TYPE2, COUNT)				\
       7    void __attribute__ ((noipa))					\
       8    test_##TYPE1##_##TYPE2##_##TYPE3 (TYPE2 *restrict r,		\
       9  				    TYPE1 *restrict pred,	\
      10  				    TYPE2 *restrict a)		\
      11    {								\
      12      for (int i = 0; i < COUNT; ++i)				\
      13        if (pred[i])						\
      14  	r[i] = a[i] / 16;					\
      15    }
      16  
      17  #define TEST_ALL(T) \
      18    T (int16_t, int8_t, 7) \
      19    T (int32_t, int8_t, 3) \
      20    T (int32_t, int16_t, 3) \
      21    T (int64_t, int8_t, 5) \
      22    T (int64_t, int16_t, 5) \
      23    T (int64_t, int32_t, 5)
      24  
      25  TEST_ALL (DEF_LOOP)
      26  
      27  /* { dg-final { scan-assembler-times {\tasrd\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b, #4\n} 3 } } */
      28  /* { dg-final { scan-assembler-times {\tasrd\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h, #4\n} 2 } } */
      29  /* { dg-final { scan-assembler-times {\tasrd\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s, #4\n} 1 } } */