[clang][modules] Don't prevent translation of FW_Private includes when explicitly...
[llvm-project.git] / llvm / test / Transforms / PhaseOrdering / loop-access-checks.ll
blobfcf1afce80ec081cb2e8ac6a27b44ad08631536e
1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
2 ; RUN: opt -passes='default<O2>' -S %s | FileCheck %s
4 ; Slightly reduced test case for a loop iterating over a std::span with libc++ hardening.
5 ; TODO: The runtime check in the loop should be removed.
7 ; #include <span>
8 ; #include <iostream>
10 ; void use(unsigned&);
12 ; void fill_with_foreach(std::span<unsigned> elems) {
13 ;  for (unsigned& x : elems)
14 ;    use(x);
15 ; }
17 %"class.std::__1::span" = type { ptr, i64 }
18 %"struct.std::__1::__bounded_iter" = type { ptr, ptr, ptr }
20 define void @test_fill_with_foreach([2 x i64] %elems.coerce) {
21 ; CHECK-LABEL: define void @test_fill_with_foreach
22 ; CHECK-SAME: ([2 x i64] [[ELEMS_COERCE:%.*]]) local_unnamed_addr {
23 ; CHECK-NEXT:  entry:
24 ; CHECK-NEXT:    [[ELEMS_COERCE_FCA_0_EXTRACT:%.*]] = extractvalue [2 x i64] [[ELEMS_COERCE]], 0
25 ; CHECK-NEXT:    [[TMP0:%.*]] = inttoptr i64 [[ELEMS_COERCE_FCA_0_EXTRACT]] to ptr
26 ; CHECK-NEXT:    [[ELEMS_COERCE_FCA_1_EXTRACT:%.*]] = extractvalue [2 x i64] [[ELEMS_COERCE]], 1
27 ; CHECK-NEXT:    [[ADD_PTR_I:%.*]] = getelementptr inbounds i32, ptr [[TMP0]], i64 [[ELEMS_COERCE_FCA_1_EXTRACT]]
28 ; CHECK-NEXT:    [[CMP_NOT_I_I_I_I:%.*]] = icmp slt i64 [[ELEMS_COERCE_FCA_1_EXTRACT]], 0
29 ; CHECK-NEXT:    br i1 [[CMP_NOT_I_I_I_I]], label [[ERROR:%.*]], label [[FOR_COND_PREHEADER_SPLIT:%.*]]
30 ; CHECK:       for.cond.preheader.split:
31 ; CHECK-NEXT:    [[CMP_I_NOT2:%.*]] = icmp eq i64 [[ELEMS_COERCE_FCA_1_EXTRACT]], 0
32 ; CHECK-NEXT:    br i1 [[CMP_I_NOT2]], label [[COMMON_RET:%.*]], label [[FOR_BODY:%.*]]
33 ; CHECK:       common.ret:
34 ; CHECK-NEXT:    ret void
35 ; CHECK:       error:
36 ; CHECK-NEXT:    tail call void @error()
37 ; CHECK-NEXT:    br label [[COMMON_RET]]
38 ; CHECK:       for.body:
39 ; CHECK-NEXT:    [[__BEGIN1_SROA_0_03:%.*]] = phi ptr [ [[INCDEC_PTR_I:%.*]], [[FOR_BODY]] ], [ [[TMP0]], [[FOR_COND_PREHEADER_SPLIT]] ]
40 ; CHECK-NEXT:    tail call void @use(ptr noundef nonnull align 4 dereferenceable(4) [[__BEGIN1_SROA_0_03]])
41 ; CHECK-NEXT:    [[INCDEC_PTR_I]] = getelementptr inbounds i32, ptr [[__BEGIN1_SROA_0_03]], i64 1
42 ; CHECK-NEXT:    [[CMP_I_NOT:%.*]] = icmp eq ptr [[INCDEC_PTR_I]], [[ADD_PTR_I]]
43 ; CHECK-NEXT:    br i1 [[CMP_I_NOT]], label [[COMMON_RET]], label [[FOR_BODY]]
45 entry:
46   %elems = alloca %"class.std::__1::span", align 8
47   %__begin1 = alloca %"struct.std::__1::__bounded_iter", align 8
48   %__end1 = alloca %"struct.std::__1::__bounded_iter", align 8
49   %elems.coerce.fca.0.extract = extractvalue [2 x i64] %elems.coerce, 0
50   store i64 %elems.coerce.fca.0.extract, ptr %elems, align 8
51   %elems.coerce.fca.1.extract = extractvalue [2 x i64] %elems.coerce, 1
52   %elems.coerce.fca.1.gep = getelementptr inbounds [2 x i64], ptr %elems, i64 0, i64 1
53   store i64 %elems.coerce.fca.1.extract, ptr %elems.coerce.fca.1.gep, align 8
54   call void @llvm.lifetime.start.p0(i64 24, ptr nonnull %__begin1) #6
55   %0 = load ptr, ptr %elems, align 8
56   %__size_.i.i = getelementptr inbounds %"class.std::__1::span", ptr %elems, i64 0, i32 1
57   %1 = load i64, ptr %__size_.i.i, align 8
58   %add.ptr.i = getelementptr inbounds i32, ptr %0, i64 %1
59   store ptr %0, ptr %__begin1, align 8
60   %__begin_.i.i.i.i = getelementptr inbounds %"struct.std::__1::__bounded_iter", ptr %__begin1, i64 0, i32 1
61   store ptr %0, ptr %__begin_.i.i.i.i, align 8
62   %__end_.i.i.i.i = getelementptr inbounds %"struct.std::__1::__bounded_iter", ptr %__begin1, i64 0, i32 2
63   store ptr %add.ptr.i, ptr %__end_.i.i.i.i, align 8
64   %cmp.not.i.i.i.i = icmp slt i64 %1, 0
65   br i1 %cmp.not.i.i.i.i, label %error, label %check.2
67 check.2:
68   call void @llvm.lifetime.start.p0(i64 24, ptr nonnull %__end1) #6
69   %l4 = load ptr, ptr %elems, align 8
70   %__size_.i.i4 = getelementptr inbounds %"class.std::__1::span", ptr %elems, i64 0, i32 1
71   %l5 = load i64, ptr %__size_.i.i4, align 8
72   %add.ptr.i5 = getelementptr inbounds i32, ptr %l4, i64 %l5
73   store ptr %add.ptr.i5, ptr %__end1, align 8
74   %__begin_.i.i.i.i6 = getelementptr inbounds %"struct.std::__1::__bounded_iter", ptr %__end1, i64 0, i32 1
75   store ptr %l4, ptr %__begin_.i.i.i.i6, align 8
76   %__end_.i.i.i.i7 = getelementptr inbounds %"struct.std::__1::__bounded_iter", ptr %__end1, i64 0, i32 2
77   store ptr %add.ptr.i5, ptr %__end_.i.i.i.i7, align 8
78   %cmp.not.i.i.i.i8 = icmp slt i64 %l5, 0
79   br i1 %cmp.not.i.i.i.i8, label %error, label %for.cond
81 error:
82   call void @error()
83   ret void
85 for.cond:
86   %l8 = load ptr, ptr %__begin1, align 8
87   %l9 = load ptr, ptr %__end1, align 8
88   %cmp.i = icmp ne ptr %l8, %l9
89   br i1 %cmp.i, label %for.body, label %for.cond.cleanup
91 for.cond.cleanup:                                 ; preds = %for.cond
92   call void @llvm.lifetime.end.p0(i64 24, ptr nonnull %__end1)
93   call void @llvm.lifetime.end.p0(i64 24, ptr nonnull %__begin1)
94   ret void
96 for.body:                                         ; preds = %for.cond
97   %l10 = load ptr, ptr %__begin1, align 8
98   %__begin_.i.i = getelementptr inbounds %"struct.std::__1::__bounded_iter", ptr %__begin1, i64 0, i32 1
99   %l11 = load ptr, ptr %__begin_.i.i, align 8
100   %cmp.not.i.i = icmp uge ptr %l10, %l11
101   %__end_.i.i = getelementptr inbounds %"struct.std::__1::__bounded_iter", ptr %__begin1, i64 0, i32 2
102   %l12 = load ptr, ptr %__end_.i.i, align 8
103   %cmp2.i.i = icmp ult ptr %l10, %l12
104   %sel = select i1 %cmp.not.i.i, i1 %cmp2.i.i, i1 false
105   br i1 %sel, label %for.latch, label %error
107 for.latch:
108   call void @use(ptr noundef nonnull align 4 dereferenceable(4) %l10)
109   %l = load ptr, ptr %__begin1, align 8
110   %incdec.ptr.i = getelementptr inbounds i32, ptr %l, i64 1
111   store ptr %incdec.ptr.i, ptr %__begin1, align 8
112   br label %for.cond
115 declare void @error()
117 declare void @llvm.lifetime.start.p0(i64 immarg, ptr nocapture)
119 declare void @use(ptr noundef nonnull align 4 dereferenceable(4))
121 declare void @llvm.lifetime.end.p0(i64 immarg, ptr nocapture)
124 ; -------------------------------------------------------------------------
125 ; Test case for runtime check removal when accessing vector elements with
126 ; hardened glibc++ (PR63125)
128 %Vector_base = type { %Vector_impl }
129 %Vector_impl = type { %Vector_impl_data }
130 %Vector_impl_data = type { ptr, ptr, ptr }
132 define void @foo(ptr noundef nonnull align 8 dereferenceable(24) noalias %vec) #0 {
133 ; CHECK-LABEL: define void @foo
134 ; CHECK-SAME: (ptr noalias nocapture noundef nonnull readonly align 8 dereferenceable(24) [[VEC:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
135 ; CHECK-NEXT:  entry:
136 ; CHECK-NEXT:    [[_M_FINISH_I_I:%.*]] = getelementptr inbounds [[VECTOR_IMPL_DATA:%.*]], ptr [[VEC]], i64 0, i32 1
137 ; CHECK-NEXT:    [[TMP0:%.*]] = load ptr, ptr [[_M_FINISH_I_I]], align 8
138 ; CHECK-NEXT:    [[TMP1:%.*]] = load ptr, ptr [[VEC]], align 8
139 ; CHECK-NEXT:    [[SUB_PTR_LHS_CAST_I_I:%.*]] = ptrtoint ptr [[TMP0]] to i64
140 ; CHECK-NEXT:    [[SUB_PTR_RHS_CAST_I_I:%.*]] = ptrtoint ptr [[TMP1]] to i64
141 ; CHECK-NEXT:    [[SUB_PTR_SUB_I_I:%.*]] = sub i64 [[SUB_PTR_LHS_CAST_I_I]], [[SUB_PTR_RHS_CAST_I_I]]
142 ; CHECK-NEXT:    [[SUB_PTR_DIV_I_I:%.*]] = ashr exact i64 [[SUB_PTR_SUB_I_I]], 3
143 ; CHECK-NEXT:    [[CMP_NOT9:%.*]] = icmp eq ptr [[TMP0]], [[TMP1]]
144 ; CHECK-NEXT:    br i1 [[CMP_NOT9]], label [[FOR_COND_CLEANUP:%.*]], label [[FOR_BODY:%.*]]
145 ; CHECK:       for.cond.cleanup:
146 ; CHECK-NEXT:    ret void
147 ; CHECK:       for.body:
148 ; CHECK-NEXT:    [[I_010:%.*]] = phi i64 [ [[INC:%.*]], [[FOR_BODY]] ], [ 0, [[ENTRY:%.*]] ]
149 ; CHECK-NEXT:    [[ADD_PTR_I:%.*]] = getelementptr inbounds double, ptr [[TMP1]], i64 [[I_010]]
150 ; CHECK-NEXT:    [[TMP2:%.*]] = load double, ptr [[ADD_PTR_I]], align 8
151 ; CHECK-NEXT:    [[ADD:%.*]] = fadd double [[TMP2]], 1.000000e+00
152 ; CHECK-NEXT:    store double [[ADD]], ptr [[ADD_PTR_I]], align 8
153 ; CHECK-NEXT:    [[INC]] = add nuw i64 [[I_010]], 1
154 ; CHECK-NEXT:    [[CMP_NOT:%.*]] = icmp eq i64 [[INC]], [[SUB_PTR_DIV_I_I]]
155 ; CHECK-NEXT:    br i1 [[CMP_NOT]], label [[FOR_COND_CLEANUP]], label [[FOR_BODY]]
157 entry:
158   %vec.addr = alloca ptr, align 8
159   %count = alloca i64, align 8
160   %i = alloca i64, align 8
161   store ptr %vec, ptr %vec.addr, align 8
162   call void @llvm.lifetime.start.p0(i64 8, ptr %count)
163   %0 = load ptr, ptr %vec.addr, align 8
164   %call = call noundef i64 @alloc(ptr noundef nonnull align 8 dereferenceable(24) %0)
165   store i64 %call, ptr %count, align 8
166   call void @llvm.lifetime.start.p0(i64 8, ptr %i)
167   store i64 0, ptr %i, align 8
168   br label %for.cond
170 for.cond:
171   %1 = load i64, ptr %i, align 8
172   %2 = load i64, ptr %count, align 8
173   %cmp = icmp ne i64 %1, %2
174   br i1 %cmp, label %for.body, label %for.cond.cleanup
176 for.cond.cleanup:
177   call void @llvm.lifetime.end.p0(i64 8, ptr %i)
178   br label %for.end
180 for.body:
181   %3 = load ptr, ptr %vec.addr, align 8
182   %4 = load i64, ptr %i, align 8
183   %call1 = call noundef nonnull align 8 dereferenceable(8) ptr @operator_acc(ptr noundef nonnull align 8 dereferenceable(24) %3, i64 noundef %4)
184   %5 = load double, ptr %call1, align 8
185   %add = fadd double %5, 1.000000e+00
186   store double %add, ptr %call1, align 8
187   br label %for.inc
189 for.inc:
190   %6 = load i64, ptr %i, align 8
191   %inc = add i64 %6, 1
192   store i64 %inc, ptr %i, align 8
193   br label %for.cond
195 for.end:
196   call void @llvm.lifetime.end.p0(i64 8, ptr %count) #5
197   ret void
200 define internal noundef i64 @alloc(ptr noundef nonnull align 8 dereferenceable(24) %__cont) {
201 entry:
202   %__cont.addr = alloca ptr, align 8
203   store ptr %__cont, ptr %__cont.addr, align 8
204   %0 = load ptr, ptr %__cont.addr, align 8
205   %call = call noundef i64 @size(ptr noundef nonnull align 8 dereferenceable(24) %0) #5
206   ret i64 %call
209 define internal noundef nonnull align 8 dereferenceable(8) ptr @operator_acc(ptr noundef nonnull align 8 dereferenceable(24) %this, i64 noundef %__n) {
210 entry:
211   %this.addr = alloca ptr, align 8
212   %__n.addr = alloca i64, align 8
213   store ptr %this, ptr %this.addr, align 8
214   store i64 %__n, ptr %__n.addr, align 8
215   %this1 = load ptr, ptr %this.addr, align 8
216   br label %do.body
218 do.body:
219   %0 = load i64, ptr %__n.addr, align 8
220   %call = call noundef i64 @size(ptr noundef nonnull align 8 dereferenceable(24) %this1)
221   %cmp = icmp ult i64 %0, %call
222   %lnot = xor i1 %cmp, true
223   %conv = zext i1 %lnot to i64
224   %tobool = icmp ne i64 %conv, 0
225   br i1 %tobool, label %if.then, label %if.end
227 if.then:
228   call void @abort() readnone
229   unreachable
231 if.end:
232   br label %do.cond
234 do.cond:
235   br label %do.end
237 do.end:
238   %_M_impl = getelementptr inbounds %Vector_base, ptr %this1, i32 0, i32 0
239   %_M_start = getelementptr inbounds %Vector_impl_data, ptr %_M_impl, i32 0, i32 0
240   %1 = load ptr, ptr %_M_start, align 8
241   %2 = load i64, ptr %__n.addr, align 8
242   %add.ptr = getelementptr inbounds double, ptr %1, i64 %2
243   ret ptr %add.ptr
246 define internal noundef i64 @size(ptr noundef nonnull align 8 dereferenceable(24) %this) {
247 entry:
248   %this.addr = alloca ptr, align 8
249   store ptr %this, ptr %this.addr, align 8
250   %this1 = load ptr, ptr %this.addr, align 8
251   %_M_impl = getelementptr inbounds %Vector_base, ptr %this1, i32 0, i32 0
252   %_M_finish = getelementptr inbounds %Vector_impl_data, ptr %_M_impl, i32 0, i32 1
253   %0 = load ptr, ptr %_M_finish, align 8
254   %_M_impl2 = getelementptr inbounds %Vector_base, ptr %this1, i32 0, i32 0
255   %_M_start = getelementptr inbounds %Vector_impl_data, ptr %_M_impl2, i32 0, i32 0
256   %1 = load ptr, ptr %_M_start, align 8
257   %sub.ptr.lhs.cast = ptrtoint ptr %0 to i64
258   %sub.ptr.rhs.cast = ptrtoint ptr %1 to i64
259   %sub.ptr.sub = sub i64 %sub.ptr.lhs.cast, %sub.ptr.rhs.cast
260   %sub.ptr.div = sdiv exact i64 %sub.ptr.sub, 8
261   ret i64 %sub.ptr.div
264 declare void @abort()
266 ; -------------------------------------------------------------------------
267 ; Test case for runtime check removal when accessing elements in a nested loop
268 ; (PR64881)
271 define void @monkey(ptr noundef %arr, i32 noundef %len) {
272 ; CHECK-LABEL: define void @monkey
273 ; CHECK-SAME: (ptr nocapture noundef [[ARR:%.*]], i32 noundef [[LEN:%.*]]) local_unnamed_addr #[[ATTR1:[0-9]+]] {
274 ; CHECK-NEXT:  entry:
275 ; CHECK-NEXT:    [[CMP8:%.*]] = icmp ugt i32 [[LEN]], 1
276 ; CHECK-NEXT:    br i1 [[CMP8]], label [[FOR_BODY4_PREHEADER:%.*]], label [[FOR_COND_CLEANUP:%.*]]
277 ; CHECK:       for.body4.preheader:
278 ; CHECK-NEXT:    [[I_09:%.*]] = phi i32 [ [[INC:%.*]], [[FOR_COND_CLEANUP3:%.*]] ], [ 1, [[ENTRY:%.*]] ]
279 ; CHECK-NEXT:    br label [[FOR_BODY4:%.*]]
280 ; CHECK:       for.cond.cleanup:
281 ; CHECK-NEXT:    ret void
282 ; CHECK:       for.cond.cleanup3:
283 ; CHECK-NEXT:    [[INC]] = add nuw i32 [[I_09]], 1
284 ; CHECK-NEXT:    [[CMP:%.*]] = icmp ult i32 [[INC]], [[LEN]]
285 ; CHECK-NEXT:    br i1 [[CMP]], label [[FOR_BODY4_PREHEADER]], label [[FOR_COND_CLEANUP]]
286 ; CHECK:       for.body4:
287 ; CHECK-NEXT:    [[K_07:%.*]] = phi i32 [ [[DEC:%.*]], [[FOR_BODY4]] ], [ [[I_09]], [[FOR_BODY4_PREHEADER]] ]
288 ; CHECK-NEXT:    [[IDX_EXT_I:%.*]] = zext i32 [[K_07]] to i64
289 ; CHECK-NEXT:    [[ADD_PTR_I:%.*]] = getelementptr inbounds i32, ptr [[ARR]], i64 [[IDX_EXT_I]]
290 ; CHECK-NEXT:    [[TMP0:%.*]] = load i32, ptr [[ADD_PTR_I]], align 4
291 ; CHECK-NEXT:    [[ADD:%.*]] = add nsw i32 [[TMP0]], 1
292 ; CHECK-NEXT:    store i32 [[ADD]], ptr [[ADD_PTR_I]], align 4
293 ; CHECK-NEXT:    [[DEC]] = add i32 [[K_07]], -1
294 ; CHECK-NEXT:    [[CMP2_NOT:%.*]] = icmp eq i32 [[DEC]], 0
295 ; CHECK-NEXT:    br i1 [[CMP2_NOT]], label [[FOR_COND_CLEANUP3]], label [[FOR_BODY4]]
297 entry:
298   %arr.addr = alloca ptr, align 8
299   %len.addr = alloca i32, align 4
300   %i = alloca i32, align 4
301   %cleanup.dest.slot = alloca i32, align 4
302   %k = alloca i32, align 4
303   store ptr %arr, ptr %arr.addr, align 8
304   store i32 %len, ptr %len.addr, align 4
305   call void @llvm.lifetime.start.p0(i64 4, ptr %i) #3
306   store i32 1, ptr %i, align 4
307   br label %for.cond
309 for.cond:                                         ; preds = %for.inc5, %entry
310   %0 = load i32, ptr %i, align 4
311   %1 = load i32, ptr %len.addr, align 4
312   %cmp = icmp ult i32 %0, %1
313   br i1 %cmp, label %for.body, label %for.cond.cleanup
315 for.cond.cleanup:                                 ; preds = %for.cond
316   store i32 2, ptr %cleanup.dest.slot, align 4
317   call void @llvm.lifetime.end.p0(i64 4, ptr %i) #3
318   br label %for.end6
320 for.body:                                         ; preds = %for.cond
321   call void @llvm.lifetime.start.p0(i64 4, ptr %k) #3
322   %2 = load i32, ptr %i, align 4
323   store i32 %2, ptr %k, align 4
324   br label %for.cond1
326 for.cond1:                                        ; preds = %for.inc, %for.body
327   %3 = load i32, ptr %k, align 4
328   %cmp2 = icmp ugt i32 %3, 0
329   br i1 %cmp2, label %for.body4, label %for.cond.cleanup3
331 for.cond.cleanup3:                                ; preds = %for.cond1
332   store i32 5, ptr %cleanup.dest.slot, align 4
333   call void @llvm.lifetime.end.p0(i64 4, ptr %k) #3
334   br label %for.end
336 for.body4:                                        ; preds = %for.cond1
337   %4 = load ptr, ptr %arr.addr, align 8
338   %5 = load i32, ptr %len.addr, align 4
339   %6 = load i32, ptr %k, align 4
340   %call = call noundef ptr @at(ptr noundef %4, i32 noundef %5, i32 noundef %6)
341   %7 = load i32, ptr %call, align 4
342   %add = add nsw i32 %7, 1
343   store i32 %add, ptr %call, align 4
344   br label %for.inc
346 for.inc:                                          ; preds = %for.body4
347   %8 = load i32, ptr %k, align 4
348   %dec = add i32 %8, -1
349   store i32 %dec, ptr %k, align 4
350   br label %for.cond1
352 for.end:                                          ; preds = %for.cond.cleanup3
353   br label %for.inc5
355 for.inc5:                                         ; preds = %for.end
356   %9 = load i32, ptr %i, align 4
357   %inc = add i32 %9, 1
358   store i32 %inc, ptr %i, align 4
359   br label %for.cond
361 for.end6:                                         ; preds = %for.cond.cleanup
362   ret void
365 define internal noundef ptr @at(ptr noundef %arr, i32 noundef %len, i32 noundef %idx) {
366 entry:
367   %arr.addr = alloca ptr, align 8
368   %len.addr = alloca i32, align 4
369   %idx.addr = alloca i32, align 4
370   store ptr %arr, ptr %arr.addr, align 8
371   store i32 %len, ptr %len.addr, align 4
372   store i32 %idx, ptr %idx.addr, align 4
373   %0 = load i32, ptr %idx.addr, align 4
374   %1 = load i32, ptr %len.addr, align 4
375   %cmp = icmp uge i32 %0, %1
376   br i1 %cmp, label %if.then, label %if.end
378 if.then:                                          ; preds = %entry
379   call void @abort()
380   unreachable
382 if.end:                                           ; preds = %entry
383   %2 = load ptr, ptr %arr.addr, align 8
384   %3 = load i32, ptr %idx.addr, align 4
385   %idx.ext = zext i32 %3 to i64
386   %add.ptr = getelementptr inbounds i32, ptr %2, i64 %idx.ext
387   ret ptr %add.ptr