xref: /llvm-project/llvm/test/Transforms/LoopDistribute/early-exit.ll (revision 344930316f4c901673461dcf44ad57ae6ade1015)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2; RUN: opt -aa-pipeline=basic-aa -passes=loop-distribute -enable-loop-distribute -verify-loop-info -verify-dom-info -S %s | FileCheck %s
3
4@B = common global ptr null, align 8
5@A = common global ptr null, align 8
6@C = common global ptr null, align 8
7@D = common global ptr null, align 8
8@E = common global ptr null, align 8
9
10define void @f() {
11; CHECK-LABEL: define void @f() {
12; CHECK-NEXT:  [[ENTRY:.*]]:
13; CHECK-NEXT:    [[A:%.*]] = load ptr, ptr @A, align 8
14; CHECK-NEXT:    [[B:%.*]] = load ptr, ptr @B, align 8
15; CHECK-NEXT:    [[C:%.*]] = load ptr, ptr @C, align 8
16; CHECK-NEXT:    [[D:%.*]] = load ptr, ptr @D, align 8
17; CHECK-NEXT:    [[E:%.*]] = load ptr, ptr @E, align 8
18; CHECK-NEXT:    br label %[[FOR_BODY:.*]]
19; CHECK:       [[FOR_BODY]]:
20; CHECK-NEXT:    [[IND:%.*]] = phi i64 [ 0, %[[ENTRY]] ], [ [[ADD:%.*]], %[[LATCH:.*]] ]
21; CHECK-NEXT:    [[ARRAYIDXA:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[IND]]
22; CHECK-NEXT:    [[LOADA:%.*]] = load i32, ptr [[ARRAYIDXA]], align 4
23; CHECK-NEXT:    [[ARRAYIDXB:%.*]] = getelementptr inbounds i32, ptr [[B]], i64 [[IND]]
24; CHECK-NEXT:    [[LOADB:%.*]] = load i32, ptr [[ARRAYIDXB]], align 4
25; CHECK-NEXT:    [[UNCOUNTABLE_C:%.*]] = icmp eq i32 [[LOADB]], 10
26; CHECK-NEXT:    br i1 [[UNCOUNTABLE_C]], label %[[FOR_END:.*]], label %[[LATCH]]
27; CHECK:       [[LATCH]]:
28; CHECK-NEXT:    [[MULA:%.*]] = mul i32 [[LOADB]], [[LOADA]]
29; CHECK-NEXT:    [[ADD]] = add nuw nsw i64 [[IND]], 1
30; CHECK-NEXT:    [[ARRAYIDXA_PLUS_4:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[ADD]]
31; CHECK-NEXT:    store i32 [[MULA]], ptr [[ARRAYIDXA_PLUS_4]], align 4
32; CHECK-NEXT:    [[ARRAYIDXD:%.*]] = getelementptr inbounds i32, ptr [[D]], i64 [[IND]]
33; CHECK-NEXT:    [[LOADD:%.*]] = load i32, ptr [[ARRAYIDXD]], align 4
34; CHECK-NEXT:    [[ARRAYIDXE:%.*]] = getelementptr inbounds i32, ptr [[E]], i64 [[IND]]
35; CHECK-NEXT:    [[LOADE:%.*]] = load i32, ptr [[ARRAYIDXE]], align 4
36; CHECK-NEXT:    [[MULC:%.*]] = mul i32 [[LOADD]], [[LOADE]]
37; CHECK-NEXT:    [[ARRAYIDXC:%.*]] = getelementptr inbounds i32, ptr [[C]], i64 [[IND]]
38; CHECK-NEXT:    store i32 [[MULC]], ptr [[ARRAYIDXC]], align 4
39; CHECK-NEXT:    [[EXITCOND:%.*]] = icmp eq i64 [[ADD]], 20
40; CHECK-NEXT:    br i1 [[EXITCOND]], label %[[FOR_END]], label %[[FOR_BODY]]
41; CHECK:       [[FOR_END]]:
42; CHECK-NEXT:    ret void
43;
44entry:
45  %a = load ptr, ptr @A, align 8
46  %b = load ptr, ptr @B, align 8
47  %c = load ptr, ptr @C, align 8
48  %d = load ptr, ptr @D, align 8
49  %e = load ptr, ptr @E, align 8
50  br label %for.body
51
52for.body:
53  %ind = phi i64 [ 0, %entry ], [ %add, %latch ]
54
55  %arrayidxA = getelementptr inbounds i32, ptr %a, i64 %ind
56  %loadA = load i32, ptr %arrayidxA, align 4
57
58  %arrayidxB = getelementptr inbounds i32, ptr %b, i64 %ind
59  %loadB = load i32, ptr %arrayidxB, align 4
60  %uncountable.c = icmp eq i32 %loadB, 10
61  br i1 %uncountable.c, label %for.end, label %latch
62
63latch:
64  %mulA = mul i32 %loadB, %loadA
65
66  %add = add nuw nsw i64 %ind, 1
67  %arrayidxA_plus_4 = getelementptr inbounds i32, ptr %a, i64 %add
68  store i32 %mulA, ptr %arrayidxA_plus_4, align 4
69
70  %arrayidxD = getelementptr inbounds i32, ptr %d, i64 %ind
71  %loadD = load i32, ptr %arrayidxD, align 4
72
73  %arrayidxE = getelementptr inbounds i32, ptr %e, i64 %ind
74  %loadE = load i32, ptr %arrayidxE, align 4
75
76  %mulC = mul i32 %loadD, %loadE
77
78  %arrayidxC = getelementptr inbounds i32, ptr %c, i64 %ind
79  store i32 %mulC, ptr %arrayidxC, align 4
80
81  %exitcond = icmp eq i64 %add, 20
82  br i1 %exitcond, label %for.end, label %for.body
83
84for.end:                                          ; preds = %for.body
85  ret void
86}
87
88attributes #0 = { nounwind readnone convergent }
89attributes #1 = { nounwind convergent }
90
91!0 = distinct !{!0, !1}
92!1 = !{!"llvm.loop.distribute.enable", i1 true}
93