xref: /llvm-project/llvm/test/Transforms/LoopLoadElim/early-exit.ll (revision e949b54a5b7cd7cd0690fa126be3363a21f05a8e)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2; RUN: opt -passes=loop-load-elim -S %s | FileCheck %s
3
4target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
5
6define void @f(ptr %A, ptr %B, ptr %C, i64 %N) {
7; CHECK-LABEL: define void @f(
8; CHECK-SAME: ptr [[A:%.*]], ptr [[B:%.*]], ptr [[C:%.*]], i64 [[N:%.*]]) {
9; CHECK-NEXT:  [[FOR_BODY_LVER_CHECK:.*]]:
10; CHECK-NEXT:    br label %[[FOR_BODY:.*]]
11; CHECK:       [[FOR_BODY]]:
12; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ 0, %[[FOR_BODY_LVER_CHECK]] ], [ [[INDVARS_IV_NEXT:%.*]], %[[LATCH:.*]] ]
13; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add nuw nsw i64 [[INDVARS_IV]], 1
14; CHECK-NEXT:    [[AIDX_NEXT:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[INDVARS_IV_NEXT]]
15; CHECK-NEXT:    [[BIDX:%.*]] = getelementptr inbounds i32, ptr [[B]], i64 [[INDVARS_IV]]
16; CHECK-NEXT:    [[CIDX:%.*]] = getelementptr inbounds i32, ptr [[C]], i64 [[INDVARS_IV]]
17; CHECK-NEXT:    [[AIDX:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[INDVARS_IV]]
18; CHECK-NEXT:    [[B:%.*]] = load i32, ptr [[BIDX]], align 4
19; CHECK-NEXT:    [[UNCOUNTABLE_C:%.*]] = icmp eq i32 [[B]], 10
20; CHECK-NEXT:    br i1 [[UNCOUNTABLE_C]], label %[[LATCH]], label %[[FOR_END:.*]]
21; CHECK:       [[LATCH]]:
22; CHECK-NEXT:    [[A_P1:%.*]] = add i32 [[B]], 2
23; CHECK-NEXT:    store i32 [[A_P1]], ptr [[AIDX_NEXT]], align 4
24; CHECK-NEXT:    [[A:%.*]] = load i32, ptr [[AIDX]], align 1
25; CHECK-NEXT:    [[C:%.*]] = mul i32 [[A]], 2
26; CHECK-NEXT:    store i32 [[C]], ptr [[CIDX]], align 4
27; CHECK-NEXT:    [[EXITCOND:%.*]] = icmp eq i64 [[INDVARS_IV_NEXT]], [[N]]
28; CHECK-NEXT:    br i1 [[EXITCOND]], label %[[FOR_END]], label %[[FOR_BODY]]
29; CHECK:       [[FOR_END]]:
30; CHECK-NEXT:    ret void
31;
32entry:
33  br label %for.body
34
35for.body:                                         ; preds = %for.body, %entry
36  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %latch ]
37  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
38
39  %Aidx_next = getelementptr inbounds i32, ptr %A, i64 %indvars.iv.next
40  %Bidx = getelementptr inbounds i32, ptr %B, i64 %indvars.iv
41  %Cidx = getelementptr inbounds i32, ptr %C, i64 %indvars.iv
42  %Aidx = getelementptr inbounds i32, ptr %A, i64 %indvars.iv
43
44  %b = load i32, ptr %Bidx, align 4
45  %uncountable.c = icmp eq i32 %b, 10
46  br i1 %uncountable.c, label  %latch, label %for.end
47
48latch:
49  %a_p1 = add i32 %b, 2
50  store i32 %a_p1, ptr %Aidx_next, align 4
51
52  %a = load i32, ptr %Aidx, align 1
53  %c = mul i32 %a, 2
54  store i32 %c, ptr %Cidx, align 4
55
56  %exitcond = icmp eq i64 %indvars.iv.next, %N
57  br i1 %exitcond, label %for.end, label %for.body
58
59for.end:                                          ; preds = %for.body
60  ret void
61}
62