blob: 8a3e5251fd692186f82ab3b70854629033b90064 [file] [log] [blame]
Chris Lattner55d47882003-10-12 21:44:18 +00001//===- LoopSimplify.cpp - Loop Canonicalization Pass ----------------------===//
Misha Brukmanb1c93172005-04-21 23:48:37 +00002//
John Criswell482202a2003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
Misha Brukmanb1c93172005-04-21 23:48:37 +00007//
John Criswell482202a2003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Chris Lattner61992f62002-09-26 16:17:31 +00009//
Chris Lattner154e4d52003-10-12 21:43:28 +000010// This pass performs several transformations to transform natural loops into a
11// simpler form, which makes subsequent analyses and transformations simpler and
12// more effective.
Chris Lattner650096a2003-02-27 20:27:08 +000013//
14// Loop pre-header insertion guarantees that there is a single, non-critical
15// entry edge from outside of the loop to the loop header. This simplifies a
16// number of analyses and transformations, such as LICM.
17//
18// Loop exit-block insertion guarantees that all exit blocks from the loop
19// (blocks which are outside of the loop that have predecessors inside of the
Chris Lattner7710f2f2003-12-10 17:20:35 +000020// loop) only have predecessors from inside of the loop (and are thus dominated
21// by the loop header). This simplifies transformations such as store-sinking
22// that are built into LICM.
Chris Lattner650096a2003-02-27 20:27:08 +000023//
Chris Lattnerc4622a62003-10-13 00:37:13 +000024// This pass also guarantees that loops will have exactly one backedge.
25//
Chris Lattner650096a2003-02-27 20:27:08 +000026// Note that the simplifycfg pass will clean up blocks which are split out but
Chris Lattner154e4d52003-10-12 21:43:28 +000027// end up being unnecessary, so usage of this pass should not pessimize
28// generated code.
29//
30// This pass obviously modifies the CFG, but updates loop information and
31// dominator information.
Chris Lattner61992f62002-09-26 16:17:31 +000032//
33//===----------------------------------------------------------------------===//
34
Chris Lattner45f966d2006-12-19 22:17:40 +000035#define DEBUG_TYPE "loopsimplify"
Chris Lattner61992f62002-09-26 16:17:31 +000036#include "llvm/Transforms/Scalar.h"
Chris Lattnerd0788122004-03-14 03:59:22 +000037#include "llvm/Constant.h"
Misha Brukman63b38bd2004-07-29 17:30:56 +000038#include "llvm/Instructions.h"
Chris Lattnerd0788122004-03-14 03:59:22 +000039#include "llvm/Function.h"
40#include "llvm/Type.h"
Chris Lattner514e8432005-03-25 06:37:22 +000041#include "llvm/Analysis/AliasAnalysis.h"
Chris Lattner031a3f82003-12-19 06:27:08 +000042#include "llvm/Analysis/Dominators.h"
43#include "llvm/Analysis/LoopInfo.h"
Chris Lattner61992f62002-09-26 16:17:31 +000044#include "llvm/Support/CFG.h"
Chris Lattner3d27be12006-08-27 12:54:02 +000045#include "llvm/Support/Compiler.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000046#include "llvm/ADT/SetOperations.h"
47#include "llvm/ADT/SetVector.h"
48#include "llvm/ADT/Statistic.h"
49#include "llvm/ADT/DepthFirstIterator.h"
Chris Lattner7710f2f2003-12-10 17:20:35 +000050using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000051
Chris Lattner45f966d2006-12-19 22:17:40 +000052STATISTIC(NumInserted, "Number of pre-header or exit blocks inserted");
53STATISTIC(NumNested , "Number of nested loops split out");
Chris Lattner61992f62002-09-26 16:17:31 +000054
Chris Lattner45f966d2006-12-19 22:17:40 +000055namespace {
Chris Lattner996795b2006-06-28 23:17:24 +000056 struct VISIBILITY_HIDDEN LoopSimplify : public FunctionPass {
Nick Lewyckye7da2d62007-05-06 13:37:16 +000057 static char ID; // Pass identification, replacement for typeid
Devang Patel09f162c2007-05-01 21:15:47 +000058 LoopSimplify() : FunctionPass((intptr_t)&ID) {}
59
Chris Lattner514e8432005-03-25 06:37:22 +000060 // AA - If we have an alias analysis object to update, this is it, otherwise
61 // this is null.
62 AliasAnalysis *AA;
Chris Lattnercffbbee2006-02-14 22:34:08 +000063 LoopInfo *LI;
Devang Pateld5258a232007-06-21 17:23:45 +000064 DominatorTree *DT;
Chris Lattner61992f62002-09-26 16:17:31 +000065 virtual bool runOnFunction(Function &F);
Misha Brukmanb1c93172005-04-21 23:48:37 +000066
Chris Lattner61992f62002-09-26 16:17:31 +000067 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
68 // We need loop information to identify the loops...
69 AU.addRequired<LoopInfo>();
Chris Lattner797cb2f2004-03-13 22:01:26 +000070 AU.addRequired<DominatorTree>();
Chris Lattner61992f62002-09-26 16:17:31 +000071
72 AU.addPreserved<LoopInfo>();
Chris Lattner61992f62002-09-26 16:17:31 +000073 AU.addPreserved<DominatorTree>();
Chris Lattner650096a2003-02-27 20:27:08 +000074 AU.addPreserved<DominanceFrontier>();
Chris Lattnerf83ce5f2005-08-10 02:07:32 +000075 AU.addPreservedID(BreakCriticalEdgesID); // No critical edges added.
Chris Lattner61992f62002-09-26 16:17:31 +000076 }
77 private:
78 bool ProcessLoop(Loop *L);
Chris Lattner650096a2003-02-27 20:27:08 +000079 BasicBlock *SplitBlockPredecessors(BasicBlock *BB, const char *Suffix,
80 const std::vector<BasicBlock*> &Preds);
Chris Lattner82782632004-04-18 22:27:10 +000081 BasicBlock *RewriteLoopExitBlock(Loop *L, BasicBlock *Exit);
Chris Lattner61992f62002-09-26 16:17:31 +000082 void InsertPreheaderForLoop(Loop *L);
Chris Lattner84170522004-04-13 05:05:33 +000083 Loop *SeparateNestedLoop(Loop *L);
Chris Lattnerc4622a62003-10-13 00:37:13 +000084 void InsertUniqueBackedgeBlock(Loop *L);
Chris Lattner6bd6da42006-09-23 08:19:21 +000085 void PlaceSplitBlockCarefully(BasicBlock *NewBB,
86 std::vector<BasicBlock*> &SplitPreds,
87 Loop *L);
Chris Lattner61992f62002-09-26 16:17:31 +000088 };
89
Devang Patel8c78a0b2007-05-03 01:11:54 +000090 char LoopSimplify::ID = 0;
Chris Lattnerc2d3d312006-08-27 22:42:52 +000091 RegisterPass<LoopSimplify>
Chris Lattner154e4d52003-10-12 21:43:28 +000092 X("loopsimplify", "Canonicalize natural loops", true);
Chris Lattner61992f62002-09-26 16:17:31 +000093}
94
95// Publically exposed interface to pass...
Chris Lattner7710f2f2003-12-10 17:20:35 +000096const PassInfo *llvm::LoopSimplifyID = X.getPassInfo();
Chris Lattner3e860842004-09-20 04:43:15 +000097FunctionPass *llvm::createLoopSimplifyPass() { return new LoopSimplify(); }
Chris Lattner61992f62002-09-26 16:17:31 +000098
Chris Lattner61992f62002-09-26 16:17:31 +000099/// runOnFunction - Run down all loops in the CFG (recursively, but we could do
100/// it in any convenient order) inserting preheaders...
101///
Chris Lattner154e4d52003-10-12 21:43:28 +0000102bool LoopSimplify::runOnFunction(Function &F) {
Chris Lattner61992f62002-09-26 16:17:31 +0000103 bool Changed = false;
Chris Lattnercffbbee2006-02-14 22:34:08 +0000104 LI = &getAnalysis<LoopInfo>();
Chris Lattner514e8432005-03-25 06:37:22 +0000105 AA = getAnalysisToUpdate<AliasAnalysis>();
Devang Pateld5258a232007-06-21 17:23:45 +0000106 DT = &getAnalysis<DominatorTree>();
Chris Lattner61992f62002-09-26 16:17:31 +0000107
Chris Lattner85d99442006-08-12 04:51:20 +0000108 // Check to see that no blocks (other than the header) in loops have
109 // predecessors that are not in loops. This is not valid for natural loops,
110 // but can occur if the blocks are unreachable. Since they are unreachable we
111 // can just shamelessly destroy their terminators to make them not branch into
112 // the loop!
113 for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) {
114 // This case can only occur for unreachable blocks. Blocks that are
115 // unreachable can't be in loops, so filter those blocks out.
116 if (LI->getLoopFor(BB)) continue;
117
118 bool BlockUnreachable = false;
119 TerminatorInst *TI = BB->getTerminator();
120
121 // Check to see if any successors of this block are non-loop-header loops
122 // that are not the header.
123 for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i) {
124 // If this successor is not in a loop, BB is clearly ok.
125 Loop *L = LI->getLoopFor(TI->getSuccessor(i));
126 if (!L) continue;
127
128 // If the succ is the loop header, and if L is a top-level loop, then this
129 // is an entrance into a loop through the header, which is also ok.
130 if (L->getHeader() == TI->getSuccessor(i) && L->getParentLoop() == 0)
131 continue;
132
133 // Otherwise, this is an entrance into a loop from some place invalid.
134 // Either the loop structure is invalid and this is not a natural loop (in
135 // which case the compiler is buggy somewhere else) or BB is unreachable.
136 BlockUnreachable = true;
137 break;
138 }
139
140 // If this block is ok, check the next one.
141 if (!BlockUnreachable) continue;
142
143 // Otherwise, this block is dead. To clean up the CFG and to allow later
144 // loop transformations to ignore this case, we delete the edges into the
145 // loop by replacing the terminator.
146
147 // Remove PHI entries from the successors.
148 for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i)
149 TI->getSuccessor(i)->removePredecessor(BB);
150
151 // Add a new unreachable instruction.
152 new UnreachableInst(TI);
153
154 // Delete the dead terminator.
155 if (AA) AA->deleteValue(&BB->back());
156 BB->getInstList().pop_back();
157 Changed |= true;
158 }
159
Chris Lattnercffbbee2006-02-14 22:34:08 +0000160 for (LoopInfo::iterator I = LI->begin(), E = LI->end(); I != E; ++I)
Chris Lattner59d2d7f2004-01-08 00:09:44 +0000161 Changed |= ProcessLoop(*I);
Chris Lattner61992f62002-09-26 16:17:31 +0000162
163 return Changed;
164}
165
Chris Lattner61992f62002-09-26 16:17:31 +0000166/// ProcessLoop - Walk the loop structure in depth first order, ensuring that
167/// all loops have preheaders.
168///
Chris Lattner154e4d52003-10-12 21:43:28 +0000169bool LoopSimplify::ProcessLoop(Loop *L) {
Chris Lattner61992f62002-09-26 16:17:31 +0000170 bool Changed = false;
Chris Lattnerf18b3962006-08-12 05:25:00 +0000171ReprocessLoop:
172
Chris Lattner9c5693f2006-02-14 23:06:02 +0000173 // Canonicalize inner loops before outer loops. Inner loop canonicalization
174 // can provide work for the outer loop to canonicalize.
175 for (Loop::iterator I = L->begin(), E = L->end(); I != E; ++I)
176 Changed |= ProcessLoop(*I);
177
Chris Lattnerd0788122004-03-14 03:59:22 +0000178 assert(L->getBlocks()[0] == L->getHeader() &&
179 "Header isn't first block in loop?");
Chris Lattnerd0788122004-03-14 03:59:22 +0000180
Chris Lattner85d99442006-08-12 04:51:20 +0000181 // Does the loop already have a preheader? If so, don't insert one.
Chris Lattner61992f62002-09-26 16:17:31 +0000182 if (L->getLoopPreheader() == 0) {
183 InsertPreheaderForLoop(L);
184 NumInserted++;
185 Changed = true;
186 }
187
Chris Lattner7710f2f2003-12-10 17:20:35 +0000188 // Next, check to make sure that all exit nodes of the loop only have
189 // predecessors that are inside of the loop. This check guarantees that the
190 // loop preheader/header will dominate the exit blocks. If the exit block has
Chris Lattner02f53ad2006-02-12 01:59:10 +0000191 // predecessors from outside of the loop, split the edge now.
192 std::vector<BasicBlock*> ExitBlocks;
193 L->getExitBlocks(ExitBlocks);
Chris Lattnercffbbee2006-02-14 22:34:08 +0000194
Chris Lattner02f53ad2006-02-12 01:59:10 +0000195 SetVector<BasicBlock*> ExitBlockSet(ExitBlocks.begin(), ExitBlocks.end());
Chris Lattnerf2c018c2004-07-15 08:20:22 +0000196 for (SetVector<BasicBlock*>::iterator I = ExitBlockSet.begin(),
197 E = ExitBlockSet.end(); I != E; ++I) {
198 BasicBlock *ExitBlock = *I;
Chris Lattnerdaa12132004-07-15 05:36:31 +0000199 for (pred_iterator PI = pred_begin(ExitBlock), PE = pred_end(ExitBlock);
200 PI != PE; ++PI)
Chris Lattner05bf90d2006-02-11 02:13:17 +0000201 // Must be exactly this loop: no subloops, parent loops, or non-loop preds
202 // allowed.
Chris Lattner02f53ad2006-02-12 01:59:10 +0000203 if (!L->contains(*PI)) {
Chris Lattnerf2c018c2004-07-15 08:20:22 +0000204 RewriteLoopExitBlock(L, ExitBlock);
Chris Lattnerdaa12132004-07-15 05:36:31 +0000205 NumInserted++;
206 Changed = true;
207 break;
208 }
Chris Lattnerf2c018c2004-07-15 08:20:22 +0000209 }
Chris Lattner650096a2003-02-27 20:27:08 +0000210
Chris Lattner84170522004-04-13 05:05:33 +0000211 // If the header has more than two predecessors at this point (from the
212 // preheader and from multiple backedges), we must adjust the loop.
Chris Lattnerf18b3962006-08-12 05:25:00 +0000213 unsigned NumBackedges = L->getNumBackEdges();
214 if (NumBackedges != 1) {
215 // If this is really a nested loop, rip it out into a child loop. Don't do
216 // this for loops with a giant number of backedges, just factor them into a
217 // common backedge instead.
218 if (NumBackedges < 8) {
219 if (Loop *NL = SeparateNestedLoop(L)) {
220 ++NumNested;
221 // This is a big restructuring change, reprocess the whole loop.
222 ProcessLoop(NL);
223 Changed = true;
224 // GCC doesn't tail recursion eliminate this.
225 goto ReprocessLoop;
226 }
Chris Lattner84170522004-04-13 05:05:33 +0000227 }
228
Chris Lattnerf18b3962006-08-12 05:25:00 +0000229 // If we either couldn't, or didn't want to, identify nesting of the loops,
230 // insert a new block that all backedges target, then make it jump to the
231 // loop header.
Chris Lattnerc4622a62003-10-13 00:37:13 +0000232 InsertUniqueBackedgeBlock(L);
233 NumInserted++;
234 Changed = true;
235 }
236
Chris Lattnerf83ce5f2005-08-10 02:07:32 +0000237 // Scan over the PHI nodes in the loop header. Since they now have only two
238 // incoming values (the loop is canonicalized), we may have simplified the PHI
239 // down to 'X = phi [X, Y]', which should be replaced with 'Y'.
240 PHINode *PN;
Chris Lattnerf83ce5f2005-08-10 02:07:32 +0000241 for (BasicBlock::iterator I = L->getHeader()->begin();
242 (PN = dyn_cast<PHINode>(I++)); )
Chris Lattner62df7982005-08-10 17:15:20 +0000243 if (Value *V = PN->hasConstantValue()) {
Chris Lattnerf83ce5f2005-08-10 02:07:32 +0000244 PN->replaceAllUsesWith(V);
245 PN->eraseFromParent();
246 }
247
Chris Lattner61992f62002-09-26 16:17:31 +0000248 return Changed;
249}
250
Chris Lattner650096a2003-02-27 20:27:08 +0000251/// SplitBlockPredecessors - Split the specified block into two blocks. We want
252/// to move the predecessors specified in the Preds list to point to the new
253/// block, leaving the remaining predecessors pointing to BB. This method
254/// updates the SSA PHINode's, but no other analyses.
255///
Chris Lattner154e4d52003-10-12 21:43:28 +0000256BasicBlock *LoopSimplify::SplitBlockPredecessors(BasicBlock *BB,
257 const char *Suffix,
Chris Lattner650096a2003-02-27 20:27:08 +0000258 const std::vector<BasicBlock*> &Preds) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000259
Chris Lattner650096a2003-02-27 20:27:08 +0000260 // Create new basic block, insert right before the original block...
Chris Lattner8d414ad2004-02-04 03:58:28 +0000261 BasicBlock *NewBB = new BasicBlock(BB->getName()+Suffix, BB->getParent(), BB);
Chris Lattner650096a2003-02-27 20:27:08 +0000262
263 // The preheader first gets an unconditional branch to the loop header...
Chris Lattnera2960002003-11-21 16:52:05 +0000264 BranchInst *BI = new BranchInst(BB, NewBB);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000265
Chris Lattner650096a2003-02-27 20:27:08 +0000266 // For every PHI node in the block, insert a PHI node into NewBB where the
267 // incoming values from the out of loop edges are moved to NewBB. We have two
268 // possible cases here. If the loop is dead, we just insert dummy entries
269 // into the PHI nodes for the new edge. If the loop is not dead, we move the
270 // incoming edges in BB into new PHI nodes in NewBB.
271 //
272 if (!Preds.empty()) { // Is the loop not obviously dead?
Chris Lattner031a3f82003-12-19 06:27:08 +0000273 // Check to see if the values being merged into the new block need PHI
274 // nodes. If so, insert them.
Alkis Evlogimenos3ce42ec2004-09-28 02:40:37 +0000275 for (BasicBlock::iterator I = BB->begin(); isa<PHINode>(I); ) {
276 PHINode *PN = cast<PHINode>(I);
Chris Lattner84170522004-04-13 05:05:33 +0000277 ++I;
278
Chris Lattner031a3f82003-12-19 06:27:08 +0000279 // Check to see if all of the values coming in are the same. If so, we
280 // don't need to create a new PHI node.
281 Value *InVal = PN->getIncomingValueForBlock(Preds[0]);
282 for (unsigned i = 1, e = Preds.size(); i != e; ++i)
283 if (InVal != PN->getIncomingValueForBlock(Preds[i])) {
284 InVal = 0;
285 break;
286 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000287
Chris Lattner031a3f82003-12-19 06:27:08 +0000288 // If the values coming into the block are not the same, we need a PHI.
289 if (InVal == 0) {
Chris Lattner6c237bc2003-12-09 23:12:55 +0000290 // Create the new PHI node, insert it into NewBB at the end of the block
291 PHINode *NewPHI = new PHINode(PN->getType(), PN->getName()+".ph", BI);
Chris Lattner514e8432005-03-25 06:37:22 +0000292 if (AA) AA->copyValue(PN, NewPHI);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000293
Chris Lattner6c237bc2003-12-09 23:12:55 +0000294 // Move all of the edges from blocks outside the loop to the new PHI
295 for (unsigned i = 0, e = Preds.size(); i != e; ++i) {
Chris Lattner84170522004-04-13 05:05:33 +0000296 Value *V = PN->removeIncomingValue(Preds[i], false);
Chris Lattner6c237bc2003-12-09 23:12:55 +0000297 NewPHI->addIncoming(V, Preds[i]);
298 }
Chris Lattner031a3f82003-12-19 06:27:08 +0000299 InVal = NewPHI;
300 } else {
301 // Remove all of the edges coming into the PHI nodes from outside of the
302 // block.
303 for (unsigned i = 0, e = Preds.size(); i != e; ++i)
304 PN->removeIncomingValue(Preds[i], false);
Chris Lattner6c237bc2003-12-09 23:12:55 +0000305 }
Chris Lattner031a3f82003-12-19 06:27:08 +0000306
307 // Add an incoming value to the PHI node in the loop for the preheader
308 // edge.
309 PN->addIncoming(InVal, NewBB);
Chris Lattner84170522004-04-13 05:05:33 +0000310
311 // Can we eliminate this phi node now?
Chris Lattner257efb22005-08-05 00:57:45 +0000312 if (Value *V = PN->hasConstantValue(true)) {
Nick Lewyckye6c64462007-04-08 01:04:30 +0000313 Instruction *I = dyn_cast<Instruction>(V);
Devang Patelbecc4662007-06-08 01:50:32 +0000314 // If I is in NewBB, the Dominator call will fail, because NewBB isn't
315 // registered in DominatorTree yet. Handle this case explicitly.
Nick Lewyckye6c64462007-04-08 01:04:30 +0000316 if (!I || (I->getParent() != NewBB &&
Devang Patelbecc4662007-06-08 01:50:32 +0000317 getAnalysis<DominatorTree>().dominates(I, PN))) {
Chris Lattnere29d6342004-10-17 21:22:38 +0000318 PN->replaceAllUsesWith(V);
Chris Lattner514e8432005-03-25 06:37:22 +0000319 if (AA) AA->deleteValue(PN);
Chris Lattnere29d6342004-10-17 21:22:38 +0000320 BB->getInstList().erase(PN);
321 }
Chris Lattner84170522004-04-13 05:05:33 +0000322 }
Chris Lattner650096a2003-02-27 20:27:08 +0000323 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000324
Chris Lattner650096a2003-02-27 20:27:08 +0000325 // Now that the PHI nodes are updated, actually move the edges from
326 // Preds to point to NewBB instead of BB.
327 //
328 for (unsigned i = 0, e = Preds.size(); i != e; ++i) {
329 TerminatorInst *TI = Preds[i]->getTerminator();
330 for (unsigned s = 0, e = TI->getNumSuccessors(); s != e; ++s)
331 if (TI->getSuccessor(s) == BB)
332 TI->setSuccessor(s, NewBB);
333 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000334
Chris Lattner650096a2003-02-27 20:27:08 +0000335 } else { // Otherwise the loop is dead...
Alkis Evlogimenos3ce42ec2004-09-28 02:40:37 +0000336 for (BasicBlock::iterator I = BB->begin(); isa<PHINode>(I); ++I) {
337 PHINode *PN = cast<PHINode>(I);
Chris Lattner650096a2003-02-27 20:27:08 +0000338 // Insert dummy values as the incoming value...
339 PN->addIncoming(Constant::getNullValue(PN->getType()), NewBB);
Alkis Evlogimenos3ce42ec2004-09-28 02:40:37 +0000340 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000341 }
Devang Pateld5258a232007-06-21 17:23:45 +0000342
Chris Lattner650096a2003-02-27 20:27:08 +0000343 return NewBB;
344}
345
Chris Lattner61992f62002-09-26 16:17:31 +0000346/// InsertPreheaderForLoop - Once we discover that a loop doesn't have a
347/// preheader, this method is called to insert one. This method has two phases:
348/// preheader insertion and analysis updating.
349///
Chris Lattner154e4d52003-10-12 21:43:28 +0000350void LoopSimplify::InsertPreheaderForLoop(Loop *L) {
Chris Lattner61992f62002-09-26 16:17:31 +0000351 BasicBlock *Header = L->getHeader();
352
353 // Compute the set of predecessors of the loop that are not in the loop.
354 std::vector<BasicBlock*> OutsideBlocks;
355 for (pred_iterator PI = pred_begin(Header), PE = pred_end(Header);
356 PI != PE; ++PI)
Chris Lattner05bf90d2006-02-11 02:13:17 +0000357 if (!L->contains(*PI)) // Coming in from outside the loop?
358 OutsideBlocks.push_back(*PI); // Keep track of it...
Misha Brukmanb1c93172005-04-21 23:48:37 +0000359
Chris Lattner608cd052006-09-23 07:40:52 +0000360 // Split out the loop pre-header.
Chris Lattner650096a2003-02-27 20:27:08 +0000361 BasicBlock *NewBB =
362 SplitBlockPredecessors(Header, ".preheader", OutsideBlocks);
Chris Lattner608cd052006-09-23 07:40:52 +0000363
Misha Brukmanb1c93172005-04-21 23:48:37 +0000364
Chris Lattner61992f62002-09-26 16:17:31 +0000365 //===--------------------------------------------------------------------===//
Misha Brukman8b2bd4e2003-10-10 17:57:28 +0000366 // Update analysis results now that we have performed the transformation
Chris Lattner61992f62002-09-26 16:17:31 +0000367 //
Misha Brukmanb1c93172005-04-21 23:48:37 +0000368
Chris Lattner61992f62002-09-26 16:17:31 +0000369 // We know that we have loop information to update... update it now.
370 if (Loop *Parent = L->getParentLoop())
Chris Lattnercffbbee2006-02-14 22:34:08 +0000371 Parent->addBasicBlockToLoop(NewBB, *LI);
Chris Lattnerf2d9f942003-02-27 22:48:57 +0000372
Devang Pateld5258a232007-06-21 17:23:45 +0000373 DT->splitBlock(NewBB);
374 if (DominanceFrontier *DF = getAnalysisToUpdate<DominanceFrontier>())
375 DF->splitBlock(NewBB);
376
Chris Lattner6bd6da42006-09-23 08:19:21 +0000377 // Make sure that NewBB is put someplace intelligent, which doesn't mess up
378 // code layout too horribly.
379 PlaceSplitBlockCarefully(NewBB, OutsideBlocks, L);
Chris Lattner650096a2003-02-27 20:27:08 +0000380}
381
Chris Lattner84170522004-04-13 05:05:33 +0000382/// RewriteLoopExitBlock - Ensure that the loop preheader dominates all exit
383/// blocks. This method is used to split exit blocks that have predecessors
384/// outside of the loop.
Chris Lattner82782632004-04-18 22:27:10 +0000385BasicBlock *LoopSimplify::RewriteLoopExitBlock(Loop *L, BasicBlock *Exit) {
Chris Lattner650096a2003-02-27 20:27:08 +0000386 std::vector<BasicBlock*> LoopBlocks;
387 for (pred_iterator I = pred_begin(Exit), E = pred_end(Exit); I != E; ++I)
388 if (L->contains(*I))
389 LoopBlocks.push_back(*I);
390
Chris Lattner10b2b052003-02-27 22:31:07 +0000391 assert(!LoopBlocks.empty() && "No edges coming in from outside the loop?");
392 BasicBlock *NewBB = SplitBlockPredecessors(Exit, ".loopexit", LoopBlocks);
393
Chris Lattnercffbbee2006-02-14 22:34:08 +0000394 // Update Loop Information - we know that the new block will be in whichever
395 // loop the Exit block is in. Note that it may not be in that immediate loop,
396 // if the successor is some other loop header. In that case, we continue
397 // walking up the loop tree to find a loop that contains both the successor
398 // block and the predecessor block.
399 Loop *SuccLoop = LI->getLoopFor(Exit);
400 while (SuccLoop && !SuccLoop->contains(L->getHeader()))
401 SuccLoop = SuccLoop->getParentLoop();
402 if (SuccLoop)
403 SuccLoop->addBasicBlockToLoop(NewBB, *LI);
Chris Lattner32a39c22003-02-28 03:07:54 +0000404
Devang Pateld5258a232007-06-21 17:23:45 +0000405 // Update Dominator Information
406 DT->splitBlock(NewBB);
407 if (DominanceFrontier *DF = getAnalysisToUpdate<DominanceFrontier>())
408 DF->splitBlock(NewBB);
409
Chris Lattner82782632004-04-18 22:27:10 +0000410 return NewBB;
Chris Lattnerc4622a62003-10-13 00:37:13 +0000411}
412
Chris Lattner84170522004-04-13 05:05:33 +0000413/// AddBlockAndPredsToSet - Add the specified block, and all of its
414/// predecessors, to the specified set, if it's not already in there. Stop
415/// predecessor traversal when we reach StopBlock.
Devang Patel83a3adc2007-04-20 20:04:37 +0000416static void AddBlockAndPredsToSet(BasicBlock *InputBB, BasicBlock *StopBlock,
Chris Lattner84170522004-04-13 05:05:33 +0000417 std::set<BasicBlock*> &Blocks) {
Devang Patel83a3adc2007-04-20 20:04:37 +0000418 std::vector<BasicBlock *> WorkList;
419 WorkList.push_back(InputBB);
420 do {
421 BasicBlock *BB = WorkList.back(); WorkList.pop_back();
422 if (Blocks.insert(BB).second && BB != StopBlock)
423 // If BB is not already processed and it is not a stop block then
424 // insert its predecessor in the work list
425 for (pred_iterator I = pred_begin(BB), E = pred_end(BB); I != E; ++I) {
426 BasicBlock *WBB = *I;
427 WorkList.push_back(WBB);
428 }
429 } while(!WorkList.empty());
Chris Lattner84170522004-04-13 05:05:33 +0000430}
431
Chris Lattnera6e22812004-04-13 15:21:18 +0000432/// FindPHIToPartitionLoops - The first part of loop-nestification is to find a
433/// PHI node that tells us how to partition the loops.
Devang Patelbecc4662007-06-08 01:50:32 +0000434static PHINode *FindPHIToPartitionLoops(Loop *L, DominatorTree *DT,
Owen Anderson59177162007-04-09 22:54:50 +0000435 AliasAnalysis *AA) {
Alkis Evlogimenos3ce42ec2004-09-28 02:40:37 +0000436 for (BasicBlock::iterator I = L->getHeader()->begin(); isa<PHINode>(I); ) {
437 PHINode *PN = cast<PHINode>(I);
Chris Lattnera6e22812004-04-13 15:21:18 +0000438 ++I;
Nate Begemanb3923212005-08-04 23:24:19 +0000439 if (Value *V = PN->hasConstantValue())
Devang Patelbecc4662007-06-08 01:50:32 +0000440 if (!isa<Instruction>(V) || DT->dominates(cast<Instruction>(V), PN)) {
Chris Lattnere29d6342004-10-17 21:22:38 +0000441 // This is a degenerate PHI already, don't modify it!
442 PN->replaceAllUsesWith(V);
Chris Lattner514e8432005-03-25 06:37:22 +0000443 if (AA) AA->deleteValue(PN);
Chris Lattnerdd3ec922005-03-06 21:35:38 +0000444 PN->eraseFromParent();
Chris Lattnere29d6342004-10-17 21:22:38 +0000445 continue;
446 }
447
448 // Scan this PHI node looking for a use of the PHI node by itself.
449 for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i)
450 if (PN->getIncomingValue(i) == PN &&
451 L->contains(PN->getIncomingBlock(i)))
452 // We found something tasty to remove.
453 return PN;
Chris Lattnera6e22812004-04-13 15:21:18 +0000454 }
455 return 0;
456}
457
Chris Lattner6bd6da42006-09-23 08:19:21 +0000458// PlaceSplitBlockCarefully - If the block isn't already, move the new block to
459// right after some 'outside block' block. This prevents the preheader from
460// being placed inside the loop body, e.g. when the loop hasn't been rotated.
461void LoopSimplify::PlaceSplitBlockCarefully(BasicBlock *NewBB,
462 std::vector<BasicBlock*>&SplitPreds,
463 Loop *L) {
464 // Check to see if NewBB is already well placed.
465 Function::iterator BBI = NewBB; --BBI;
466 for (unsigned i = 0, e = SplitPreds.size(); i != e; ++i) {
467 if (&*BBI == SplitPreds[i])
468 return;
469 }
470
471 // If it isn't already after an outside block, move it after one. This is
472 // always good as it makes the uncond branch from the outside block into a
473 // fall-through.
474
475 // Figure out *which* outside block to put this after. Prefer an outside
476 // block that neighbors a BB actually in the loop.
477 BasicBlock *FoundBB = 0;
478 for (unsigned i = 0, e = SplitPreds.size(); i != e; ++i) {
479 Function::iterator BBI = SplitPreds[i];
480 if (++BBI != NewBB->getParent()->end() &&
481 L->contains(BBI)) {
482 FoundBB = SplitPreds[i];
483 break;
484 }
485 }
486
487 // If our heuristic for a *good* bb to place this after doesn't find
488 // anything, just pick something. It's likely better than leaving it within
489 // the loop.
490 if (!FoundBB)
491 FoundBB = SplitPreds[0];
492 NewBB->moveAfter(FoundBB);
493}
494
495
Chris Lattner84170522004-04-13 05:05:33 +0000496/// SeparateNestedLoop - If this loop has multiple backedges, try to pull one of
497/// them out into a nested loop. This is important for code that looks like
498/// this:
499///
500/// Loop:
501/// ...
502/// br cond, Loop, Next
503/// ...
504/// br cond2, Loop, Out
505///
506/// To identify this common case, we look at the PHI nodes in the header of the
507/// loop. PHI nodes with unchanging values on one backedge correspond to values
508/// that change in the "outer" loop, but not in the "inner" loop.
509///
510/// If we are able to separate out a loop, return the new outer loop that was
511/// created.
512///
513Loop *LoopSimplify::SeparateNestedLoop(Loop *L) {
Devang Patelbecc4662007-06-08 01:50:32 +0000514 PHINode *PN = FindPHIToPartitionLoops(L, DT, AA);
Chris Lattnera6e22812004-04-13 15:21:18 +0000515 if (PN == 0) return 0; // No known way to partition.
Chris Lattner84170522004-04-13 05:05:33 +0000516
Chris Lattnera6e22812004-04-13 15:21:18 +0000517 // Pull out all predecessors that have varying values in the loop. This
518 // handles the case when a PHI node has multiple instances of itself as
519 // arguments.
Chris Lattner84170522004-04-13 05:05:33 +0000520 std::vector<BasicBlock*> OuterLoopPreds;
Chris Lattnera6e22812004-04-13 15:21:18 +0000521 for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i)
522 if (PN->getIncomingValue(i) != PN ||
523 !L->contains(PN->getIncomingBlock(i)))
524 OuterLoopPreds.push_back(PN->getIncomingBlock(i));
Chris Lattner84170522004-04-13 05:05:33 +0000525
Chris Lattner89e959b2004-04-13 16:23:25 +0000526 BasicBlock *Header = L->getHeader();
Chris Lattner84170522004-04-13 05:05:33 +0000527 BasicBlock *NewBB = SplitBlockPredecessors(Header, ".outer", OuterLoopPreds);
528
Devang Pateld5258a232007-06-21 17:23:45 +0000529 // Update dominator information
530 DT->splitBlock(NewBB);
531 if (DominanceFrontier *DF = getAnalysisToUpdate<DominanceFrontier>())
532 DF->splitBlock(NewBB);
Chris Lattner84170522004-04-13 05:05:33 +0000533
Chris Lattner6bd6da42006-09-23 08:19:21 +0000534 // Make sure that NewBB is put someplace intelligent, which doesn't mess up
535 // code layout too horribly.
536 PlaceSplitBlockCarefully(NewBB, OuterLoopPreds, L);
537
Chris Lattner84170522004-04-13 05:05:33 +0000538 // Create the new outer loop.
539 Loop *NewOuter = new Loop();
540
Chris Lattner84170522004-04-13 05:05:33 +0000541 // Change the parent loop to use the outer loop as its child now.
542 if (Loop *Parent = L->getParentLoop())
543 Parent->replaceChildLoopWith(L, NewOuter);
544 else
Chris Lattnercffbbee2006-02-14 22:34:08 +0000545 LI->changeTopLevelLoop(L, NewOuter);
Chris Lattner84170522004-04-13 05:05:33 +0000546
547 // This block is going to be our new header block: add it to this loop and all
548 // parent loops.
Chris Lattnercffbbee2006-02-14 22:34:08 +0000549 NewOuter->addBasicBlockToLoop(NewBB, *LI);
Chris Lattner84170522004-04-13 05:05:33 +0000550
551 // L is now a subloop of our outer loop.
552 NewOuter->addChildLoop(L);
553
Chris Lattner84170522004-04-13 05:05:33 +0000554 for (unsigned i = 0, e = L->getBlocks().size(); i != e; ++i)
555 NewOuter->addBlockEntry(L->getBlocks()[i]);
556
557 // Determine which blocks should stay in L and which should be moved out to
558 // the Outer loop now.
Chris Lattner84170522004-04-13 05:05:33 +0000559 std::set<BasicBlock*> BlocksInL;
560 for (pred_iterator PI = pred_begin(Header), E = pred_end(Header); PI!=E; ++PI)
Devang Patelbecc4662007-06-08 01:50:32 +0000561 if (DT->dominates(Header, *PI))
Chris Lattner84170522004-04-13 05:05:33 +0000562 AddBlockAndPredsToSet(*PI, Header, BlocksInL);
563
564
565 // Scan all of the loop children of L, moving them to OuterLoop if they are
566 // not part of the inner loop.
567 for (Loop::iterator I = L->begin(); I != L->end(); )
568 if (BlocksInL.count((*I)->getHeader()))
569 ++I; // Loop remains in L
570 else
571 NewOuter->addChildLoop(L->removeChildLoop(I));
572
573 // Now that we know which blocks are in L and which need to be moved to
574 // OuterLoop, move any blocks that need it.
575 for (unsigned i = 0; i != L->getBlocks().size(); ++i) {
576 BasicBlock *BB = L->getBlocks()[i];
577 if (!BlocksInL.count(BB)) {
578 // Move this block to the parent, updating the exit blocks sets
579 L->removeBlockFromLoop(BB);
Chris Lattnercffbbee2006-02-14 22:34:08 +0000580 if ((*LI)[BB] == L)
581 LI->changeLoopFor(BB, NewOuter);
Chris Lattner84170522004-04-13 05:05:33 +0000582 --i;
583 }
584 }
585
Chris Lattner84170522004-04-13 05:05:33 +0000586 return NewOuter;
587}
588
589
590
Chris Lattnerc4622a62003-10-13 00:37:13 +0000591/// InsertUniqueBackedgeBlock - This method is called when the specified loop
592/// has more than one backedge in it. If this occurs, revector all of these
593/// backedges to target a new basic block and have that block branch to the loop
594/// header. This ensures that loops have exactly one backedge.
595///
596void LoopSimplify::InsertUniqueBackedgeBlock(Loop *L) {
597 assert(L->getNumBackEdges() > 1 && "Must have > 1 backedge!");
598
599 // Get information about the loop
600 BasicBlock *Preheader = L->getLoopPreheader();
601 BasicBlock *Header = L->getHeader();
602 Function *F = Header->getParent();
603
604 // Figure out which basic blocks contain back-edges to the loop header.
605 std::vector<BasicBlock*> BackedgeBlocks;
606 for (pred_iterator I = pred_begin(Header), E = pred_end(Header); I != E; ++I)
607 if (*I != Preheader) BackedgeBlocks.push_back(*I);
608
609 // Create and insert the new backedge block...
610 BasicBlock *BEBlock = new BasicBlock(Header->getName()+".backedge", F);
Chris Lattnera2960002003-11-21 16:52:05 +0000611 BranchInst *BETerminator = new BranchInst(Header, BEBlock);
Chris Lattnerc4622a62003-10-13 00:37:13 +0000612
613 // Move the new backedge block to right after the last backedge block.
614 Function::iterator InsertPos = BackedgeBlocks.back(); ++InsertPos;
615 F->getBasicBlockList().splice(InsertPos, F->getBasicBlockList(), BEBlock);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000616
Chris Lattnerc4622a62003-10-13 00:37:13 +0000617 // Now that the block has been inserted into the function, create PHI nodes in
618 // the backedge block which correspond to any PHI nodes in the header block.
Alkis Evlogimenos3ce42ec2004-09-28 02:40:37 +0000619 for (BasicBlock::iterator I = Header->begin(); isa<PHINode>(I); ++I) {
620 PHINode *PN = cast<PHINode>(I);
Chris Lattnerc4622a62003-10-13 00:37:13 +0000621 PHINode *NewPN = new PHINode(PN->getType(), PN->getName()+".be",
622 BETerminator);
Chris Lattnerd8e20182005-01-29 00:39:08 +0000623 NewPN->reserveOperandSpace(BackedgeBlocks.size());
Chris Lattner514e8432005-03-25 06:37:22 +0000624 if (AA) AA->copyValue(PN, NewPN);
Chris Lattnerc4622a62003-10-13 00:37:13 +0000625
626 // Loop over the PHI node, moving all entries except the one for the
627 // preheader over to the new PHI node.
628 unsigned PreheaderIdx = ~0U;
629 bool HasUniqueIncomingValue = true;
630 Value *UniqueValue = 0;
631 for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
632 BasicBlock *IBB = PN->getIncomingBlock(i);
633 Value *IV = PN->getIncomingValue(i);
634 if (IBB == Preheader) {
635 PreheaderIdx = i;
636 } else {
637 NewPN->addIncoming(IV, IBB);
638 if (HasUniqueIncomingValue) {
639 if (UniqueValue == 0)
640 UniqueValue = IV;
641 else if (UniqueValue != IV)
642 HasUniqueIncomingValue = false;
643 }
644 }
645 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000646
Chris Lattnerc4622a62003-10-13 00:37:13 +0000647 // Delete all of the incoming values from the old PN except the preheader's
648 assert(PreheaderIdx != ~0U && "PHI has no preheader entry??");
649 if (PreheaderIdx != 0) {
650 PN->setIncomingValue(0, PN->getIncomingValue(PreheaderIdx));
651 PN->setIncomingBlock(0, PN->getIncomingBlock(PreheaderIdx));
652 }
Chris Lattnerd8e20182005-01-29 00:39:08 +0000653 // Nuke all entries except the zero'th.
654 for (unsigned i = 0, e = PN->getNumIncomingValues()-1; i != e; ++i)
655 PN->removeIncomingValue(e-i, false);
Chris Lattnerc4622a62003-10-13 00:37:13 +0000656
657 // Finally, add the newly constructed PHI node as the entry for the BEBlock.
658 PN->addIncoming(NewPN, BEBlock);
659
660 // As an optimization, if all incoming values in the new PhiNode (which is a
661 // subset of the incoming values of the old PHI node) have the same value,
662 // eliminate the PHI Node.
663 if (HasUniqueIncomingValue) {
664 NewPN->replaceAllUsesWith(UniqueValue);
Chris Lattner514e8432005-03-25 06:37:22 +0000665 if (AA) AA->deleteValue(NewPN);
Chris Lattnerc4622a62003-10-13 00:37:13 +0000666 BEBlock->getInstList().erase(NewPN);
667 }
668 }
669
670 // Now that all of the PHI nodes have been inserted and adjusted, modify the
671 // backedge blocks to just to the BEBlock instead of the header.
672 for (unsigned i = 0, e = BackedgeBlocks.size(); i != e; ++i) {
673 TerminatorInst *TI = BackedgeBlocks[i]->getTerminator();
674 for (unsigned Op = 0, e = TI->getNumSuccessors(); Op != e; ++Op)
675 if (TI->getSuccessor(Op) == Header)
676 TI->setSuccessor(Op, BEBlock);
677 }
678
679 //===--- Update all analyses which we must preserve now -----------------===//
680
681 // Update Loop Information - we know that this block is now in the current
682 // loop and all parent loops.
Chris Lattnercffbbee2006-02-14 22:34:08 +0000683 L->addBasicBlockToLoop(BEBlock, *LI);
Chris Lattnerc4622a62003-10-13 00:37:13 +0000684
Devang Pateld5258a232007-06-21 17:23:45 +0000685 // Update dominator information
686 DT->splitBlock(BEBlock);
687 if (DominanceFrontier *DF = getAnalysisToUpdate<DominanceFrontier>())
688 DF->splitBlock(BEBlock);
Chris Lattner61992f62002-09-26 16:17:31 +0000689}
Brian Gaeke960707c2003-11-11 22:41:34 +0000690
Owen Anderson83efbc82007-04-09 22:25:09 +0000691