1 /* Make sure that vector comaprison results are not unnecessarily ANDed
2    with vectors of 1.  */
3 /* { dg-do compile } */
4 /* { dg-options "-O2 -ftree-vectorize" } */
5 
6 #pragma GCC target "+nosve"
7 
8 #define COUNT1(X) if (X) count += 1
9 #define COUNT2(X) if (X) count -= 1
10 #define COUNT3(X) count += (X)
11 #define COUNT4(X) count -= (X)
12 
13 #define COND1(X) (X)
14 #define COND2(X) ((X) ? 1 : 0)
15 #define COND3(X) ((X) ? -1 : 0)
16 #define COND4(X) ((X) ? 0 : 1)
17 #define COND5(X) ((X) ? 0 : -1)
18 
19 #define TEST_LT(X, Y) ((X) < (Y))
20 #define TEST_LE(X, Y) ((X) <= (Y))
21 #define TEST_GT(X, Y) ((X) > (Y))
22 #define TEST_GE(X, Y) ((X) >= (Y))
23 #define TEST_EQ(X, Y) ((X) == (Y))
24 #define TEST_NE(X, Y) ((X) != (Y))
25 
26 #define COUNT_LOOP(ID, TYPE, CMP_ARRAY, TEST, COUNT) \
27   TYPE \
28   reduc_##ID (__typeof__ (CMP_ARRAY[0]) x) \
29   { \
30     TYPE count = 0; \
31     for (unsigned int i = 0; i < 1024; ++i) \
32       COUNT (TEST (CMP_ARRAY[i], x)); \
33     return count; \
34   }
35 
36 #define COND_LOOP(ID, ARRAY, CMP_ARRAY, TEST, COND) \
37   void \
38   plus_##ID (__typeof__ (CMP_ARRAY[0]) x) \
39   { \
40     for (unsigned int i = 0; i < 1024; ++i) \
41       ARRAY[i] += COND (TEST (CMP_ARRAY[i], x)); \
42   } \
43   void \
44   plusc_##ID (void) \
45   { \
46     for (unsigned int i = 0; i < 1024; ++i) \
47       ARRAY[i] += COND (TEST (CMP_ARRAY[i], 10)); \
48   } \
49   void \
50   minus_##ID (__typeof__ (CMP_ARRAY[0]) x) \
51   { \
52     for (unsigned int i = 0; i < 1024; ++i) \
53       ARRAY[i] -= COND (TEST (CMP_ARRAY[i], x)); \
54   } \
55   void \
56   minusc_##ID (void) \
57   { \
58     for (unsigned int i = 0; i < 1024; ++i) \
59       ARRAY[i] += COND (TEST (CMP_ARRAY[i], 1)); \
60   }
61 
62 #define ALL_LOOPS(ID, ARRAY, CMP_ARRAY, TEST) \
63   typedef __typeof__(ARRAY[0]) ID##_type; \
64   COUNT_LOOP (ID##_1, ID##_type, CMP_ARRAY, TEST, COUNT1) \
65   COUNT_LOOP (ID##_2, ID##_type, CMP_ARRAY, TEST, COUNT2) \
66   COUNT_LOOP (ID##_3, ID##_type, CMP_ARRAY, TEST, COUNT3) \
67   COUNT_LOOP (ID##_4, ID##_type, CMP_ARRAY, TEST, COUNT4) \
68   COND_LOOP (ID##_1, ARRAY, CMP_ARRAY, TEST, COND1) \
69   COND_LOOP (ID##_2, ARRAY, CMP_ARRAY, TEST, COND2) \
70   COND_LOOP (ID##_3, ARRAY, CMP_ARRAY, TEST, COND3) \
71   COND_LOOP (ID##_4, ARRAY, CMP_ARRAY, TEST, COND4) \
72   COND_LOOP (ID##_5, ARRAY, CMP_ARRAY, TEST, COND5)
73 
74 signed int asi[1024] __attribute__ ((aligned (16)));
75 unsigned int aui[1024] __attribute__ ((aligned (16)));
76 signed long long asl[1024] __attribute__ ((aligned (16)));
77 unsigned long long aul[1024] __attribute__ ((aligned (16)));
78 float af[1024] __attribute__ ((aligned (16)));
79 double ad[1024] __attribute__ ((aligned (16)));
80 
81 ALL_LOOPS (si_si, aui, asi, TEST_LT)
82 ALL_LOOPS (ui_si, aui, asi, TEST_LE)
83 ALL_LOOPS (si_ui, aui, asi, TEST_GT)
84 ALL_LOOPS (ui_ui, aui, asi, TEST_GE)
85 ALL_LOOPS (sl_sl, asl, asl, TEST_NE)
86 ALL_LOOPS (ul_ul, aul, aul, TEST_EQ)
87 ALL_LOOPS (si_f, asi, af, TEST_LE)
88 ALL_LOOPS (ui_f, aui, af, TEST_GT)
89 ALL_LOOPS (sl_d, asl, ad, TEST_GE)
90 ALL_LOOPS (ul_d, aul, ad, TEST_GT)
91 
92 /* { dg-final { scan-assembler-not "\tand\t" } } */
93 /* { dg-final { scan-assembler-not "\tld\[^\t\]*\t\[wx\]" } } */
94 /* { dg-final { scan-assembler-not "\tst\[^\t\]*\t\[wx\]" } } */
95 /* { dg-final { scan-assembler "\tldr\tq" } } */
96 /* { dg-final { scan-assembler "\tstr\tq" } } */
97