[LLVM][IR] Use splat syntax when printing ConstantExpr based splats. (#116856)
[llvm-project.git] / llvm / test / Transforms / CorrelatedValuePropagation / select.ll
blobbe027fbc698744f46b2e62a0e97b1d806b7c4e0f
1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
2 ; RUN: opt < %s -passes=correlated-propagation -S | FileCheck %s
4 declare void @use(i8, i8)
6 define i8 @simple_phi(i1 %c, i8 %a, i8 %b) {
7 ; CHECK-LABEL: define i8 @simple_phi
8 ; CHECK-SAME: (i1 [[C:%.*]], i8 [[A:%.*]], i8 [[B:%.*]]) {
9 ; CHECK-NEXT:  entry:
10 ; CHECK-NEXT:    br i1 [[C]], label [[THEN:%.*]], label [[ELSE:%.*]]
11 ; CHECK:       then:
12 ; CHECK-NEXT:    ret i8 [[A]]
13 ; CHECK:       else:
14 ; CHECK-NEXT:    ret i8 [[B]]
16 entry:
17   %s = select i1 %c, i8 %a, i8 %b
18   br i1 %c, label %then, label %else
20 then:
21   %phi1 = phi i8 [ %s, %entry ]
22   ret i8 %phi1
24 else:
25   %phi2 = phi i8 [ %s, %entry ]
26   ret i8 %phi2
29 define i8 @phi_other_edge(i1 %c, i8 %a, i8 %b, i32 %sw) {
30 ; CHECK-LABEL: define i8 @phi_other_edge
31 ; CHECK-SAME: (i1 [[C:%.*]], i8 [[A:%.*]], i8 [[B:%.*]], i32 [[SW:%.*]]) {
32 ; CHECK-NEXT:  entry:
33 ; CHECK-NEXT:    switch i32 [[SW]], label [[TEST:%.*]] [
34 ; CHECK-NEXT:      i32 0, label [[THEN:%.*]]
35 ; CHECK-NEXT:      i32 1, label [[ELSE:%.*]]
36 ; CHECK-NEXT:    ]
37 ; CHECK:       test:
38 ; CHECK-NEXT:    br i1 [[C]], label [[THEN]], label [[ELSE]]
39 ; CHECK:       then:
40 ; CHECK-NEXT:    [[PHI1:%.*]] = phi i8 [ [[A]], [[TEST]] ], [ 1, [[ENTRY:%.*]] ]
41 ; CHECK-NEXT:    ret i8 [[PHI1]]
42 ; CHECK:       else:
43 ; CHECK-NEXT:    [[PHI2:%.*]] = phi i8 [ [[B]], [[TEST]] ], [ 2, [[ENTRY]] ]
44 ; CHECK-NEXT:    ret i8 [[PHI2]]
46 entry:
47   switch i32 %sw, label %test [
48   i32 0, label %then
49   i32 1, label %else
50   ]
52 test:
53   %s = select i1 %c, i8 %a, i8 %b
54   br i1 %c, label %then, label %else
56 then:
57   %phi1 = phi i8 [ %s, %test ], [ 1, %entry ]
58   ret i8 %phi1
60 else:
61   %phi2 = phi i8 [ %s, %test ], [ 2, %entry ]
62   ret i8 %phi2
65 define void @phi_loop(i32) {
66 ; CHECK-LABEL: define void @phi_loop
67 ; CHECK-SAME: (i32 [[TMP0:%.*]]) {
68 ; CHECK-NEXT:  entry:
69 ; CHECK-NEXT:    br label [[LOOP:%.*]]
70 ; CHECK:       loop:
71 ; CHECK-NEXT:    [[IDX:%.*]] = phi i32 [ [[TMP0]], [[ENTRY:%.*]] ], [ [[TMP2:%.*]], [[LOOP]] ]
72 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i32 [[IDX]], 0
73 ; CHECK-NEXT:    [[TMP2]] = add i32 [[IDX]], -1
74 ; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[TMP1]], i32 0, i32 [[TMP2]]
75 ; CHECK-NEXT:    br i1 [[TMP1]], label [[OUT:%.*]], label [[LOOP]]
76 ; CHECK:       out:
77 ; CHECK-NEXT:    ret void
79 entry:
80   br label %loop
82 loop:
83   %idx = phi i32 [ %0, %entry ], [ %sel, %loop ]
84   %1 = icmp eq i32 %idx, 0
85   %2 = add i32 %idx, -1
86   %sel = select i1 %1, i32 0, i32 %2
87   br i1 %1, label %out, label %loop
89 out:
90   ret void
93 define i8 @simple_non_phi(i1 %c, i8 %a, i8 %b) {
94 ; CHECK-LABEL: define i8 @simple_non_phi
95 ; CHECK-SAME: (i1 [[C:%.*]], i8 [[A:%.*]], i8 [[B:%.*]]) {
96 ; CHECK-NEXT:  entry:
97 ; CHECK-NEXT:    br i1 [[C]], label [[THEN:%.*]], label [[ELSE:%.*]]
98 ; CHECK:       then:
99 ; CHECK-NEXT:    ret i8 [[A]]
100 ; CHECK:       else:
101 ; CHECK-NEXT:    ret i8 [[B]]
103 entry:
104   %s = select i1 %c, i8 %a, i8 %b
105   br i1 %c, label %then, label %else
107 then:
108   ret i8 %s
110 else:
111   ret i8 %s
115 define void @simple_multiple_uses(i1 %c, i8 %a, i8 %b) {
116 ; CHECK-LABEL: define void @simple_multiple_uses
117 ; CHECK-SAME: (i1 [[C:%.*]], i8 [[A:%.*]], i8 [[B:%.*]]) {
118 ; CHECK-NEXT:  entry:
119 ; CHECK-NEXT:    [[S:%.*]] = select i1 [[C]], i8 [[A]], i8 [[B]]
120 ; CHECK-NEXT:    call void @use(i8 [[S]], i8 [[S]])
121 ; CHECK-NEXT:    br i1 [[C]], label [[THEN:%.*]], label [[ELSE:%.*]]
122 ; CHECK:       then:
123 ; CHECK-NEXT:    call void @use(i8 [[A]], i8 [[A]])
124 ; CHECK-NEXT:    ret void
125 ; CHECK:       else:
126 ; CHECK-NEXT:    call void @use(i8 [[B]], i8 [[B]])
127 ; CHECK-NEXT:    ret void
129 entry:
130   %s = select i1 %c, i8 %a, i8 %b
131   call void @use(i8 %s, i8 %s)
132   br i1 %c, label %then, label %else
134 then:
135   call void @use(i8 %s, i8 %s)
136   ret void
138 else:
139   call void @use(i8 %s, i8 %s)
140   ret void
143 define i8 @not_correlated(i1, i1) {
144 ; CHECK-LABEL: define range(i8 0, 2) i8 @not_correlated
145 ; CHECK-SAME: (i1 [[TMP0:%.*]], i1 [[TMP1:%.*]]) {
146 ; CHECK-NEXT:  entry:
147 ; CHECK-NEXT:    [[S:%.*]] = select i1 [[TMP0]], i8 0, i8 1
148 ; CHECK-NEXT:    br i1 [[TMP1]], label [[THEN:%.*]], label [[ELSE:%.*]]
149 ; CHECK:       then:
150 ; CHECK-NEXT:    ret i8 [[S]]
151 ; CHECK:       else:
152 ; CHECK-NEXT:    ret i8 [[S]]
154 entry:
155   %s = select i1 %0, i8 0, i8 1
156   br i1 %1, label %then, label %else
158 then:
159   %a = phi i8 [ %s, %entry ]
160   ret i8 %a
162 else:
163   %b = phi i8 [ %s, %entry ]
164   ret i8 %b
167 @c = global i32 0, align 4
168 @b = global i32 0, align 4
170 define i32 @PR23752() {
171 ; CHECK-LABEL: define i32 @PR23752() {
172 ; CHECK-NEXT:  entry:
173 ; CHECK-NEXT:    br label [[FOR_BODY:%.*]]
174 ; CHECK:       for.body:
175 ; CHECK-NEXT:    [[CMP2:%.*]] = icmp sgt ptr @b, @c
176 ; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP2]], i32 0, i32 1
177 ; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[SEL]], 1
178 ; CHECK-NEXT:    br i1 [[CMP]], label [[FOR_BODY]], label [[IF_END:%.*]]
179 ; CHECK:       if.end:
180 ; CHECK-NEXT:    ret i32 1
182 entry:
183   br label %for.body
185 for.body:
186   %phi = phi i32 [ 0, %entry ], [ %sel, %for.body ]
187   %cmp2 = icmp sgt ptr @b, @c
188   %sel = select i1 %cmp2, i32 %phi, i32 1
189   %cmp = icmp ne i32 %sel, 1
190   br i1 %cmp, label %for.body, label %if.end
193 if.end:
194   ret i32 %sel
197 define i1 @test1(ptr %p, i1 %unknown) {
198 ; CHECK-LABEL: define i1 @test1
199 ; CHECK-SAME: (ptr [[P:%.*]], i1 [[UNKNOWN:%.*]]) {
200 ; CHECK-NEXT:    [[PVAL:%.*]] = load i32, ptr [[P]], align 4
201 ; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[PVAL]], 255
202 ; CHECK-NEXT:    br i1 [[CMP1]], label [[NEXT:%.*]], label [[EXIT:%.*]]
203 ; CHECK:       next:
204 ; CHECK-NEXT:    [[MIN:%.*]] = select i1 [[UNKNOWN]], i32 [[PVAL]], i32 5
205 ; CHECK-NEXT:    ret i1 false
206 ; CHECK:       exit:
207 ; CHECK-NEXT:    ret i1 true
209   %pval = load i32, ptr %p
210   %cmp1 = icmp slt i32 %pval, 255
211   br i1 %cmp1, label %next, label %exit
213 next:
214   %min = select i1 %unknown, i32 %pval, i32 5
215   %res = icmp eq i32 %min, 255
216   ret i1 %res
218 exit:
219   ret i1 true
222 ; Check that we take a conservative meet
223 define i1 @test2(ptr %p, i32 %qval, i1 %unknown) {
224 ; CHECK-LABEL: define i1 @test2
225 ; CHECK-SAME: (ptr [[P:%.*]], i32 [[QVAL:%.*]], i1 [[UNKNOWN:%.*]]) {
226 ; CHECK-NEXT:    [[PVAL:%.*]] = load i32, ptr [[P]], align 4
227 ; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[PVAL]], 255
228 ; CHECK-NEXT:    br i1 [[CMP1]], label [[NEXT:%.*]], label [[EXIT:%.*]]
229 ; CHECK:       next:
230 ; CHECK-NEXT:    [[MIN:%.*]] = select i1 [[UNKNOWN]], i32 [[PVAL]], i32 [[QVAL]]
231 ; CHECK-NEXT:    [[RES:%.*]] = icmp eq i32 [[MIN]], 255
232 ; CHECK-NEXT:    ret i1 [[RES]]
233 ; CHECK:       exit:
234 ; CHECK-NEXT:    ret i1 true
236   %pval = load i32, ptr %p
237   %cmp1 = icmp slt i32 %pval, 255
238   br i1 %cmp1, label %next, label %exit
240 next:
241   %min = select i1 %unknown, i32 %pval, i32 %qval
242   %res = icmp eq i32 %min, 255
243   ret i1 %res
245 exit:
246   ret i1 true
249 ; Same as @test2, but for the opposite select input
250 define i1 @test3(ptr %p, i32 %qval, i1 %unknown) {
251 ; CHECK-LABEL: define i1 @test3
252 ; CHECK-SAME: (ptr [[P:%.*]], i32 [[QVAL:%.*]], i1 [[UNKNOWN:%.*]]) {
253 ; CHECK-NEXT:    [[PVAL:%.*]] = load i32, ptr [[P]], align 4
254 ; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[PVAL]], 255
255 ; CHECK-NEXT:    br i1 [[CMP1]], label [[NEXT:%.*]], label [[EXIT:%.*]]
256 ; CHECK:       next:
257 ; CHECK-NEXT:    [[MIN:%.*]] = select i1 [[UNKNOWN]], i32 [[QVAL]], i32 [[PVAL]]
258 ; CHECK-NEXT:    [[RES:%.*]] = icmp eq i32 [[MIN]], 255
259 ; CHECK-NEXT:    ret i1 [[RES]]
260 ; CHECK:       exit:
261 ; CHECK-NEXT:    ret i1 true
263   %pval = load i32, ptr %p
264   %cmp1 = icmp slt i32 %pval, 255
265   br i1 %cmp1, label %next, label %exit
267 next:
268   %min = select i1 %unknown, i32 %qval, i32 %pval
269   %res = icmp eq i32 %min, 255
270   ret i1 %res
272 exit:
273   ret i1 true
276 ; Conflicting constants (i.e. isOverdefined result)
277 ; NOTE: Using doubles in this version is a bit of a hack.  This
278 ; is to get around the fact that all integers (including constants
279 ; and non-constants) are actually represented as constant-ranges.
280 define i1 @test4(ptr %p, i32 %qval, i1 %unknown) {
281 ; CHECK-LABEL: define i1 @test4
282 ; CHECK-SAME: (ptr [[P:%.*]], i32 [[QVAL:%.*]], i1 [[UNKNOWN:%.*]]) {
283 ; CHECK-NEXT:    [[PVAL:%.*]] = load i32, ptr [[P]], align 4
284 ; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[PVAL]], 255
285 ; CHECK-NEXT:    br i1 [[CMP1]], label [[NEXT:%.*]], label [[EXIT:%.*]]
286 ; CHECK:       next:
287 ; CHECK-NEXT:    [[MIN:%.*]] = select i1 [[UNKNOWN]], double 1.000000e+00, double 0.000000e+00
288 ; CHECK-NEXT:    [[RES:%.*]] = fcmp oeq double [[MIN]], 3.000000e+02
289 ; CHECK-NEXT:    ret i1 [[RES]]
290 ; CHECK:       exit:
291 ; CHECK-NEXT:    ret i1 true
293   %pval = load i32, ptr %p
294   %cmp1 = icmp slt i32 %pval, 255
295   br i1 %cmp1, label %next, label %exit
297 next:
298   %min = select i1 %unknown, double 1.0, double 0.0
299   %res = fcmp oeq double %min, 300.0
300   ret i1 %res
302 exit:
303   ret i1 true
306 ;; Using the condition to clamp the result
309 define i1 @test5(ptr %p, i1 %unknown) {
310 ; CHECK-LABEL: define i1 @test5
311 ; CHECK-SAME: (ptr [[P:%.*]], i1 [[UNKNOWN:%.*]]) {
312 ; CHECK-NEXT:    [[PVAL:%.*]] = load i32, ptr [[P]], align 4, !noundef [[META0:![0-9]+]]
313 ; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[PVAL]], 255
314 ; CHECK-NEXT:    br i1 [[CMP1]], label [[NEXT:%.*]], label [[EXIT:%.*]]
315 ; CHECK:       next:
316 ; CHECK-NEXT:    [[COND:%.*]] = icmp sgt i32 [[PVAL]], 0
317 ; CHECK-NEXT:    [[MIN:%.*]] = select i1 [[COND]], i32 [[PVAL]], i32 5
318 ; CHECK-NEXT:    ret i1 false
319 ; CHECK:       exit:
320 ; CHECK-NEXT:    ret i1 true
322   %pval = load i32, ptr %p, !noundef !0
323   %cmp1 = icmp slt i32 %pval, 255
324   br i1 %cmp1, label %next, label %exit
326 next:
327   %cond = icmp sgt i32 %pval, 0
328   %min = select i1 %cond, i32 %pval, i32 5
329   %res = icmp eq i32 %min, -1
330   ret i1 %res
332 exit:
333   ret i1 true
336 define i1 @test6(ptr %p, i1 %unknown) {
337 ; CHECK-LABEL: define i1 @test6
338 ; CHECK-SAME: (ptr [[P:%.*]], i1 [[UNKNOWN:%.*]]) {
339 ; CHECK-NEXT:    [[PVAL:%.*]] = load i32, ptr [[P]], align 4, !noundef [[META0]]
340 ; CHECK-NEXT:    [[CMP1:%.*]] = icmp ult i32 [[PVAL]], 255
341 ; CHECK-NEXT:    br i1 [[CMP1]], label [[NEXT:%.*]], label [[EXIT:%.*]]
342 ; CHECK:       next:
343 ; CHECK-NEXT:    [[COND:%.*]] = icmp ne i32 [[PVAL]], 254
344 ; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[COND]], i32 [[PVAL]], i32 1
345 ; CHECK-NEXT:    ret i1 true
346 ; CHECK:       exit:
347 ; CHECK-NEXT:    ret i1 true
349   %pval = load i32, ptr %p, !noundef !0
350   %cmp1 = icmp ult i32 %pval, 255
351   br i1 %cmp1, label %next, label %exit
353 next:
354   %cond = icmp ne i32 %pval, 254
355   %sel = select i1 %cond, i32 %pval, i32 1
356   %res = icmp slt i32 %sel, 254
357   ret i1 %res
359 exit:
360   ret i1 true
363 define i64 @select_cond_may_undef(i32 %a) {
364 ; CHECK-LABEL: define range(i64 -2147483648, 2147483648) i64 @select_cond_may_undef
365 ; CHECK-SAME: (i32 [[A:%.*]]) {
366 ; CHECK-NEXT:    [[IS_A_NONNEGATIVE:%.*]] = icmp sgt i32 [[A]], 1
367 ; CHECK-NEXT:    [[NARROW:%.*]] = select i1 [[IS_A_NONNEGATIVE]], i32 [[A]], i32 0
368 ; CHECK-NEXT:    [[MAX:%.*]] = sext i32 [[NARROW]] to i64
369 ; CHECK-NEXT:    ret i64 [[MAX]]
371   %is_a_nonnegative = icmp sgt i32 %a, 1
372   %narrow = select i1 %is_a_nonnegative, i32 %a, i32 0
373   %max = sext i32 %narrow to i64
374   ret i64 %max
377 define i32 @test_solve_select_at_use(i32 %a, i32 %b, i32 %c) {
378 ; CHECK-LABEL: define i32 @test_solve_select_at_use
379 ; CHECK-SAME: (i32 [[A:%.*]], i32 [[B:%.*]], i32 [[C:%.*]]) {
380 ; CHECK-NEXT:  entry:
381 ; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[A]], 0
382 ; CHECK-NEXT:    [[COND:%.*]] = icmp sgt i32 [[A]], -1
383 ; CHECK-NEXT:    br i1 [[COND]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
384 ; CHECK:       if.then:
385 ; CHECK-NEXT:    ret i32 [[C]]
386 ; CHECK:       if.else:
387 ; CHECK-NEXT:    ret i32 [[B]]
389 entry:
390   %cmp = icmp slt i32 %a, 0
391   %retval = select i1 %cmp, i32 %b, i32 %c
392   %cond = icmp sgt i32 %a, -1
393   br i1 %cond, label %if.then, label %if.else
394 if.then:
395   ret i32 %retval
396 if.else:
397   ret i32 %retval
400 !0 = !{}