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