xref: /llvm-project/llvm/test/Transforms/NaryReassociate/nary-umax.ll (revision 63691eef9ba7f0e54448ec7ccdd27ef23fea4ddd) 
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt < %s -passes=nary-reassociate -S | FileCheck %s
3; RUN: opt < %s -passes='nary-reassociate' -S | FileCheck %s
4
5declare i32 @llvm.umax.i32(i32 %a, i32 %b)
6
7; m1 = umax(a,b) ; has side uses
8; m2 = umax(umax((b,c), a) -> m2 = umax(m1, c)
9define i32 @umax_test1(i32 %a, i32 %b, i32 %c) {
10; CHECK-LABEL: @umax_test1(
11; CHECK-NEXT:    [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]
12; CHECK-NEXT:    [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]
13; CHECK-NEXT:    [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])
14; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]
15; CHECK-NEXT:    ret i32 [[RES]]
16;
17  %c1 = icmp ugt i32 %a, %b
18  %umax1 = select i1 %c1, i32 %a, i32 %b
19  %c2 = icmp ugt i32 %b, %c
20  %umax2 = select i1 %c2, i32 %b, i32 %c
21  %c3 = icmp ugt i32 %umax2, %a
22  %umax3 = select i1 %c3, i32 %umax2, i32 %a
23  %res = add i32 %umax1, %umax3
24  ret i32 %res
25}
26
27; m1 = umax(a,b) ; has side uses
28; m2 = umax(b, (umax(a, c))) -> m2 = umax(m1, c)
29define i32 @umax_test2(i32 %a, i32 %b, i32 %c) {
30; CHECK-LABEL: @umax_test2(
31; CHECK-NEXT:    [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]
32; CHECK-NEXT:    [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]
33; CHECK-NEXT:    [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])
34; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]
35; CHECK-NEXT:    ret i32 [[RES]]
36;
37  %c1 = icmp ugt i32 %a, %b
38  %umax1 = select i1 %c1, i32 %a, i32 %b
39  %c2 = icmp ugt i32 %a, %c
40  %umax2 = select i1 %c2, i32 %a, i32 %c
41  %c3 = icmp ugt i32 %b, %umax2
42  %umax3 = select i1 %c3, i32 %b, i32 %umax2
43  %res = add i32 %umax1, %umax3
44  ret i32 %res
45}
46
47; Same test as umax_test1 but uses @llvm.umax intrinsic
48define i32 @umax_test3(i32 %a, i32 %b, i32 %c) {
49; CHECK-LABEL: @umax_test3(
50; CHECK-NEXT:    [[UMAX1:%.*]] = call i32 @llvm.umax.i32(i32 [[A:%.*]], i32 [[B:%.*]])
51; CHECK-NEXT:    [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])
52; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]
53; CHECK-NEXT:    ret i32 [[RES]]
54;
55  %umax1 = call i32 @llvm.umax.i32(i32 %a, i32 %b)
56  %umax2 = call i32 @llvm.umax.i32(i32 %b, i32 %c)
57  %umax3 = call i32 @llvm.umax.i32(i32 %umax2, i32 %a)
58  %res = add i32 %umax1, %umax3
59  ret i32 %res
60}
61
62; m1 = umax(a,b) ; has side uses
63; m2 = umax(umax_or_eq((b,c), a) -> m2 = umax(m1, c)
64define i32 @umax_test4(i32 %a, i32 %b, i32 %c) {
65; CHECK-LABEL: @umax_test4(
66; CHECK-NEXT:    [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]
67; CHECK-NEXT:    [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]
68; CHECK-NEXT:    [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])
69; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]
70; CHECK-NEXT:    ret i32 [[RES]]
71;
72  %c1 = icmp ugt i32 %a, %b
73  %umax1 = select i1 %c1, i32 %a, i32 %b
74  %c2 = icmp uge i32 %b, %c
75  %umax_or_eq2 = select i1 %c2, i32 %b, i32 %c
76  %c3 = icmp ugt i32 %umax_or_eq2, %a
77  %umax3 = select i1 %c3, i32 %umax_or_eq2, i32 %a
78  %res = add i32 %umax1, %umax3
79  ret i32 %res
80}
81
82; m1 = umax_or_eq(a,b) ; has side uses
83; m2 = umax_or_eq(umax((b,c), a) -> m2 = umax(m1, c)
84define i32 @umax_test5(i32 %a, i32 %b, i32 %c) {
85; CHECK-LABEL: @umax_test5(
86; CHECK-NEXT:    [[C1:%.*]] = icmp uge i32 [[A:%.*]], [[B:%.*]]
87; CHECK-NEXT:    [[UMAX_OR_EQ1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]
88; CHECK-NEXT:    [[UMAX_OR_EQ3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX_OR_EQ1]], i32 [[C:%.*]])
89; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX_OR_EQ1]], [[UMAX_OR_EQ3_NARY]]
90; CHECK-NEXT:    ret i32 [[RES]]
91;
92  %c1 = icmp uge i32 %a, %b
93  %umax_or_eq1 = select i1 %c1, i32 %a, i32 %b
94  %c2 = icmp ugt i32 %b, %c
95  %umax2 = select i1 %c2, i32 %b, i32 %c
96  %c3 = icmp uge i32 %umax2, %a
97  %umax_or_eq3 = select i1 %c3, i32 %umax2, i32 %a
98  %res = add i32 %umax_or_eq1, %umax_or_eq3
99  ret i32 %res
100}
101
102; m1 = umax(a,b) ; has side uses
103; m2 = umax(smax((b,c), a) ; check that signed and unsigned maxs are not mixed
104define i32 @umax_test6(i32 %a, i32 %b, i32 %c) {
105; CHECK-LABEL: @umax_test6(
106; CHECK-NEXT:    [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]
107; CHECK-NEXT:    [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]
108; CHECK-NEXT:    [[C2:%.*]] = icmp sgt i32 [[B]], [[C:%.*]]
109; CHECK-NEXT:    [[SMAX2:%.*]] = select i1 [[C2]], i32 [[B]], i32 [[C]]
110; CHECK-NEXT:    [[C3:%.*]] = icmp ugt i32 [[SMAX2]], [[A]]
111; CHECK-NEXT:    [[UMAX3:%.*]] = select i1 [[C3]], i32 [[SMAX2]], i32 [[A]]
112; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3]]
113; CHECK-NEXT:    ret i32 [[RES]]
114;
115  %c1 = icmp ugt i32 %a, %b
116  %umax1 = select i1 %c1, i32 %a, i32 %b
117  %c2 = icmp sgt i32 %b, %c
118  %smax2 = select i1 %c2, i32 %b, i32 %c
119  %c3 = icmp ugt i32 %smax2, %a
120  %umax3 = select i1 %c3, i32 %smax2, i32 %a
121  %res = add i32 %umax1, %umax3
122  ret i32 %res
123}
124
125; m1 = umax(a,b) ; has side uses
126; m2 = umax(umin((b,c), a) ; check that max and min are not mixed
127define i32 @umax_test7(i32 %a, i32 %b, i32 %c) {
128; CHECK-LABEL: @umax_test7(
129; CHECK-NEXT:    [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]
130; CHECK-NEXT:    [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]
131; CHECK-NEXT:    [[C2:%.*]] = icmp ult i32 [[B]], [[C:%.*]]
132; CHECK-NEXT:    [[UMAX2:%.*]] = select i1 [[C2]], i32 [[B]], i32 [[C]]
133; CHECK-NEXT:    [[C3:%.*]] = icmp ugt i32 [[UMAX2]], [[A]]
134; CHECK-NEXT:    [[UMAX3:%.*]] = select i1 [[C3]], i32 [[UMAX2]], i32 [[A]]
135; CHECK-NEXT:    [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3]]
136; CHECK-NEXT:    ret i32 [[RES]]
137;
138  %c1 = icmp ugt i32 %a, %b
139  %umax1 = select i1 %c1, i32 %a, i32 %b
140  %c2 = icmp ult i32 %b, %c
141  %umax2 = select i1 %c2, i32 %b, i32 %c
142  %c3 = icmp ugt i32 %umax2, %a
143  %umax3 = select i1 %c3, i32 %umax2, i32 %a
144  %res = add i32 %umax1, %umax3
145  ret i32 %res
146}
147
148; Pointer types are not supported yet
149define ptr @umax_test8(ptr %a, ptr %b, ptr %c) {
150; CHECK-LABEL: @umax_test8(
151; CHECK-NEXT:    [[C1:%.*]] = icmp ugt ptr [[A:%.*]], [[B:%.*]]
152; CHECK-NEXT:    [[SMAX1:%.*]] = select i1 [[C1]], ptr [[A]], ptr [[B]]
153; CHECK-NEXT:    [[C2:%.*]] = icmp ugt ptr [[B]], [[C:%.*]]
154; CHECK-NEXT:    [[SMAX2:%.*]] = select i1 [[C2]], ptr [[B]], ptr [[C]]
155; CHECK-NEXT:    [[C3:%.*]] = icmp ugt ptr [[SMAX2]], [[A]]
156; CHECK-NEXT:    [[SMAX3:%.*]] = select i1 [[C3]], ptr [[SMAX2]], ptr [[A]]
157; CHECK-NEXT:    [[C4:%.*]] = icmp ugt ptr [[SMAX1]], [[SMAX3]]
158; CHECK-NEXT:    [[RES:%.*]] = select i1 [[C4]], ptr [[SMAX1]], ptr [[SMAX3]]
159; CHECK-NEXT:    ret ptr [[RES]]
160;
161  %c1 = icmp ugt ptr %a, %b
162  %smax1 = select i1 %c1, ptr %a, ptr %b
163  %c2 = icmp ugt ptr %b, %c
164  %smax2 = select i1 %c2, ptr %b, ptr %c
165  %c3 = icmp ugt ptr %smax2, %a
166  %smax3 = select i1 %c3, ptr %smax2, ptr %a
167  %c4 = icmp ugt ptr %smax1, %smax3
168  %res = select i1 %c4, ptr %smax1, ptr %smax3
169  ret ptr %res
170}
171