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 (TYPE2 *__restrict r,			\
9 			  TYPE2 *__restrict a,			\
10 			  TYPE1 *__restrict pred)		\
11   {								\
12     for (int i = 0; i < COUNT; ++i)				\
13       r[i] = pred[i] ? !a[i] : a[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 {\tcnot\tz[0-9]+\.b, p[0-7]/m,} 3 } } */
27 /* { dg-final { scan-assembler-times {\tcnot\tz[0-9]+\.h, p[0-7]/m,} 2 } } */
28 /* { dg-final { scan-assembler-times {\tcnot\tz[0-9]+\.s, p[0-7]/m,} 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} } } */
33