1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize -moverride=sve_width=256" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define abs(A) ((A) < 0 ? -(A) : (A))
       7  #define neg(A) (-(A))
       8  #define not(A) (~(A))
       9  
      10  #define DEF_LOOP(TYPE1, TYPE2, COUNT, OP)			\
      11    void __attribute__ ((noipa))					\
      12    test_##TYPE1##_##TYPE2##_##OP (TYPE2 *__restrict r,		\
      13  				 TYPE2 *__restrict a,		\
      14  				 TYPE2 *__restrict b,		\
      15  				 TYPE1 *__restrict pred)	\
      16    {								\
      17      for (int i = 0; i < COUNT; ++i)				\
      18        {								\
      19  	TYPE2 bi = b[i];					\
      20  	r[i] = pred[i] ? OP (a[i]) : bi;			\
      21        }								\
      22    }
      23  
      24  #define TEST_TYPES(T, TYPE1, TYPE2, COUNT) \
      25    T (TYPE1, TYPE2, COUNT, abs) \
      26    T (TYPE1, TYPE2, COUNT, neg) \
      27    T (TYPE1, TYPE2, COUNT, not)
      28  
      29  #define TEST_ALL(T) \
      30    TEST_TYPES (T, int16_t, int8_t, 7) \
      31    TEST_TYPES (T, int32_t, int8_t, 3) \
      32    TEST_TYPES (T, int32_t, int16_t, 3) \
      33    TEST_TYPES (T, int64_t, int8_t, 5) \
      34    TEST_TYPES (T, int64_t, int16_t, 5) \
      35    TEST_TYPES (T, int64_t, int32_t, 5)
      36  
      37  TEST_ALL (DEF_LOOP)
      38  
      39  /* { dg-final { scan-assembler-times {\tabs\tz[0-9]+\.b, p[0-7]/m,} 3 } } */
      40  /* { dg-final { scan-assembler-times {\tabs\tz[0-9]+\.h, p[0-7]/m,} 2 } } */
      41  /* { dg-final { scan-assembler-times {\tabs\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      42  
      43  /* { dg-final { scan-assembler-times {\tneg\tz[0-9]+\.b, p[0-7]/m,} 3 } } */
      44  /* { dg-final { scan-assembler-times {\tneg\tz[0-9]+\.h, p[0-7]/m,} 2 } } */
      45  /* { dg-final { scan-assembler-times {\tneg\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      46  
      47  /* { dg-final { scan-assembler-times {\tnot\tz[0-9]+\.b, p[0-7]/m,} 3 } } */
      48  /* { dg-final { scan-assembler-times {\tnot\tz[0-9]+\.h, p[0-7]/m,} 2 } } */
      49  /* { dg-final { scan-assembler-times {\tnot\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      50  
      51  /* { dg-final { scan-assembler-not {\tmov\tz} } } */
      52  /* { dg-final { scan-assembler-not {\tmovprfx\t} } } */
      53  /* { dg-final { scan-assembler-not {\tsel\t} } } */