xref: /llvm-project/llvm/test/Transforms/LICM/invariant.start.ll (revision e390c229a438ed1eb3396df8fbeeda89c49474e6)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -passes=licm < %s -S | FileCheck %s
3; RUN: opt -aa-pipeline=basic-aa -passes='require<aa>,require<target-ir>,require<scalar-evolution>,require<opt-remark-emit>,loop-mssa(licm)' < %s -S | FileCheck %s
4
5define void @test1(i1 %cond, ptr %ptr) {
6; CHECK-LABEL: @test1(
7; CHECK-NEXT:  entry:
8; CHECK-NEXT:    [[TMP0:%.*]] = call ptr @llvm.invariant.start.p0(i64 4, ptr [[PTR:%.*]])
9; CHECK-NEXT:    [[VAL:%.*]] = load i32, ptr [[PTR]], align 4
10; CHECK-NEXT:    br label [[LOOP:%.*]]
11; CHECK:       loop:
12; CHECK-NEXT:    [[X:%.*]] = phi i32 [ 0, [[ENTRY:%.*]] ], [ [[X_INC:%.*]], [[LOOP]] ]
13; CHECK-NEXT:    [[X_INC]] = add i32 [[X]], [[VAL]]
14; CHECK-NEXT:    br label [[LOOP]]
15;
16
17entry:
18  br label %loop
19
20loop:
21  %x = phi i32 [ 0, %entry ], [ %x.inc, %loop ]
22  call ptr @llvm.invariant.start.p0(i64 4, ptr %ptr)
23  %val = load i32, ptr %ptr
24  %x.inc = add i32 %x, %val
25  br label %loop
26}
27
28;; despite the loop varying invariant.start, we should be
29;; able to hoist the load
30define void @test2(i1 %cond, ptr %ptr) {
31; CHECK-LABEL: @test2(
32; CHECK-NEXT:  entry:
33; CHECK-NEXT:    [[VAL:%.*]] = load i32, ptr [[PTR:%.*]], align 4
34; CHECK-NEXT:    br label [[LOOP:%.*]]
35; CHECK:       loop:
36; CHECK-NEXT:    [[X:%.*]] = phi i32 [ 0, [[ENTRY:%.*]] ], [ [[X_INC:%.*]], [[LOOP]] ]
37; CHECK-NEXT:    [[PIV:%.*]] = getelementptr i32, ptr [[PTR]], i32 [[X]]
38; CHECK-NEXT:    [[TMP0:%.*]] = call ptr @llvm.invariant.start.p0(i64 4, ptr [[PIV]])
39; CHECK-NEXT:    [[X_INC]] = add i32 [[X]], [[VAL]]
40; CHECK-NEXT:    br label [[LOOP]]
41;
42entry:
43  br label %loop
44
45loop:
46  %x = phi i32 [ 0, %entry ], [ %x.inc, %loop ]
47  %piv = getelementptr i32, ptr %ptr, i32 %x
48  call ptr @llvm.invariant.start.p0(i64 4, ptr %piv)
49  %val = load i32, ptr %ptr
50  %x.inc = add i32 %x, %val
51  br label %loop
52}
53
54define void @test3(i1 %cond, ptr %ptr) {
55; CHECK-LABEL: @test3(
56; CHECK-NEXT:  entry:
57; CHECK-NEXT:    [[TMP0:%.*]] = call ptr @llvm.invariant.start.p0(i64 4, ptr [[PTR:%.*]])
58; CHECK-NEXT:    [[VAL:%.*]] = load i32, ptr [[PTR]], align 4
59; CHECK-NEXT:    [[P2:%.*]] = getelementptr i32, ptr [[PTR]], i32 1
60; CHECK-NEXT:    br label [[LOOP:%.*]]
61; CHECK:       loop:
62; CHECK-NEXT:    [[X:%.*]] = phi i32 [ 0, [[ENTRY:%.*]] ], [ [[X_INC:%.*]], [[LOOP]] ]
63; CHECK-NEXT:    store volatile i32 0, ptr [[P2]], align 4
64; CHECK-NEXT:    [[X_INC]] = add i32 [[X]], [[VAL]]
65; CHECK-NEXT:    br label [[LOOP]]
66;
67entry:
68  br label %loop
69
70loop:
71  %x = phi i32 [ 0, %entry ], [ %x.inc, %loop ]
72  call ptr @llvm.invariant.start.p0(i64 4, ptr %ptr)
73  %val = load i32, ptr %ptr
74  %p2 = getelementptr i32, ptr %ptr, i32 1
75  store volatile i32 0, ptr %p2
76  %x.inc = add i32 %x, %val
77  br label %loop
78}
79
80; can't hoist due to init in loop, only well defined if loop exits
81; on first iteration, but we don't bother checking for that currently
82define void @test4(i1 %cond, ptr %ptr) {
83; CHECK-LABEL: @test4(
84; CHECK-NEXT:  entry:
85; CHECK-NEXT:    br label [[LOOP:%.*]]
86; CHECK:       loop:
87; CHECK-NEXT:    [[X:%.*]] = phi i32 [ 0, [[ENTRY:%.*]] ], [ [[X_INC:%.*]], [[LOOP]] ]
88; CHECK-NEXT:    store i32 0, ptr [[PTR:%.*]], align 4
89; CHECK-NEXT:    [[TMP0:%.*]] = call ptr @llvm.invariant.start.p0(i64 4, ptr [[PTR]])
90; CHECK-NEXT:    [[X_INC]] = add i32 [[X]], 0
91; CHECK-NEXT:    br label [[LOOP]]
92;
93entry:
94  br label %loop
95
96loop:
97  %x = phi i32 [ 0, %entry ], [ %x.inc, %loop ]
98  store i32 0, ptr %ptr
99  call ptr @llvm.invariant.start.p0(i64 4, ptr %ptr)
100  %val = load i32, ptr %ptr
101  %x.inc = add i32 %x, %val
102  br label %loop
103}
104
105; don't try to reason about scopes
106define void @test5(i1 %cond, ptr %ptr) {
107; CHECK-LABEL: @test5(
108; CHECK-NEXT:  entry:
109; CHECK-NEXT:    br label [[LOOP:%.*]]
110; CHECK:       loop:
111; CHECK-NEXT:    [[X:%.*]] = phi i32 [ 0, [[ENTRY:%.*]] ], [ [[X_INC:%.*]], [[LOOP]] ]
112; CHECK-NEXT:    store i32 0, ptr [[PTR:%.*]], align 4
113; CHECK-NEXT:    [[SCOPE:%.*]] = call ptr @llvm.invariant.start.p0(i64 4, ptr [[PTR]])
114; CHECK-NEXT:    [[VAL:%.*]] = load i32, ptr [[PTR]], align 4
115; CHECK-NEXT:    call void @llvm.invariant.end.p0(ptr [[SCOPE]], i64 4, ptr [[PTR]])
116; CHECK-NEXT:    [[X_INC]] = add i32 [[X]], [[VAL]]
117; CHECK-NEXT:    br label [[LOOP]]
118;
119entry:
120  br label %loop
121
122loop:
123  %x = phi i32 [ 0, %entry ], [ %x.inc, %loop ]
124  store i32 0, ptr %ptr
125  %scope = call ptr @llvm.invariant.start.p0(i64 4, ptr %ptr)
126  %val = load i32, ptr %ptr
127  call void @llvm.invariant.end.p0(ptr %scope, i64 4, ptr %ptr)
128  %x.inc = add i32 %x, %val
129  br label %loop
130}
131
132declare ptr @llvm.invariant.start.p0(i64, ptr)
133declare void @llvm.invariant.end.p0(ptr, i64, ptr)
134