Skip to content

Commit 73863a4

Browse files
committed
[InstCombine] Add tests for folding (add/sub/disjoint_or/icmp C, (ctpop (not x))); NFC
1 parent f9b089a commit 73863a4

File tree

1 file changed

+235
-0
lines changed

1 file changed

+235
-0
lines changed
Lines changed: 235 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,235 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2+
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
3+
4+
declare void @use.i8(i8)
5+
declare i8 @llvm.ctpop.i8(i8)
6+
declare i2 @llvm.ctpop.i2(i2)
7+
declare <2 x i8> @llvm.ctpop.v2i8(<2 x i8>)
8+
9+
define i8 @fold_sub_c_ctpop(i8 %x) {
10+
; CHECK-LABEL: @fold_sub_c_ctpop(
11+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
12+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0:![0-9]+]]
13+
; CHECK-NEXT: [[R:%.*]] = sub nuw nsw i8 12, [[CNT]]
14+
; CHECK-NEXT: ret i8 [[R]]
15+
;
16+
%nx = xor i8 %x, -1
17+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
18+
%r = sub i8 12, %cnt
19+
ret i8 %r
20+
}
21+
22+
define i8 @fold_sub_var_ctpop_fail(i8 %x, i8 %y) {
23+
; CHECK-LABEL: @fold_sub_var_ctpop_fail(
24+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
25+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
26+
; CHECK-NEXT: [[R:%.*]] = sub i8 [[Y:%.*]], [[CNT]]
27+
; CHECK-NEXT: ret i8 [[R]]
28+
;
29+
%nx = xor i8 %x, -1
30+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
31+
%r = sub i8 %y, %cnt
32+
ret i8 %r
33+
}
34+
35+
define <2 x i8> @fold_sub_ctpop_c(<2 x i8> %x) {
36+
; CHECK-LABEL: @fold_sub_ctpop_c(
37+
; CHECK-NEXT: [[NX:%.*]] = xor <2 x i8> [[X:%.*]], <i8 -1, i8 -1>
38+
; CHECK-NEXT: [[CNT:%.*]] = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> [[NX]]), !range [[RNG0]]
39+
; CHECK-NEXT: [[R:%.*]] = add nuw nsw <2 x i8> [[CNT]], <i8 -63, i8 -64>
40+
; CHECK-NEXT: ret <2 x i8> [[R]]
41+
;
42+
%nx = xor <2 x i8> %x, <i8 -1, i8 -1>
43+
%cnt = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> %nx)
44+
%r = sub <2 x i8> %cnt, <i8 63, i8 64>
45+
ret <2 x i8> %r
46+
}
47+
48+
define i8 @fold_add_ctpop_c(i8 %x) {
49+
; CHECK-LABEL: @fold_add_ctpop_c(
50+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
51+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
52+
; CHECK-NEXT: [[R:%.*]] = add nuw nsw i8 [[CNT]], 63
53+
; CHECK-NEXT: ret i8 [[R]]
54+
;
55+
%nx = xor i8 %x, -1
56+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
57+
%r = add i8 %cnt, 63
58+
ret i8 %r
59+
}
60+
61+
define i8 @fold_distjoint_or_ctpop_c(i8 %x) {
62+
; CHECK-LABEL: @fold_distjoint_or_ctpop_c(
63+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
64+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
65+
; CHECK-NEXT: [[R:%.*]] = or disjoint i8 [[CNT]], 64
66+
; CHECK-NEXT: ret i8 [[R]]
67+
;
68+
%nx = xor i8 %x, -1
69+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
70+
%r = or i8 %cnt, 64
71+
ret i8 %r
72+
}
73+
74+
define i8 @fold_or_ctpop_c_fail(i8 %x) {
75+
; CHECK-LABEL: @fold_or_ctpop_c_fail(
76+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
77+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
78+
; CHECK-NEXT: [[R:%.*]] = or i8 [[CNT]], 65
79+
; CHECK-NEXT: ret i8 [[R]]
80+
;
81+
%nx = xor i8 %x, -1
82+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
83+
%r = or i8 %cnt, 65
84+
ret i8 %r
85+
}
86+
87+
define i8 @fold_add_ctpop_var_fail(i8 %x, i8 %y) {
88+
; CHECK-LABEL: @fold_add_ctpop_var_fail(
89+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
90+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
91+
; CHECK-NEXT: [[R:%.*]] = add i8 [[CNT]], [[Y:%.*]]
92+
; CHECK-NEXT: ret i8 [[R]]
93+
;
94+
%nx = xor i8 %x, -1
95+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
96+
%r = add i8 %cnt, %y
97+
ret i8 %r
98+
}
99+
100+
define i1 @fold_icmp_sgt_ctpop_c_i2_fail(i2 %x, i2 %C) {
101+
; CHECK-LABEL: @fold_icmp_sgt_ctpop_c_i2_fail(
102+
; CHECK-NEXT: ret i1 false
103+
;
104+
%nx = xor i2 %x, -1
105+
%cnt = tail call i2 @llvm.ctpop.i2(i2 %nx)
106+
%r = icmp sgt i2 %cnt, 1
107+
ret i1 %r
108+
}
109+
110+
define i1 @fold_cmp_eq_ctpop_c(i8 %x) {
111+
; CHECK-LABEL: @fold_cmp_eq_ctpop_c(
112+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
113+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
114+
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[CNT]], 2
115+
; CHECK-NEXT: ret i1 [[R]]
116+
;
117+
%nx = xor i8 %x, -1
118+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
119+
%r = icmp eq i8 %cnt, 2
120+
ret i1 %r
121+
}
122+
123+
define i1 @fold_cmp_eq_ctpop_c_multiuse_fail(i8 %x) {
124+
; CHECK-LABEL: @fold_cmp_eq_ctpop_c_multiuse_fail(
125+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
126+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
127+
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[CNT]], 2
128+
; CHECK-NEXT: call void @use.i8(i8 [[CNT]])
129+
; CHECK-NEXT: ret i1 [[R]]
130+
;
131+
%nx = xor i8 %x, -1
132+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
133+
%r = icmp eq i8 %cnt, 2
134+
call void @use.i8(i8 %cnt)
135+
ret i1 %r
136+
}
137+
138+
define <2 x i1> @fold_cmp_ne_ctpop_c(<2 x i8> %x) {
139+
; CHECK-LABEL: @fold_cmp_ne_ctpop_c(
140+
; CHECK-NEXT: [[NX:%.*]] = xor <2 x i8> [[X:%.*]], <i8 -1, i8 -1>
141+
; CHECK-NEXT: [[CNT:%.*]] = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> [[NX]]), !range [[RNG0]]
142+
; CHECK-NEXT: [[R:%.*]] = icmp ne <2 x i8> [[CNT]], <i8 44, i8 3>
143+
; CHECK-NEXT: ret <2 x i1> [[R]]
144+
;
145+
%nx = xor <2 x i8> %x, <i8 -1, i8 -1>
146+
%cnt = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> %nx)
147+
%r = icmp ne <2 x i8> %cnt, <i8 44, i8 3>
148+
ret <2 x i1> %r
149+
}
150+
151+
define <2 x i1> @fold_cmp_ne_ctpop_var_fail(<2 x i8> %x, <2 x i8> %y) {
152+
; CHECK-LABEL: @fold_cmp_ne_ctpop_var_fail(
153+
; CHECK-NEXT: [[NX:%.*]] = xor <2 x i8> [[X:%.*]], <i8 -1, i8 -1>
154+
; CHECK-NEXT: [[CNT:%.*]] = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> [[NX]]), !range [[RNG0]]
155+
; CHECK-NEXT: [[R:%.*]] = icmp ne <2 x i8> [[CNT]], [[Y:%.*]]
156+
; CHECK-NEXT: ret <2 x i1> [[R]]
157+
;
158+
%nx = xor <2 x i8> %x, <i8 -1, i8 -1>
159+
%cnt = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> %nx)
160+
%r = icmp ne <2 x i8> %cnt, %y
161+
ret <2 x i1> %r
162+
}
163+
164+
define i1 @fold_cmp_ult_ctpop_c(i8 %x, i8 %y, i1 %cond) {
165+
; CHECK-LABEL: @fold_cmp_ult_ctpop_c(
166+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
167+
; CHECK-NEXT: [[NY:%.*]] = add i8 [[Y:%.*]], 15
168+
; CHECK-NEXT: [[N:%.*]] = select i1 [[COND:%.*]], i8 [[NX]], i8 [[NY]]
169+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[N]]), !range [[RNG0]]
170+
; CHECK-NEXT: [[R:%.*]] = icmp ult i8 [[CNT]], 5
171+
; CHECK-NEXT: ret i1 [[R]]
172+
;
173+
%nx = xor i8 %x, -1
174+
%ny = add i8 %y, 15
175+
%n = select i1 %cond, i8 %nx, i8 %ny
176+
%cnt = call i8 @llvm.ctpop.i8(i8 %n)
177+
%r = icmp ult i8 %cnt, 5
178+
ret i1 %r
179+
}
180+
181+
define i1 @fold_cmp_sle_ctpop_c(i8 %x, i8 %y, i1 %cond) {
182+
; CHECK-LABEL: @fold_cmp_sle_ctpop_c(
183+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -1
184+
; CHECK-NEXT: [[NY:%.*]] = add i8 [[Y:%.*]], 15
185+
; CHECK-NEXT: [[N:%.*]] = select i1 [[COND:%.*]], i8 [[NX]], i8 [[NY]]
186+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[N]]), !range [[RNG0]]
187+
; CHECK-NEXT: [[R:%.*]] = icmp ult i8 [[CNT]], 4
188+
; CHECK-NEXT: ret i1 [[R]]
189+
;
190+
%nx = xor i8 %x, -1
191+
%ny = add i8 %y, 15
192+
%n = select i1 %cond, i8 %nx, i8 %ny
193+
%cnt = call i8 @llvm.ctpop.i8(i8 %n)
194+
%r = icmp sle i8 %cnt, 3
195+
ret i1 %r
196+
}
197+
198+
define i1 @fold_cmp_ult_ctpop_c_no_not_inst_save_fail(i8 %x) {
199+
; CHECK-LABEL: @fold_cmp_ult_ctpop_c_no_not_inst_save_fail(
200+
; CHECK-NEXT: [[NX:%.*]] = xor i8 [[X:%.*]], -2
201+
; CHECK-NEXT: [[CNT:%.*]] = call i8 @llvm.ctpop.i8(i8 [[NX]]), !range [[RNG0]]
202+
; CHECK-NEXT: [[R:%.*]] = icmp ult i8 [[CNT]], 5
203+
; CHECK-NEXT: ret i1 [[R]]
204+
;
205+
%nx = xor i8 %x, -2
206+
%cnt = call i8 @llvm.ctpop.i8(i8 %nx)
207+
%r = icmp ult i8 %cnt, 5
208+
ret i1 %r
209+
}
210+
211+
define <2 x i1> @fold_cmp_ugt_ctpop_c(<2 x i8> %x) {
212+
; CHECK-LABEL: @fold_cmp_ugt_ctpop_c(
213+
; CHECK-NEXT: [[NX:%.*]] = xor <2 x i8> [[X:%.*]], <i8 -1, i8 -1>
214+
; CHECK-NEXT: [[CNT:%.*]] = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> [[NX]]), !range [[RNG0]]
215+
; CHECK-NEXT: [[R:%.*]] = icmp ugt <2 x i8> [[CNT]], <i8 8, i8 6>
216+
; CHECK-NEXT: ret <2 x i1> [[R]]
217+
;
218+
%nx = xor <2 x i8> %x, <i8 -1, i8 -1>
219+
%cnt = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> %nx)
220+
%r = icmp ugt <2 x i8> %cnt, <i8 8, i8 6>
221+
ret <2 x i1> %r
222+
}
223+
224+
define <2 x i1> @fold_cmp_ugt_ctpop_c_out_of_range_fail(<2 x i8> %x) {
225+
; CHECK-LABEL: @fold_cmp_ugt_ctpop_c_out_of_range_fail(
226+
; CHECK-NEXT: [[NX:%.*]] = xor <2 x i8> [[X:%.*]], <i8 -1, i8 -1>
227+
; CHECK-NEXT: [[CNT:%.*]] = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> [[NX]]), !range [[RNG0]]
228+
; CHECK-NEXT: [[R:%.*]] = icmp ugt <2 x i8> [[CNT]], <i8 2, i8 10>
229+
; CHECK-NEXT: ret <2 x i1> [[R]]
230+
;
231+
%nx = xor <2 x i8> %x, <i8 -1, i8 -1>
232+
%cnt = call <2 x i8> @llvm.ctpop.v2i8(<2 x i8> %nx)
233+
%r = icmp ugt <2 x i8> %cnt, <i8 2, i8 10>
234+
ret <2 x i1> %r
235+
}

0 commit comments

Comments
 (0)