1  /* { dg-do compile } */
       2  /* { dg-options "-O2 -ftree-vectorize" } */
       3  
       4  #include <stdint.h>
       5  
       6  #define DEF_LOOP(TYPE, NAME, OP)					\
       7    void __attribute__ ((noipa))						\
       8    test_##TYPE##_##NAME (TYPE *__restrict r, TYPE *__restrict a,		\
       9  			TYPE *__restrict b, int n)			\
      10    {									\
      11      for (int i = 0; i < n; ++i)						\
      12        r[i] = a[i] > 20 ? b[i] OP 3 : 72;				\
      13    }
      14  
      15  #define TEST_TYPE(T, TYPE) \
      16    T (TYPE, shl, <<) \
      17    T (TYPE, shr, >>)
      18  
      19  #define TEST_ALL(T) \
      20    TEST_TYPE (T, int8_t) \
      21    TEST_TYPE (T, uint8_t) \
      22    TEST_TYPE (T, int16_t) \
      23    TEST_TYPE (T, uint16_t) \
      24    TEST_TYPE (T, int32_t) \
      25    TEST_TYPE (T, uint32_t) \
      26    TEST_TYPE (T, int64_t) \
      27    TEST_TYPE (T, uint64_t)
      28  
      29  TEST_ALL (DEF_LOOP)
      30  
      31  /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.b, p[0-7]/m,} 2 } } */
      32  /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.h, p[0-7]/m,} 2 } } */
      33  /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.s, p[0-7]/m,} 2 } } */
      34  /* { dg-final { scan-assembler-times {\tlsl\tz[0-9]+\.d, p[0-7]/m,} 2 } } */
      35  
      36  /* { dg-final { scan-assembler-times {\tasr\tz[0-9]+\.b, p[0-7]/m,} 1 } } */
      37  /* { dg-final { scan-assembler-times {\tasr\tz[0-9]+\.h, p[0-7]/m,} 1 } } */
      38  /* { dg-final { scan-assembler-times {\tasr\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      39  /* { dg-final { scan-assembler-times {\tasr\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      40  
      41  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.b, p[0-7]/m,} 1 } } */
      42  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.h, p[0-7]/m,} 1 } } */
      43  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.s, p[0-7]/m,} 1 } } */
      44  /* { dg-final { scan-assembler-times {\tlsr\tz[0-9]+\.d, p[0-7]/m,} 1 } } */
      45  
      46  /* { dg-final { scan-assembler-not {\tmov\tz[^,]*z} } } */
      47  /* { dg-final { scan-assembler-not {\tmovprfx\t} } } */
      48  /* { dg-final { scan-assembler-times {\tsel\t} 16 } } */