Skip to content

Commit d603d6e

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

File tree

1 file changed

+268
-0
lines changed

1 file changed

+268
-0
lines changed

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

Lines changed: 268 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,268 @@
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 <2 x i1> @umul_greater_than_or_overflow_const_vector_splat(<2 x i64> %in) {
121+
; CHECK-LABEL: define <2 x i1> @umul_greater_than_or_overflow_const_vector_splat(
122+
; CHECK-SAME: <2 x i64> [[IN:%.*]]) {
123+
; 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))
124+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[TMP2]], 0
125+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[TMP2]], 1
126+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt <2 x i64> [[TMP3]], splat (i64 9223372036854775800)
127+
; CHECK-NEXT: [[TMP6:%.*]] = or <2 x i1> [[TMP4]], [[TMP5]]
128+
; CHECK-NEXT: ret <2 x i1> [[TMP6]]
129+
;
130+
%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>)
131+
%mul = extractvalue { <2 x i64>, <2 x i1> } %mwo, 0
132+
%ovf = extractvalue { <2 x i64>, <2 x i1> } %mwo, 1
133+
%cmp = icmp ugt <2 x i64> %mul, <i64 9223372036854775800, i64 9223372036854775800>
134+
%ret = or <2 x i1> %ovf, %cmp
135+
ret <2 x i1> %ret
136+
}
137+
138+
; Negative test
139+
define <4 x i1> @umul_greater_than_or_overflow_const_vector_non_splat_negative(<4 x i64> %in) {
140+
; CHECK-LABEL: define <4 x i1> @umul_greater_than_or_overflow_const_vector_non_splat_negative(
141+
; CHECK-SAME: <4 x i64> [[IN:%.*]]) {
142+
; 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>)
143+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { <4 x i64>, <4 x i1> } [[MWO]], 0
144+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { <4 x i64>, <4 x i1> } [[MWO]], 1
145+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt <4 x i64> [[MUL]], <i64 9223372036854775000, i64 9223372036854775800, i64 -16, i64 -16>
146+
; CHECK-NEXT: [[RET:%.*]] = or <4 x i1> [[OVF]], [[CMP]]
147+
; CHECK-NEXT: ret <4 x i1> [[RET]]
148+
;
149+
%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>)
150+
%mul = extractvalue { <4 x i64>, <4 x i1> } %mwo, 0
151+
%ovf = extractvalue { <4 x i64>, <4 x i1> } %mwo, 1
152+
%cmp = icmp ugt <4 x i64> %mul, <i64 9223372036854775000, i64 9223372036854775800, i64 -16, i64 -16>
153+
%ret = or <4 x i1> %ovf, %cmp
154+
ret <4 x i1> %ret
155+
}
156+
157+
; Negative test
158+
define <2 x i1> @umul_greater_than_or_overflow_const_vector_poison_non_splat_negative(<2 x i64> %in) {
159+
; CHECK-LABEL: define <2 x i1> @umul_greater_than_or_overflow_const_vector_poison_non_splat_negative(
160+
; CHECK-SAME: <2 x i64> [[IN:%.*]]) {
161+
; 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>)
162+
; CHECK-NEXT: [[MUL:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[MWO]], 0
163+
; CHECK-NEXT: [[OVF:%.*]] = extractvalue { <2 x i64>, <2 x i1> } [[MWO]], 1
164+
; CHECK-NEXT: [[CMP:%.*]] = icmp ugt <2 x i64> [[MUL]], <i64 9223372036854775800, i64 poison>
165+
; CHECK-NEXT: [[RET:%.*]] = or <2 x i1> [[OVF]], [[CMP]]
166+
; CHECK-NEXT: ret <2 x i1> [[RET]]
167+
;
168+
%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>)
169+
%mul = extractvalue { <2 x i64>, <2 x i1> } %mwo, 0
170+
%ovf = extractvalue { <2 x i64>, <2 x i1> } %mwo, 1
171+
%cmp = icmp ugt <2 x i64> %mul, <i64 9223372036854775800, i64 poison>
172+
%ret = or <2 x i1> %ovf, %cmp
173+
ret <2 x i1> %ret
174+
}
175+
176+
; Negative test
177+
define i1 @umul_greater_than_and_overflow_const_negative(i64 %in) {
178+
; CHECK-LABEL: define i1 @umul_greater_than_and_overflow_const_negative(
179+
; CHECK-SAME: i64 [[IN:%.*]]) {
180+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
181+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
182+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
183+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ult i64 [[TMP3]], 9223372036854775800
184+
; CHECK-NEXT: [[TMP6:%.*]] = and i1 [[TMP4]], [[TMP5]]
185+
; CHECK-NEXT: ret i1 [[TMP6]]
186+
;
187+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
188+
%mul = extractvalue { i64, i1 } %mwo, 0
189+
%ovf = extractvalue { i64, i1 } %mwo, 1
190+
%cmp = icmp ult i64 %mul, 9223372036854775800
191+
%ret = and i1 %ovf, %cmp
192+
ret i1 %ret
193+
}
194+
195+
; Negative test
196+
define i1 @umul_less_than_or_overflow_const_negative(i64 %in) {
197+
; CHECK-LABEL: define i1 @umul_less_than_or_overflow_const_negative(
198+
; CHECK-SAME: i64 [[IN:%.*]]) {
199+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
200+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
201+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
202+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ult i64 [[TMP3]], 9223372036854775800
203+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
204+
; CHECK-NEXT: ret i1 [[TMP6]]
205+
;
206+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
207+
%mul = extractvalue { i64, i1 } %mwo, 0
208+
%ovf = extractvalue { i64, i1 } %mwo, 1
209+
%cmp = icmp ult i64 %mul, 9223372036854775800
210+
%ret = or i1 %ovf, %cmp
211+
ret i1 %ret
212+
}
213+
214+
; Negative test
215+
define i1 @umul_greater_than_or_overflow_const_multiuse_icmp_negative(i64 %in) {
216+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_multiuse_icmp_negative(
217+
; CHECK-SAME: i64 [[IN:%.*]]) {
218+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
219+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
220+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
221+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i64 [[TMP3]], 9223372036854775800
222+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
223+
; CHECK-NEXT: tail call void @use.i1(i1 [[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 ugt i64 %mul, 9223372036854775800
230+
%ret = or i1 %ovf, %cmp
231+
tail call void @use.i1(i1 %cmp)
232+
ret i1 %ret
233+
}
234+
235+
; Negative test
236+
define i1 @umul_greater_than_or_overflow_const_multiuse_umul_call_negative(i64 %in) {
237+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_multiuse_umul_call_negative(
238+
; CHECK-SAME: i64 [[IN:%.*]]) {
239+
; CHECK-NEXT: [[TMP2:%.*]] = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 [[IN]], i64 48)
240+
; CHECK-NEXT: [[TMP3:%.*]] = extractvalue { i64, i1 } [[TMP2]], 0
241+
; CHECK-NEXT: [[TMP4:%.*]] = extractvalue { i64, i1 } [[TMP2]], 1
242+
; CHECK-NEXT: [[TMP5:%.*]] = icmp ugt i64 [[TMP3]], 9223372036854775800
243+
; CHECK-NEXT: [[TMP6:%.*]] = or i1 [[TMP4]], [[TMP5]]
244+
; CHECK-NEXT: tail call void @use.i64i1({ i64, i1 } [[TMP2]])
245+
; CHECK-NEXT: ret i1 [[TMP6]]
246+
;
247+
%mwo = tail call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 48)
248+
%mul = extractvalue { i64, i1 } %mwo, 0
249+
%ovf = extractvalue { i64, i1 } %mwo, 1
250+
%cmp = icmp ugt i64 %mul, 9223372036854775800
251+
%ret = or i1 %ovf, %cmp
252+
tail call void @use.i64i1({ i64, i1 } %mwo)
253+
ret i1 %ret
254+
}
255+
256+
; Negative test. The umul.with.overflow should be folded away before.
257+
define i1 @umul_greater_than_or_overflow_const_0_negative(i64 %in) {
258+
; CHECK-LABEL: define i1 @umul_greater_than_or_overflow_const_0_negative(
259+
; CHECK-SAME: i64 [[IN:%.*]]) {
260+
; CHECK-NEXT: ret i1 false
261+
;
262+
%mwo = call { i64, i1 } @llvm.umul.with.overflow.i64(i64 %in, i64 0)
263+
%mul = extractvalue { i64, i1 } %mwo, 0
264+
%ovf = extractvalue { i64, i1 } %mwo, 1
265+
%cmp = icmp ugt i64 %mul, 0
266+
%ret = or i1 %ovf, %cmp
267+
ret i1 %ret
268+
}

0 commit comments

Comments
 (0)