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 : a[i];				\
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-times {\tmovprfx\tz[0-9]+\.b, p[0-7]/m, z[0-9]+\.b\n} 4 } } */
47 /* { dg-final { scan-assembler-times {\tmovprfx\tz[0-9]+\.h, p[0-7]/m, z[0-9]+\.h\n} 4 } } */
48 /* { dg-final { scan-assembler-times {\tmovprfx\tz[0-9]+\.s, p[0-7]/m, z[0-9]+\.s\n} 4 } } */
49 /* { dg-final { scan-assembler-times {\tmovprfx\tz[0-9]+\.d, p[0-7]/m, z[0-9]+\.d\n} 4 } } */
50 
51 /* { dg-final { scan-assembler-not {\tmov\tz[^,]*z} } } */
52 /* { dg-final { scan-assembler-not {\tsel\t} } } */
53