xref: /llvm-project/llvm/test/CodeGen/PowerPC/loop-hoist-toc-save.ll (revision 427fb35192f1f7bb694a5910b05abc5925a798b2)
1; RUN: llc -verify-machineinstrs -mtriple=powerpc64le-unknown-linux-gnu -hoist-const-stores -ppc-stack-ptr-caller-preserved < %s | FileCheck %s
2; RUN: llc -verify-machineinstrs -mtriple=powerpc64-unknown-linux-gnu -hoist-const-stores -ppc-stack-ptr-caller-preserved < %s | FileCheck %s -check-prefix=CHECKBE
3
4; Test hoist out of single loop
5define signext i32 @test1(i32 signext %lim, ptr nocapture %Func) {
6entry:
7; CHECK-LABEL: test1
8; CHECK: for.body.preheader
9; CHECK: std 2, 24(1)
10; CHECK: for.body
11; CHECK-NOT: std 2, 24(1)
12; CHECKBE-LABEL: test1
13; CHECKBE: for.body.preheader
14; CHECKBE: std 2, 40(1)
15; CHECKBE: for.body
16; CHECKBE-NOT: std 2, 40(1)
17
18  %cmp6 = icmp sgt i32 %lim, 0
19  br i1 %cmp6, label %for.body.preheader, label %for.cond.cleanup
20
21for.body.preheader:                               ; preds = %entry
22  br label %for.body
23
24for.cond.cleanup:                                 ; preds = %for.body, %entry
25  %Sum.0.lcssa = phi i32 [ 0, %entry ], [ %add, %for.body ]
26  ret i32 %Sum.0.lcssa
27
28for.body:                                         ; preds = %for.body.preheader, %for.body
29  %i.08 = phi i32 [ %inc, %for.body ], [ 0, %for.body.preheader ]
30  %Sum.07 = phi i32 [ %add, %for.body ], [ 0, %for.body.preheader ]
31  %call = tail call signext i32 %Func(i32 signext %i.08)
32  %add = add nsw i32 %call, %Sum.07
33  %inc = add nuw nsw i32 %i.08, 1
34  %exitcond = icmp eq i32 %inc, %lim
35  br i1 %exitcond, label %for.cond.cleanup, label %for.body
36}
37
38; Test hoist of nested loop goes to outter loop preheader
39define signext i32 @test2(i32 signext %lim, ptr nocapture %Func) {
40entry:
41; CHECK-LABEL: test2
42; CHECK: for.body4.lr.ph.preheader
43; CHECK: std 2, 24(1)
44; CHECK: for.body4.lr.ph
45; CHECK-NOT: std 2, 24(1)
46; CHECKBE-LABEL: test2
47; CHECKBE: for.body4.lr.ph.preheader
48; CHECKBE: std 2, 40(1)
49; CHECKBE: for.body4.lr.ph
50; CHECKBE-NOT: std 2, 40(1)
51
52  %cmp20 = icmp sgt i32 %lim, 0
53  br i1 %cmp20, label %for.body4.lr.ph.preheader, label %for.cond.cleanup
54
55for.body4.lr.ph.preheader:                        ; preds = %entry
56  br label %for.body4.lr.ph
57
58for.cond.cleanup:                                 ; preds = %for.cond.cleanup3, %entry
59  %Sum.0.lcssa = phi i32 [ 0, %entry ], [ %add, %for.cond.cleanup3 ]
60  ret i32 %Sum.0.lcssa
61
62for.body4.lr.ph:                                  ; preds = %for.body4.lr.ph.preheader, %for.cond.cleanup3
63  %j.022 = phi i32 [ %inc6, %for.cond.cleanup3 ], [ 0, %for.body4.lr.ph.preheader ]
64  %Sum.021 = phi i32 [ %add, %for.cond.cleanup3 ], [ 0, %for.body4.lr.ph.preheader ]
65  br label %for.body4
66
67for.cond.cleanup3:                                ; preds = %for.body4
68  %inc6 = add nuw nsw i32 %j.022, 1
69  %exitcond24 = icmp eq i32 %inc6, %lim
70  br i1 %exitcond24, label %for.cond.cleanup, label %for.body4.lr.ph
71
72for.body4:                                        ; preds = %for.body4, %for.body4.lr.ph
73  %i.019 = phi i32 [ %j.022, %for.body4.lr.ph ], [ %inc, %for.body4 ]
74  %Sum.118 = phi i32 [ %Sum.021, %for.body4.lr.ph ], [ %add, %for.body4 ]
75  %call = tail call signext i32 %Func(i32 signext %i.019)
76  %add = add nsw i32 %call, %Sum.118
77  %inc = add nuw nsw i32 %i.019, 1
78  %exitcond = icmp eq i32 %inc, %lim
79  br i1 %exitcond, label %for.cond.cleanup3, label %for.body4
80}
81
82; Test hoist out of if statement with low branch probability
83; FIXME: we shouldn't hoist in such cases as it could increase the number
84; of stores after hoisting.
85define signext i32 @test3(i32 signext %lim, ptr nocapture %Func) {
86entry:
87; CHECK-LABEL: test3
88; CHECK: %for.body.lr.ph
89; CHECK: std 2, 24(1)
90; CHECK: %for.body
91; CHECK-NOT: std 2, 24(1)
92; CHECKBE-LABEL: test3
93; CHECKBE: %for.body.lr.ph
94; CHECKBE: std 2, 40(1)
95; CHECKBE: %for.body
96; CHECKBE-NOT: std 2, 40(1)
97
98  %cmp13 = icmp sgt i32 %lim, 0
99  br i1 %cmp13, label %for.body.lr.ph, label %for.cond.cleanup
100
101for.body.lr.ph:                                   ; preds = %entry
102  %sub = add nsw i32 %lim, -1
103  br label %for.body
104
105for.cond.cleanup:                                 ; preds = %if.end, %entry
106  %Sum.0.lcssa = phi i32 [ 0, %entry ], [ %add3, %if.end ]
107  ret i32 %Sum.0.lcssa
108
109for.body:                                         ; preds = %if.end, %for.body.lr.ph
110  %i.015 = phi i32 [ 0, %for.body.lr.ph ], [ %inc, %if.end ]
111  %Sum.014 = phi i32 [ 0, %for.body.lr.ph ], [ %add3, %if.end ]
112  %cmp1 = icmp eq i32 %i.015, %sub
113  br i1 %cmp1, label %if.then, label %if.end
114
115if.then:                                          ; preds = %for.body
116  %call = tail call signext i32 %Func(i32 signext %sub)
117  %add = add nsw i32 %call, %Sum.014
118  br label %if.end
119
120if.end:                                           ; preds = %if.then, %for.body
121  %Sum.1 = phi i32 [ %add, %if.then ], [ %Sum.014, %for.body ]
122  %call2 = tail call signext i32 @func(i32 signext %i.015)
123  %add3 = add nsw i32 %call2, %Sum.1
124  %inc = add nuw nsw i32 %i.015, 1
125  %exitcond = icmp eq i32 %inc, %lim
126  br i1 %exitcond, label %for.cond.cleanup, label %for.body
127}
128
129declare signext i32 @func(i32 signext)
130