Skip to content

Commit cc51c1e

Browse files
fhahngithub-actions[bot]
fhahn
authored and
github-actions[bot]
committed
Automerge: [IndVars] Add additional tests with ICMP_NE loop guards.
Extra test coverage for llvm/llvm-project#163787.
2 parents d860799 + 0590c9e commit cc51c1e

File tree

1 file changed

+166
-0
lines changed

1 file changed

+166
-0
lines changed

‎llvm/test/Transforms/IndVarSimplify/pointer-loop-guards.ll‎

Lines changed: 166 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 6
22
; RUN: opt -p indvars -S %s | FileCheck %s
3+
; RUN: opt -p indvars -data-layout='n32:64' -S %s | FileCheck --check-prefix=N32 %s
34

45
declare i1 @cond()
56

@@ -28,6 +29,32 @@ define i64 @test_ptr_compare_guard(ptr %start, ptr %end) {
2829
; CHECK-NEXT: [[RES:%.*]] = phi i64 [ 0, %[[ENTRY]] ], [ [[RES_PH]], %[[EXIT_LOOPEXIT]] ]
2930
; CHECK-NEXT: ret i64 [[RES]]
3031
;
32+
; N32-LABEL: define i64 @test_ptr_compare_guard(
33+
; N32-SAME: ptr [[START:%.*]], ptr [[END:%.*]]) {
34+
; N32-NEXT: [[ENTRY:.*]]:
35+
; N32-NEXT: [[START2:%.*]] = ptrtoint ptr [[START]] to i64
36+
; N32-NEXT: [[END1:%.*]] = ptrtoint ptr [[END]] to i64
37+
; N32-NEXT: [[C_0:%.*]] = icmp eq ptr [[START]], [[END]]
38+
; N32-NEXT: br i1 [[C_0]], label %[[EXIT:.*]], label %[[LOOP_HEADER_PREHEADER:.*]]
39+
; N32: [[LOOP_HEADER_PREHEADER]]:
40+
; N32-NEXT: [[TMP0:%.*]] = add i64 [[END1]], -1
41+
; N32-NEXT: [[TMP1:%.*]] = sub i64 [[TMP0]], [[START2]]
42+
; N32-NEXT: br label %[[LOOP_HEADER:.*]]
43+
; N32: [[LOOP_HEADER]]:
44+
; N32-NEXT: [[PTR_IV:%.*]] = phi ptr [ [[PTR_IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ], [ [[START]], %[[LOOP_HEADER_PREHEADER]] ]
45+
; N32-NEXT: [[C_1:%.*]] = call i1 @cond()
46+
; N32-NEXT: br i1 [[C_1]], label %[[LOOP_LATCH]], label %[[EXIT_LOOPEXIT:.*]]
47+
; N32: [[LOOP_LATCH]]:
48+
; N32-NEXT: [[PTR_IV_NEXT]] = getelementptr i8, ptr [[PTR_IV]], i64 1
49+
; N32-NEXT: [[C_2:%.*]] = icmp eq ptr [[PTR_IV_NEXT]], [[END]]
50+
; N32-NEXT: br i1 [[C_2]], label %[[EXIT_LOOPEXIT]], label %[[LOOP_HEADER]]
51+
; N32: [[EXIT_LOOPEXIT]]:
52+
; N32-NEXT: [[RES_PH:%.*]] = phi i64 [ 0, %[[LOOP_HEADER]] ], [ [[TMP1]], %[[LOOP_LATCH]] ]
53+
; N32-NEXT: br label %[[EXIT]]
54+
; N32: [[EXIT]]:
55+
; N32-NEXT: [[RES:%.*]] = phi i64 [ 0, %[[ENTRY]] ], [ [[RES_PH]], %[[EXIT_LOOPEXIT]] ]
56+
; N32-NEXT: ret i64 [[RES]]
57+
;
3158
entry:
3259
%c.0 = icmp eq ptr %start, %end
3360
br i1 %c.0, label %exit, label %loop.header
@@ -48,3 +75,142 @@ exit:
4875
%res = phi i64 [ 0, %entry ], [ %i64.iv, %loop.latch ], [ 0, %loop.header ]
4976
ret i64 %res
5077
}
78+
79+
define void @test_sub_cmp(ptr align 8 %start, ptr %end) {
80+
; CHECK-LABEL: define void @test_sub_cmp(
81+
; CHECK-SAME: ptr align 8 [[START:%.*]], ptr [[END:%.*]]) {
82+
; CHECK-NEXT: [[ENTRY:.*:]]
83+
; CHECK-NEXT: [[START_INT:%.*]] = ptrtoint ptr [[START]] to i64
84+
; CHECK-NEXT: [[END_INT:%.*]] = ptrtoint ptr [[END]] to i64
85+
; CHECK-NEXT: [[PTR_DIFF:%.*]] = sub i64 [[START_INT]], [[END_INT]]
86+
; CHECK-NEXT: [[CMP_ENTRY:%.*]] = icmp eq ptr [[START]], [[END]]
87+
; CHECK-NEXT: br i1 [[CMP_ENTRY]], label %[[EXIT:.*]], label %[[LOOP_HEADER_PREHEADER:.*]]
88+
; CHECK: [[LOOP_HEADER_PREHEADER]]:
89+
; CHECK-NEXT: br label %[[LOOP_HEADER:.*]]
90+
; CHECK: [[LOOP_HEADER]]:
91+
; CHECK-NEXT: [[IV:%.*]] = phi i64 [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ], [ 0, %[[LOOP_HEADER_PREHEADER]] ]
92+
; CHECK-NEXT: [[C_1:%.*]] = call i1 @cond()
93+
; CHECK-NEXT: br i1 [[C_1]], label %[[EXIT_EARLY:.*]], label %[[LOOP_LATCH]]
94+
; CHECK: [[LOOP_LATCH]]:
95+
; CHECK-NEXT: [[IV_NEXT]] = add nuw i64 [[IV]], 1
96+
; CHECK-NEXT: [[CMP_LATCH:%.*]] = icmp ult i64 [[IV_NEXT]], [[PTR_DIFF]]
97+
; CHECK-NEXT: br i1 [[CMP_LATCH]], label %[[LOOP_HEADER]], label %[[EXIT_LOOPEXIT:.*]]
98+
; CHECK: [[EXIT_EARLY]]:
99+
; CHECK-NEXT: br label %[[EXIT]]
100+
; CHECK: [[EXIT_LOOPEXIT]]:
101+
; CHECK-NEXT: br label %[[EXIT]]
102+
; CHECK: [[EXIT]]:
103+
; CHECK-NEXT: ret void
104+
;
105+
; N32-LABEL: define void @test_sub_cmp(
106+
; N32-SAME: ptr align 8 [[START:%.*]], ptr [[END:%.*]]) {
107+
; N32-NEXT: [[ENTRY:.*:]]
108+
; N32-NEXT: [[START_INT:%.*]] = ptrtoint ptr [[START]] to i64
109+
; N32-NEXT: [[END_INT:%.*]] = ptrtoint ptr [[END]] to i64
110+
; N32-NEXT: [[PTR_DIFF:%.*]] = sub i64 [[START_INT]], [[END_INT]]
111+
; N32-NEXT: [[CMP_ENTRY:%.*]] = icmp eq ptr [[START]], [[END]]
112+
; N32-NEXT: br i1 [[CMP_ENTRY]], label %[[EXIT:.*]], label %[[LOOP_HEADER_PREHEADER:.*]]
113+
; N32: [[LOOP_HEADER_PREHEADER]]:
114+
; N32-NEXT: [[UMAX:%.*]] = call i64 @llvm.umax.i64(i64 [[PTR_DIFF]], i64 1)
115+
; N32-NEXT: br label %[[LOOP_HEADER:.*]]
116+
; N32: [[LOOP_HEADER]]:
117+
; N32-NEXT: [[IV:%.*]] = phi i64 [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ], [ 0, %[[LOOP_HEADER_PREHEADER]] ]
118+
; N32-NEXT: [[C_1:%.*]] = call i1 @cond()
119+
; N32-NEXT: br i1 [[C_1]], label %[[EXIT_EARLY:.*]], label %[[LOOP_LATCH]]
120+
; N32: [[LOOP_LATCH]]:
121+
; N32-NEXT: [[IV_NEXT]] = add nuw i64 [[IV]], 1
122+
; N32-NEXT: [[EXITCOND:%.*]] = icmp ne i64 [[IV_NEXT]], [[UMAX]]
123+
; N32-NEXT: br i1 [[EXITCOND]], label %[[LOOP_HEADER]], label %[[EXIT_LOOPEXIT:.*]]
124+
; N32: [[EXIT_EARLY]]:
125+
; N32-NEXT: br label %[[EXIT]]
126+
; N32: [[EXIT_LOOPEXIT]]:
127+
; N32-NEXT: br label %[[EXIT]]
128+
; N32: [[EXIT]]:
129+
; N32-NEXT: ret void
130+
;
131+
entry:
132+
%start.int = ptrtoint ptr %start to i64
133+
%end.int = ptrtoint ptr %end to i64
134+
%ptr.diff = sub i64 %start.int, %end.int
135+
%cmp.entry = icmp eq ptr %start, %end
136+
br i1 %cmp.entry, label %exit, label %loop.header
137+
138+
loop.header:
139+
%iv = phi i64 [ 0, %entry ], [ %iv.next, %loop.latch ]
140+
%c.1 = call i1 @cond()
141+
br i1 %c.1, label %exit.early, label %loop.latch
142+
143+
loop.latch:
144+
%iv.next = add i64 %iv, 1
145+
%cmp.latch = icmp ult i64 %iv.next, %ptr.diff
146+
br i1 %cmp.latch, label %loop.header, label %exit
147+
148+
exit.early:
149+
br label %exit
150+
151+
exit:
152+
ret void
153+
}
154+
155+
156+
define void @test_ptr_diff_with_assume(ptr align 8 %start, ptr align 8 %end, ptr %P) {
157+
; CHECK-LABEL: define void @test_ptr_diff_with_assume(
158+
; CHECK-SAME: ptr align 8 [[START:%.*]], ptr align 8 [[END:%.*]], ptr [[P:%.*]]) {
159+
; CHECK-NEXT: [[ENTRY:.*:]]
160+
; CHECK-NEXT: [[START_INT:%.*]] = ptrtoint ptr [[START]] to i64
161+
; CHECK-NEXT: [[END_INT:%.*]] = ptrtoint ptr [[END]] to i64
162+
; CHECK-NEXT: [[PTR_DIFF:%.*]] = sub i64 [[START_INT]], [[END_INT]]
163+
; CHECK-NEXT: [[DIFF_CMP:%.*]] = icmp ult i64 [[PTR_DIFF]], 2
164+
; CHECK-NEXT: call void @llvm.assume(i1 [[DIFF_CMP]])
165+
; CHECK-NEXT: [[ENTRY_CMP:%.*]] = icmp eq ptr [[START]], [[END]]
166+
; CHECK-NEXT: br i1 [[ENTRY_CMP]], label %[[EXIT:.*]], label %[[LOOP_BODY_PREHEADER:.*]]
167+
; CHECK: [[LOOP_BODY_PREHEADER]]:
168+
; CHECK-NEXT: br label %[[LOOP_BODY:.*]]
169+
; CHECK: [[LOOP_BODY]]:
170+
; CHECK-NEXT: [[TMP0:%.*]] = call i1 @cond()
171+
; CHECK-NEXT: br i1 true, label %[[EXIT_LOOPEXIT:.*]], label %[[LOOP_BODY]]
172+
; CHECK: [[EXIT_LOOPEXIT]]:
173+
; CHECK-NEXT: br label %[[EXIT]]
174+
; CHECK: [[EXIT]]:
175+
; CHECK-NEXT: ret void
176+
;
177+
; N32-LABEL: define void @test_ptr_diff_with_assume(
178+
; N32-SAME: ptr align 8 [[START:%.*]], ptr align 8 [[END:%.*]], ptr [[P:%.*]]) {
179+
; N32-NEXT: [[ENTRY:.*:]]
180+
; N32-NEXT: [[START_INT:%.*]] = ptrtoint ptr [[START]] to i64
181+
; N32-NEXT: [[END_INT:%.*]] = ptrtoint ptr [[END]] to i64
182+
; N32-NEXT: [[PTR_DIFF:%.*]] = sub i64 [[START_INT]], [[END_INT]]
183+
; N32-NEXT: [[DIFF_CMP:%.*]] = icmp ult i64 [[PTR_DIFF]], 2
184+
; N32-NEXT: call void @llvm.assume(i1 [[DIFF_CMP]])
185+
; N32-NEXT: [[ENTRY_CMP:%.*]] = icmp eq ptr [[START]], [[END]]
186+
; N32-NEXT: br i1 [[ENTRY_CMP]], label %[[EXIT:.*]], label %[[LOOP_BODY_PREHEADER:.*]]
187+
; N32: [[LOOP_BODY_PREHEADER]]:
188+
; N32-NEXT: br label %[[LOOP_BODY:.*]]
189+
; N32: [[LOOP_BODY]]:
190+
; N32-NEXT: [[TMP0:%.*]] = call i1 @cond()
191+
; N32-NEXT: br i1 true, label %[[EXIT_LOOPEXIT:.*]], label %[[LOOP_BODY]]
192+
; N32: [[EXIT_LOOPEXIT]]:
193+
; N32-NEXT: br label %[[EXIT]]
194+
; N32: [[EXIT]]:
195+
; N32-NEXT: ret void
196+
;
197+
entry:
198+
%start.int = ptrtoint ptr %start to i64
199+
%end.int = ptrtoint ptr %end to i64
200+
%ptr.diff = sub i64 %start.int, %end.int
201+
%diff.cmp = icmp ult i64 %ptr.diff, 2
202+
call void @llvm.assume(i1 %diff.cmp)
203+
%computed.end = getelementptr i8, ptr %start, i64 %ptr.diff
204+
%entry.cmp = icmp eq ptr %start, %end
205+
br i1 %entry.cmp, label %exit, label %loop.body
206+
207+
loop.body:
208+
%iv = phi ptr [ %start, %entry ], [ %iv.next, %loop.body ]
209+
call i1 @cond()
210+
%iv.next = getelementptr i8, ptr %iv, i64 1
211+
%loop.cmp = icmp eq ptr %iv.next, %computed.end
212+
br i1 %loop.cmp, label %exit, label %loop.body
213+
214+
exit:
215+
ret void
216+
}

0 commit comments

Comments
 (0)