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