Skip to content

Commit e5cb233

Browse files
mskampmskamp
committed
[InstCombine] Add Tests for umul.overflow(x, c1) | x*c1 > c2; NFC
1 parent 4c27279 commit e5cb233

File tree

1 file changed

+267
-0
lines changed

1 file changed

+267
-0
lines changed

‎llvm/test/Transforms/InstCombine/icmp_or_umul_overflow.ll

Lines changed: 267 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,267 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2+
; RUN: opt -S -passes=instcombine < %s | FileCheck %s
3+
4+
declare void @use.i1(i1 %x)
5+
declare void @use.i64(i64 %x)
6+
declare void @use.i64i1({i64, i1} %x)
7+
8+
define i1 @umul_greater_than_or_overflow_const(i64 %in) {
9+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const(
10+
; CHECK-SAME: i64 [[IN:%.*]]) {
11+
; CHECK-NEXT: [[TMP2:%.*]] = call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 168)
12+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
13+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
14+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i64 [[TMP3]], -16
15+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
16+
; CHECK-NEXT: ret i1 [[TMP6]]
17+
;
18+
%mwo = call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 168)
19+
%mul = extractvalue { i64, i1 } %mwo, 0
20+
%ovf = extractvalue { i64, i1 } %mwo, 1
21+
%cmp = icmp ugt i64 %mul, -16
22+
%ret = or i1 %ovf, %cmp
23+
ret i1 %ret
24+
}
25+
26+
define i1 @umul_greater_than_or_overflow_const_i8(i8 %in) {
27+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_i8(
28+
; CHECK-SAME: i8 [[IN:%.*]]) {
29+
; CHECK-NEXT: [[TMP2:%.*]] = call { i8, i1 } @llvm.umul.with.overflow.i8(i8 [[IN]], i8 24)
30+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i8, i1 } [[TMP2]], 0
31+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i8, i1 } [[TMP2]], 1
32+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i8 [[TMP3]], -16
33+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
34+
; CHECK-NEXT: ret i1 [[TMP6]]
35+
;
36+
%mwo = call { i8, i1 } @llvm.umul.with.overflow.i8(i8 %in, i8 24)
37+
%mul = extractvalue { i8, i1 } %mwo, 0
38+
%ovf = extractvalue { i8, i1 } %mwo, 1
39+
%cmp = icmp ugt i8 %mul, -16
40+
%ret = or i1 %ovf, %cmp
41+
ret i1 %ret
42+
}
43+
44+
define i1 @umul_greater_than_or_overflow_const_commuted(i64 %in) {
45+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_commuted(
46+
; CHECK-SAME: i64 [[IN:%.*]]) {
47+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
48+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
49+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
50+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i64 [[TMP3]], 9223372036854775800
51+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP5]], [[TMP4]]
52+
; CHECK-NEXT: ret i1 [[TMP6]]
53+
;
54+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
55+
%mul = extractvalue { i64, i1 } %mwo, 0
56+
%ovf = extractvalue { i64, i1 } %mwo, 1
57+
%cmp = icmp ugt i64 %mul, 9223372036854775800
58+
%ret = or i1 %cmp, %ovf
59+
ret i1 %ret
60+
}
61+
62+
define i1 @umul_greater_than_or_overflow_const_disjoint(i64 %in) {
63+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_disjoint(
64+
; CHECK-SAME: i64 [[IN:%.*]]) {
65+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 40)
66+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
67+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
68+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i64 [[TMP3]], 9223372036854775800
69+
; CHECK-NEXT: [[TMP6:%.*]] = or disjoint i1 [[TMP4]], [[TMP5]]
70+
; CHECK-NEXT: ret i1 [[TMP6]]
71+
;
72+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 40)
73+
%mul = extractvalue { i64, i1 } %mwo, 0
74+
%ovf = extractvalue { i64, i1 } %mwo, 1
75+
%cmp = icmp ugt i64 %mul, 9223372036854775800
76+
%ret = or disjoint i1 %ovf, %cmp
77+
ret i1 %ret
78+
}
79+
80+
define i1 @umul_greater_than_or_overflow_const_multiuse_mul(i64 %in) {
81+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_multiuse_mul(
82+
; CHECK-SAME: i64 [[IN:%.*]]) {
83+
; CHECK-NEXT: [[MWO:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
84+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { i64, i1 } [[MWO]], 0
85+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { i64, i1 } [[MWO]], 1
86+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[MUL]], 9223372036854775800
87+
; CHECK-NEXT: [[RET:%.*]] = or i1 [[OVF]], [[CMP]]
88+
; CHECK-NEXT: tail call void @use.i64(i64 [[MUL]])
89+
; CHECK-NEXT: ret i1 [[RET]]
90+
;
91+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
92+
%mul = extractvalue { i64, i1 } %mwo, 0
93+
%ovf = extractvalue { i64, i1 } %mwo, 1
94+
%cmp = icmp ugt i64 %mul, 9223372036854775800
95+
%ret = or i1 %ovf, %cmp
96+
tail call void @use.i64(i64 %mul)
97+
ret i1 %ret
98+
}
99+
100+
define i1 @umul_greater_than_or_overflow_const_multiuse_overflow(i64 %in) {
101+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_multiuse_overflow(
102+
; CHECK-SAME: i64 [[IN:%.*]]) {
103+
; CHECK-NEXT: [[MWO:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
104+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { i64, i1 } [[MWO]], 0
105+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { i64, i1 } [[MWO]], 1
106+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[MUL]], 9223372036854775800
107+
; CHECK-NEXT: [[RET:%.*]] = or i1 [[OVF]], [[CMP]]
108+
; CHECK-NEXT: tail call void @use.i1(i1 [[OVF]])
109+
; CHECK-NEXT: ret i1 [[RET]]
110+
;
111+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
112+
%mul = extractvalue { i64, i1 } %mwo, 0
113+
%ovf = extractvalue { i64, i1 } %mwo, 1
114+
%cmp = icmp ugt i64 %mul, 9223372036854775800
115+
%ret = or i1 %ovf, %cmp
116+
tail call void @use.i1(i1 %ovf)
117+
ret i1 %ret
118+
}
119+
120+
define i1 @umul_greater_than_or_overflow_const_multiuse_umul_call(i64 %in) {
121+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_multiuse_umul_call(
122+
; CHECK-SAME: i64 [[IN:%.*]]) {
123+
; CHECK-NEXT: [[MWO:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
124+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { i64, i1 } [[MWO]], 0
125+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { i64, i1 } [[MWO]], 1
126+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt i64 [[MUL]], 9223372036854775800
127+
; CHECK-NEXT: [[RET:%.*]] = or i1 [[OVF]], [[CMP]]
128+
; CHECK-NEXT: tail call void @use.i64i1({ i64, i1 } [[MWO]])
129+
; CHECK-NEXT: ret i1 [[RET]]
130+
;
131+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
132+
%mul = extractvalue { i64, i1 } %mwo, 0
133+
%ovf = extractvalue { i64, i1 } %mwo, 1
134+
%cmp = icmp ugt i64 %mul, 9223372036854775800
135+
%ret = or i1 %ovf, %cmp
136+
tail call void @use.i64i1({ i64, i1 } %mwo)
137+
ret i1 %ret
138+
}
139+
140+
define <2 x i1> @umul_greater_than_or_overflow_const_vector_splat(<2 x i64> %in) {
141+
; CHECK-LABEL: define <2 x i1> @umul_greater_than_or_overflow_const_vector_splat(
142+
; CHECK-SAME: <2 x i64> [[IN:%.*]]) {
143+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { <2 x i64>, <2 x i1> } @llvm.umul.with.overflow.v2i64(<2 x i64> [[IN]], <2 x i64> splat (i64 1424))
144+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[TMP2]], 0
145+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[TMP2]], 1
146+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt <2 x i64> [[TMP3]], splat (i64 9223372036854775800)
147+
; CHECK-NEXT: [[TMP6:%.*]] = or <2 x i1> [[TMP4]], [[TMP5]]
148+
; CHECK-NEXT: ret <2 x i1> [[TMP6]]
149+
;
150+
%mwo = tail call { <2 x i64>, <2 x i1> } @llvm.umul.with.overflow.v2i64(<2 x i64> %in, <2 x i64> <i64 1424, i64 1424>)
151+
%mul = extractvalue { <2 x i64>, <2 x i1> } %mwo, 0
152+
%ovf = extractvalue { <2 x i64>, <2 x i1> } %mwo, 1
153+
%cmp = icmp ugt <2 x i64> %mul, <i64 9223372036854775800, i64 9223372036854775800>
154+
%ret = or <2 x i1> %ovf, %cmp
155+
ret <2 x i1> %ret
156+
}
157+
158+
; Negative test
159+
define <4 x i1> @umul_greater_than_or_overflow_const_vector_non_splat_negative(<4 x i64> %in) {
160+
; CHECK-LABEL: define <4 x i1> @umul_greater_than_or_overflow_const_vector_non_splat_negative(
161+
; CHECK-SAME: <4 x i64> [[IN:%.*]]) {
162+
; CHECK-NEXT: [[MWO:%.*]] = tail call { <4 x i64>, <4 x i1> } @llvm.umul.with.overflow.v4i64(<4 x i64> [[IN]], <4 x i64> <i64 24, i64 1424, i64 0, i64 -1>)
163+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { <4 x i64>, <4 x i1> } [[MWO]], 0
164+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { <4 x i64>, <4 x i1> } [[MWO]], 1
165+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt <4 x i64> [[MUL]], <i64 9223372036854775000, i64 9223372036854775800, i64 -16, i64 -16>
166+
; CHECK-NEXT: [[RET:%.*]] = or <4 x i1> [[OVF]], [[CMP]]
167+
; CHECK-NEXT: ret <4 x i1> [[RET]]
168+
;
169+
%mwo = tail call { <4 x i64>, <4 x i1> } @llvm.umul.with.overflow.v2i64(<4 x i64> %in, <4 x i64> <i64 24, i64 1424, i64 0, i64 -1>)
170+
%mul = extractvalue { <4 x i64>, <4 x i1> } %mwo, 0
171+
%ovf = extractvalue { <4 x i64>, <4 x i1> } %mwo, 1
172+
%cmp = icmp ugt <4 x i64> %mul, <i64 9223372036854775000, i64 9223372036854775800, i64 -16, i64 -16>
173+
%ret = or <4 x i1> %ovf, %cmp
174+
ret <4 x i1> %ret
175+
}
176+
177+
; Negative test
178+
define <2 x i1> @umul_greater_than_or_overflow_const_vector_poison_non_splat_negative(<2 x i64> %in) {
179+
; CHECK-LABEL: define <2 x i1> @umul_greater_than_or_overflow_const_vector_poison_non_splat_negative(
180+
; CHECK-SAME: <2 x i64> [[IN:%.*]]) {
181+
; CHECK-NEXT: [[MWO:%.*]] = tail call { <2 x i64>, <2 x i1> } @llvm.umul.with.overflow.v2i64(<2 x i64> [[IN]], <2 x i64> <i64 poison, i64 1424>)
182+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[MWO]], 0
183+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[MWO]], 1
184+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt <2 x i64> [[MUL]], <i64 9223372036854775800, i64 poison>
185+
; CHECK-NEXT: [[RET:%.*]] = or <2 x i1> [[OVF]], [[CMP]]
186+
; CHECK-NEXT: ret <2 x i1> [[RET]]
187+
;
188+
%mwo = tail call { <2 x i64>, <2 x i1> } @llvm.umul.with.overflow.v2i64(<2 x i64> %in, <2 x i64> <i64 poison, i64 1424>)
189+
%mul = extractvalue { <2 x i64>, <2 x i1> } %mwo, 0
190+
%ovf = extractvalue { <2 x i64>, <2 x i1> } %mwo, 1
191+
%cmp = icmp ugt <2 x i64> %mul, <i64 9223372036854775800, i64 poison>
192+
%ret = or <2 x i1> %ovf, %cmp
193+
ret <2 x i1> %ret
194+
}
195+
196+
; Negative test
197+
define i1 @umul_greater_than_and_overflow_const_negative(i64 %in) {
198+
; CHECK-LABEL: define i1 @umul_greater_than_and_overflow_const_negative(
199+
; CHECK-SAME: i64 [[IN:%.*]]) {
200+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
201+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
202+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
203+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ult i64 [[TMP3]], 9223372036854775800
204+
; CHECK-NEXT: [[TMP6:%.*]] = and i1 [[TMP4]], [[TMP5]]
205+
; CHECK-NEXT: ret i1 [[TMP6]]
206+
;
207+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
208+
%mul = extractvalue { i64, i1 } %mwo, 0
209+
%ovf = extractvalue { i64, i1 } %mwo, 1
210+
%cmp = icmp ult i64 %mul, 9223372036854775800
211+
%ret = and i1 %ovf, %cmp
212+
ret i1 %ret
213+
}
214+
215+
; Negative test
216+
define i1 @umul_less_than_or_overflow_const_negative(i64 %in) {
217+
; CHECK-LABEL: define i1 @umul_less_than_or_overflow_const_negative(
218+
; CHECK-SAME: i64 [[IN:%.*]]) {
219+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
220+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
221+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
222+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ult i64 [[TMP3]], 9223372036854775800
223+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
224+
; CHECK-NEXT: ret i1 [[TMP6]]
225+
;
226+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
227+
%mul = extractvalue { i64, i1 } %mwo, 0
228+
%ovf = extractvalue { i64, i1 } %mwo, 1
229+
%cmp = icmp ult i64 %mul, 9223372036854775800
230+
%ret = or i1 %ovf, %cmp
231+
ret i1 %ret
232+
}
233+
234+
; Negative test
235+
define i1 @umul_greater_than_or_overflow_const_multiuse_icmp_negative(i64 %in) {
236+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_multiuse_icmp_negative(
237+
; CHECK-SAME: i64 [[IN:%.*]]) {
238+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
239+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
240+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
241+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i64 [[TMP3]], 9223372036854775800
242+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
243+
; CHECK-NEXT: tail call void @use.i1(i1 [[TMP5]])
244+
; CHECK-NEXT: ret i1 [[TMP6]]
245+
;
246+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
247+
%mul = extractvalue { i64, i1 } %mwo, 0
248+
%ovf = extractvalue { i64, i1 } %mwo, 1
249+
%cmp = icmp ugt i64 %mul, 9223372036854775800
250+
%ret = or i1 %ovf, %cmp
251+
tail call void @use.i1(i1 %cmp)
252+
ret i1 %ret
253+
}
254+
255+
; Negative test. The umul.with.overflow should be folded away before.
256+
define i1 @umul_greater_than_or_overflow_const_0_negative(i64 %in) {
257+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_0_negative(
258+
; CHECK-SAME: i64 [[IN:%.*]]) {
259+
; CHECK-NEXT: ret i1 false
260+
;
261+
%mwo = call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 0)
262+
%mul = extractvalue { i64, i1 } %mwo, 0
263+
%ovf = extractvalue { i64, i1 } %mwo, 1
264+
%cmp = icmp ugt i64 %mul, 0
265+
%ret = or i1 %ovf, %cmp
266+
ret i1 %ret
267+
}

0 commit comments

Comments
 (0)