LoopSimplify.cpp revision 2ab36d350293c77fc8941ce1023e4899df7e3a82
167a9801bc510ff2c28068361fb30ae397fd1e026Chris Lattner//===- LoopSimplify.cpp - Loop Canonicalization Pass ----------------------===//
2fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman//
3b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//                     The LLVM Compiler Infrastructure
4b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//
54ee451de366474b9c228b4e5fa573795a715216dChris Lattner// This file is distributed under the University of Illinois Open Source
64ee451de366474b9c228b4e5fa573795a715216dChris Lattner// License. See LICENSE.TXT for details.
7fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman//
8b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//===----------------------------------------------------------------------===//
938acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner//
10ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// This pass performs several transformations to transform natural loops into a
11ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// simpler form, which makes subsequent analyses and transformations simpler and
12ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// more effective.
13dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner//
14dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner// Loop pre-header insertion guarantees that there is a single, non-critical
15dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner// entry edge from outside of the loop to the loop header.  This simplifies a
16dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner// number of analyses and transformations, such as LICM.
17dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner//
18dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner// Loop exit-block insertion guarantees that all exit blocks from the loop
19dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner// (blocks which are outside of the loop that have predecessors inside of the
2066ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattner// loop) only have predecessors from inside of the loop (and are thus dominated
2166ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattner// by the loop header).  This simplifies transformations such as store-sinking
2266ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattner// that are built into LICM.
23dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner//
242ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner// This pass also guarantees that loops will have exactly one backedge.
252ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner//
26f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman// Indirectbr instructions introduce several complications. If the loop
27f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman// contains or is entered by an indirectbr instruction, it may not be possible
28f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman// to transform the loop and make these guarantees. Client code should check
29f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman// that these conditions are true before relying on them.
30f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman//
31dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner// Note that the simplifycfg pass will clean up blocks which are split out but
32ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// end up being unnecessary, so usage of this pass should not pessimize
33ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// generated code.
34ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner//
35ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// This pass obviously modifies the CFG, but updates loop information and
36ee2c50cca57272b92cf9e0d1fb238d14d57ea1ddChris Lattner// dominator information.
3738acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner//
3838acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner//===----------------------------------------------------------------------===//
3938acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
40d216e8ba60494caacf919cbf5fef110d48f0d162Chris Lattner#define DEBUG_TYPE "loopsimplify"
4138acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner#include "llvm/Transforms/Scalar.h"
423cb63ddd5183a1469e4557b3e22735ed3ace05b2Chris Lattner#include "llvm/Constants.h"
4347b14a4a6a455c7be169cfd312fcbe796f0ad426Misha Brukman#include "llvm/Instructions.h"
44689fac02268929b756086753b4656d6dabc5cf2dDevang Patel#include "llvm/IntrinsicInst.h"
452ef703ec429900c5b49d94d82332e7a216a2d7c4Chris Lattner#include "llvm/Function.h"
460a205a459884ec745df1c529396dd921f029dafdOwen Anderson#include "llvm/LLVMContext.h"
472ef703ec429900c5b49d94d82332e7a216a2d7c4Chris Lattner#include "llvm/Type.h"
48cec5b8831d4ee3d81990bf1af41ce1d4f4cf9704Chris Lattner#include "llvm/Analysis/AliasAnalysis.h"
49ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman#include "llvm/Analysis/ScalarEvolution.h"
500f98e75adff9024dcfe1d2afbfa83625d60ebaa8Chris Lattner#include "llvm/Analysis/Dominators.h"
51d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman#include "llvm/Analysis/LoopPass.h"
5254b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner#include "llvm/Transforms/Utils/BasicBlockUtils.h"
534b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman#include "llvm/Transforms/Utils/Local.h"
5438acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner#include "llvm/Support/CFG.h"
55c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman#include "llvm/Support/Debug.h"
56551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/ADT/SetOperations.h"
57551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/ADT/SetVector.h"
58551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/ADT/Statistic.h"
59551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/ADT/DepthFirstIterator.h"
6066ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattnerusing namespace llvm;
61d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke
62d216e8ba60494caacf919cbf5fef110d48f0d162Chris LattnerSTATISTIC(NumInserted, "Number of pre-header or exit blocks inserted");
63d216e8ba60494caacf919cbf5fef110d48f0d162Chris LattnerSTATISTIC(NumNested  , "Number of nested loops split out");
6438acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
65d216e8ba60494caacf919cbf5fef110d48f0d162Chris Lattnernamespace {
666726b6d75a8b679068a58cb954ba97cf9d1690baNick Lewycky  struct LoopSimplify : public LoopPass {
67ecd94c804a563f2a86572dcf1d2e81f397e19daaNick Lewycky    static char ID; // Pass identification, replacement for typeid
6890c579de5a383cee278acc3f7e7b9d0a656e6a35Owen Anderson    LoopSimplify() : LoopPass(ID) {}
69794fd75c67a2cdc128d67342c6d88a504d186896Devang Patel
70cec5b8831d4ee3d81990bf1af41ce1d4f4cf9704Chris Lattner    // AA - If we have an alias analysis object to update, this is it, otherwise
71cec5b8831d4ee3d81990bf1af41ce1d4f4cf9704Chris Lattner    // this is null.
72cec5b8831d4ee3d81990bf1af41ce1d4f4cf9704Chris Lattner    AliasAnalysis *AA;
73c27e056d4fd7f6ecdd8e40eb92230be380c5c8c9Chris Lattner    LoopInfo *LI;
740e7f728ad1ac25b0ed450fe0f8b86a38d3c2a93aDevang Patel    DominatorTree *DT;
75ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman    ScalarEvolution *SE;
76d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    Loop *L;
77d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    virtual bool runOnLoop(Loop *L, LPPassManager &LPM);
78fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
7938acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
8038acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner      // We need loop information to identify the loops...
81052f0001588a1613f845c84c04b38ced28ad6711Dan Gohman      AU.addRequired<DominatorTree>();
8238acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner      AU.addPreserved<DominatorTree>();
831e381fcd553a3955a10338fd305efc023d7d22e1Dan Gohman
84052f0001588a1613f845c84c04b38ced28ad6711Dan Gohman      AU.addRequired<LoopInfo>();
851e381fcd553a3955a10338fd305efc023d7d22e1Dan Gohman      AU.addPreserved<LoopInfo>();
861e381fcd553a3955a10338fd305efc023d7d22e1Dan Gohman
874c37c07ee3bfacaaf90ea57165ef6855b4ed8b22Devang Patel      AU.addPreserved<AliasAnalysis>();
88ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman      AU.addPreserved<ScalarEvolution>();
8994f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner      AU.addPreservedID(BreakCriticalEdgesID);  // No critical edges added.
90052f0001588a1613f845c84c04b38ced28ad6711Dan Gohman      AU.addPreserved<DominanceFrontier>();
91052f0001588a1613f845c84c04b38ced28ad6711Dan Gohman      AU.addPreservedID(LCSSAID);
9238acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner    }
9358e0ef1e90c3f6dbae213612b44e56f7d6d65ea7Devang Patel
94f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    /// verifyAnalysis() - Verify LoopSimplifyForm's guarantees.
95f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    void verifyAnalysis() const;
9658e0ef1e90c3f6dbae213612b44e56f7d6d65ea7Devang Patel
9738acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  private:
98d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    bool ProcessLoop(Loop *L, LPPassManager &LPM);
9959fb87d469b9b38b0f4c1e31a2f34fa8f09b981dChris Lattner    BasicBlock *RewriteLoopExitBlock(Loop *L, BasicBlock *Exit);
1000df6e09d43d6d733555a10d22572ddb0006e7d23Dan Gohman    BasicBlock *InsertPreheaderForLoop(Loop *L);
101d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    Loop *SeparateNestedLoop(Loop *L, LPPassManager &LPM);
102f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    BasicBlock *InsertUniqueBackedgeBlock(Loop *L, BasicBlock *Preheader);
103120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner    void PlaceSplitBlockCarefully(BasicBlock *NewBB,
10454b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                                  SmallVectorImpl<BasicBlock*> &SplitPreds,
105120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner                                  Loop *L);
10638acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  };
10738acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner}
10838acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
109844731a7f1909f55935e3514c9e713a62d67662eDan Gohmanchar LoopSimplify::ID = 0;
1102ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_BEGIN(LoopSimplify, "loopsimplify",
1112ab36d350293c77fc8941ce1023e4899df7e3a82Owen Anderson                "Canonicalize natural loops", true, false)
1122ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_DEPENDENCY(DominatorTree)
1132ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_DEPENDENCY(LoopInfo)
1142ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_DEPENDENCY(ScalarEvolution)
1152ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_DEPENDENCY(BreakCriticalEdges)
1162ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_DEPENDENCY(DominanceFrontier)
1172ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_DEPENDENCY(LCSSA)
1182ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_AG_DEPENDENCY(AliasAnalysis)
1192ab36d350293c77fc8941ce1023e4899df7e3a82Owen AndersonINITIALIZE_PASS_END(LoopSimplify, "loopsimplify",
120ce665bd2e2b581ab0858d1afe359192bac96b868Owen Anderson                "Canonicalize natural loops", true, false)
121844731a7f1909f55935e3514c9e713a62d67662eDan Gohman
12238acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner// Publically exposed interface to pass...
12390c579de5a383cee278acc3f7e7b9d0a656e6a35Owen Andersonchar &llvm::LoopSimplifyID = LoopSimplify::ID;
124d84db1133345234738b646c70b907bf8a0983ac9Dan GohmanPass *llvm::createLoopSimplifyPass() { return new LoopSimplify(); }
12538acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
12634d2b90d09226ebf6189775acfd2801e127b10ecDan Gohman/// runOnLoop - Run down all loops in the CFG (recursively, but we could do
12738acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner/// it in any convenient order) inserting preheaders...
12838acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner///
129d84db1133345234738b646c70b907bf8a0983ac9Dan Gohmanbool LoopSimplify::runOnLoop(Loop *l, LPPassManager &LPM) {
130d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  L = l;
13138acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  bool Changed = false;
132c27e056d4fd7f6ecdd8e40eb92230be380c5c8c9Chris Lattner  LI = &getAnalysis<LoopInfo>();
1331465d61bdd36cfd6021036a527895f0dd358e97dDuncan Sands  AA = getAnalysisIfAvailable<AliasAnalysis>();
1340e7f728ad1ac25b0ed450fe0f8b86a38d3c2a93aDevang Patel  DT = &getAnalysis<DominatorTree>();
135ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman  SE = getAnalysisIfAvailable<ScalarEvolution>();
13638acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
137d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  Changed |= ProcessLoop(L, LPM);
13838acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
13938acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  return Changed;
14038acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner}
14138acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
14238acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner/// ProcessLoop - Walk the loop structure in depth first order, ensuring that
14338acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner/// all loops have preheaders.
14438acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner///
145d84db1133345234738b646c70b907bf8a0983ac9Dan Gohmanbool LoopSimplify::ProcessLoop(Loop *L, LPPassManager &LPM) {
14638acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  bool Changed = false;
1473bb4657488f700bbe3376fb547017163b8fbbd8fChris LattnerReprocessLoop:
148d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman
1492d0a91cd6c3df32014d547255d6a615bd1bc84fbDan Gohman  // Check to see that no blocks (other than the header) in this loop have
150d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  // predecessors that are not in the loop.  This is not valid for natural
151d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  // loops, but can occur if the blocks are unreachable.  Since they are
152d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  // unreachable we can just shamelessly delete those CFG edges!
153d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  for (Loop::block_iterator BB = L->block_begin(), E = L->block_end();
154d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman       BB != E; ++BB) {
155d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    if (*BB == L->getHeader()) continue;
156d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman
157481c4c07347c40fa666d09f3b31fbe2ca27e2d52Gabor Greif    SmallPtrSet<BasicBlock*, 4> BadPreds;
158481c4c07347c40fa666d09f3b31fbe2ca27e2d52Gabor Greif    for (pred_iterator PI = pred_begin(*BB),
159481c4c07347c40fa666d09f3b31fbe2ca27e2d52Gabor Greif         PE = pred_end(*BB); PI != PE; ++PI) {
1609672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      BasicBlock *P = *PI;
1619672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      if (!L->contains(P))
1629672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif        BadPreds.insert(P);
1639672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    }
164d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman
165d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    // Delete each unique out-of-loop (and thus dead) predecessor.
166481c4c07347c40fa666d09f3b31fbe2ca27e2d52Gabor Greif    for (SmallPtrSet<BasicBlock*, 4>::iterator I = BadPreds.begin(),
167d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman         E = BadPreds.end(); I != E; ++I) {
168c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
169c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman      DEBUG(dbgs() << "LoopSimplify: Deleting edge from dead predecessor ";
170c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman            WriteAsOperand(dbgs(), *I, false);
171c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman            dbgs() << "\n");
172c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
173d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      // Inform each successor of each dead pred.
174d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      for (succ_iterator SI = succ_begin(*I), SE = succ_end(*I); SI != SE; ++SI)
175d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman        (*SI)->removePredecessor(*I);
176d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      // Zap the dead pred's terminator and replace it with unreachable.
177d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      TerminatorInst *TI = (*I)->getTerminator();
178d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman       TI->replaceAllUsesWith(UndefValue::get(TI->getType()));
179d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      (*I)->getTerminator()->eraseFromParent();
180d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      new UnreachableInst((*I)->getContext(), *I);
181d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      Changed = true;
182d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman    }
183d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  }
1842ef703ec429900c5b49d94d82332e7a216a2d7c4Chris Lattner
18585669637139089eaed8def1583ac04266c9654e2Dan Gohman  // If there are exiting blocks with branches on undef, resolve the undef in
18685669637139089eaed8def1583ac04266c9654e2Dan Gohman  // the direction which will exit the loop. This will help simplify loop
18785669637139089eaed8def1583ac04266c9654e2Dan Gohman  // trip count computations.
18885669637139089eaed8def1583ac04266c9654e2Dan Gohman  SmallVector<BasicBlock*, 8> ExitingBlocks;
18985669637139089eaed8def1583ac04266c9654e2Dan Gohman  L->getExitingBlocks(ExitingBlocks);
19085669637139089eaed8def1583ac04266c9654e2Dan Gohman  for (SmallVectorImpl<BasicBlock *>::iterator I = ExitingBlocks.begin(),
19185669637139089eaed8def1583ac04266c9654e2Dan Gohman       E = ExitingBlocks.end(); I != E; ++I)
19285669637139089eaed8def1583ac04266c9654e2Dan Gohman    if (BranchInst *BI = dyn_cast<BranchInst>((*I)->getTerminator()))
19385669637139089eaed8def1583ac04266c9654e2Dan Gohman      if (BI->isConditional()) {
19485669637139089eaed8def1583ac04266c9654e2Dan Gohman        if (UndefValue *Cond = dyn_cast<UndefValue>(BI->getCondition())) {
195c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
196c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman          DEBUG(dbgs() << "LoopSimplify: Resolving \"br i1 undef\" to exit in ";
197c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman                WriteAsOperand(dbgs(), *I, false);
198c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman                dbgs() << "\n");
199c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
20085669637139089eaed8def1583ac04266c9654e2Dan Gohman          BI->setCondition(ConstantInt::get(Cond->getType(),
20185669637139089eaed8def1583ac04266c9654e2Dan Gohman                                            !L->contains(BI->getSuccessor(0))));
20285669637139089eaed8def1583ac04266c9654e2Dan Gohman          Changed = true;
20385669637139089eaed8def1583ac04266c9654e2Dan Gohman        }
20485669637139089eaed8def1583ac04266c9654e2Dan Gohman      }
20585669637139089eaed8def1583ac04266c9654e2Dan Gohman
206fa78946482a2cc73a1485887dfd12edd12b742a4Chris Lattner  // Does the loop already have a preheader?  If so, don't insert one.
2070df6e09d43d6d733555a10d22572ddb0006e7d23Dan Gohman  BasicBlock *Preheader = L->getLoopPreheader();
2080df6e09d43d6d733555a10d22572ddb0006e7d23Dan Gohman  if (!Preheader) {
2090df6e09d43d6d733555a10d22572ddb0006e7d23Dan Gohman    Preheader = InsertPreheaderForLoop(L);
210f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    if (Preheader) {
211fe60104ac97f3a8736dcfbfdf9547c7b7cc7b951Dan Gohman      ++NumInserted;
212f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      Changed = true;
213f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    }
21438acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  }
21538acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
21666ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattner  // Next, check to make sure that all exit nodes of the loop only have
21766ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattner  // predecessors that are inside of the loop.  This check guarantees that the
21866ea98e85c5f9c03aac139563d7874e93dc345c6Chris Lattner  // loop preheader/header will dominate the exit blocks.  If the exit block has
219ee628cfefb93e0261ee3e56686d3fffa4e81f371Chris Lattner  // predecessors from outside of the loop, split the edge now.
220b7211a2ce13a0365e0e1dd2f27adda2ee3d1288bDevang Patel  SmallVector<BasicBlock*, 8> ExitBlocks;
221ee628cfefb93e0261ee3e56686d3fffa4e81f371Chris Lattner  L->getExitBlocks(ExitBlocks);
222c27e056d4fd7f6ecdd8e40eb92230be380c5c8c9Chris Lattner
22317146baef5b79114f05e0f99fcba389f2764b65dDan Gohman  SmallSetVector<BasicBlock *, 8> ExitBlockSet(ExitBlocks.begin(),
22417146baef5b79114f05e0f99fcba389f2764b65dDan Gohman                                               ExitBlocks.end());
22517146baef5b79114f05e0f99fcba389f2764b65dDan Gohman  for (SmallSetVector<BasicBlock *, 8>::iterator I = ExitBlockSet.begin(),
226fed22aac4337c589841c443be70fe05559693f6aChris Lattner         E = ExitBlockSet.end(); I != E; ++I) {
227fed22aac4337c589841c443be70fe05559693f6aChris Lattner    BasicBlock *ExitBlock = *I;
228de7aee760e77d49877ec308bc47dc455b2b754afChris Lattner    for (pred_iterator PI = pred_begin(ExitBlock), PE = pred_end(ExitBlock);
229de7aee760e77d49877ec308bc47dc455b2b754afChris Lattner         PI != PE; ++PI)
2308587eb3a51117b630c18236cc53eb865e76faf2dChris Lattner      // Must be exactly this loop: no subloops, parent loops, or non-loop preds
2318587eb3a51117b630c18236cc53eb865e76faf2dChris Lattner      // allowed.
232ee628cfefb93e0261ee3e56686d3fffa4e81f371Chris Lattner      if (!L->contains(*PI)) {
233f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        if (RewriteLoopExitBlock(L, ExitBlock)) {
234fe60104ac97f3a8736dcfbfdf9547c7b7cc7b951Dan Gohman          ++NumInserted;
235f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman          Changed = true;
236f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        }
237de7aee760e77d49877ec308bc47dc455b2b754afChris Lattner        break;
238de7aee760e77d49877ec308bc47dc455b2b754afChris Lattner      }
239fed22aac4337c589841c443be70fe05559693f6aChris Lattner  }
240dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner
241529b28da455a703d226a31a03400e6662ff569feChris Lattner  // If the header has more than two predecessors at this point (from the
242529b28da455a703d226a31a03400e6662ff569feChris Lattner  // preheader and from multiple backedges), we must adjust the loop.
243f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  BasicBlock *LoopLatch = L->getLoopLatch();
244f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  if (!LoopLatch) {
2453bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner    // If this is really a nested loop, rip it out into a child loop.  Don't do
2463bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner    // this for loops with a giant number of backedges, just factor them into a
2473bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner    // common backedge instead.
248f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    if (L->getNumBackEdges() < 8) {
249d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman      if (SeparateNestedLoop(L, LPM)) {
2503bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner        ++NumNested;
2513bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner        // This is a big restructuring change, reprocess the whole loop.
2523bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner        Changed = true;
2533bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner        // GCC doesn't tail recursion eliminate this.
2543bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner        goto ReprocessLoop;
2553bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner      }
256529b28da455a703d226a31a03400e6662ff569feChris Lattner    }
257529b28da455a703d226a31a03400e6662ff569feChris Lattner
2583bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner    // If we either couldn't, or didn't want to, identify nesting of the loops,
2593bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner    // insert a new block that all backedges target, then make it jump to the
2603bb4657488f700bbe3376fb547017163b8fbbd8fChris Lattner    // loop header.
261f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    LoopLatch = InsertUniqueBackedgeBlock(L, Preheader);
262f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    if (LoopLatch) {
263fe60104ac97f3a8736dcfbfdf9547c7b7cc7b951Dan Gohman      ++NumInserted;
264f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      Changed = true;
265f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    }
2662ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  }
2672ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
26894f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner  // Scan over the PHI nodes in the loop header.  Since they now have only two
26994f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner  // incoming values (the loop is canonicalized), we may have simplified the PHI
27094f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner  // down to 'X = phi [X, Y]', which should be replaced with 'Y'.
27194f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner  PHINode *PN;
27294f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner  for (BasicBlock::iterator I = L->getHeader()->begin();
27394f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner       (PN = dyn_cast<PHINode>(I++)); )
274bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman    if (Value *V = PN->hasConstantValue(DT)) {
2754c37c07ee3bfacaaf90ea57165ef6855b4ed8b22Devang Patel      if (AA) AA->deleteValue(PN);
2764c37c07ee3bfacaaf90ea57165ef6855b4ed8b22Devang Patel      PN->replaceAllUsesWith(V);
2774c37c07ee3bfacaaf90ea57165ef6855b4ed8b22Devang Patel      PN->eraseFromParent();
2784c37c07ee3bfacaaf90ea57165ef6855b4ed8b22Devang Patel    }
27994f40324481c04ae8718967b4b5a3d7ca22370e6Chris Lattner
2805cd8770412f98f6e6416c439e01222b3643b9e22Bob Wilson  // If this loop has multiple exits and the exits all go to the same
2814b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // block, attempt to merge the exits. This helps several passes, such
2824b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // as LoopRotation, which do not support loops with multiple exits.
2834b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // SimplifyCFG also does this (and this code uses the same utility
2844b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // function), however this code is loop-aware, where SimplifyCFG is
2854b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // not. That gives it the advantage of being able to hoist
2864b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // loop-invariant instructions out of the way to open up more
2874b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // opportunities, and the disadvantage of having the responsibility
2884b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  // to preserve dominator information.
289b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman  bool UniqueExit = true;
290b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman  if (!ExitBlocks.empty())
291b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman    for (unsigned i = 1, e = ExitBlocks.size(); i != e; ++i)
292b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman      if (ExitBlocks[i] != ExitBlocks[0]) {
293b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman        UniqueExit = false;
294b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman        break;
295b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman      }
296b1dc915a8d4971880a016e678ccf563d1a03a916Dan Gohman  if (UniqueExit) {
2974b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman    for (unsigned i = 0, e = ExitingBlocks.size(); i != e; ++i) {
2984b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      BasicBlock *ExitingBlock = ExitingBlocks[i];
2994b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      if (!ExitingBlock->getSinglePredecessor()) continue;
3004b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      BranchInst *BI = dyn_cast<BranchInst>(ExitingBlock->getTerminator());
3014b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      if (!BI || !BI->isConditional()) continue;
3024b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      CmpInst *CI = dyn_cast<CmpInst>(BI->getCondition());
3034b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      if (!CI || CI->getParent() != ExitingBlock) continue;
3044b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman
3054b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // Attempt to hoist out all instructions except for the
3064b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // comparison and the branch.
3074b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      bool AllInvariant = true;
3082aa93efa0c983449e5464165e80ebd9c0fb5f6c1Dan Gohman      for (BasicBlock::iterator I = ExitingBlock->begin(); &*I != BI; ) {
3094b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman        Instruction *Inst = I++;
310689fac02268929b756086753b4656d6dabc5cf2dDevang Patel        // Skip debug info intrinsics.
311689fac02268929b756086753b4656d6dabc5cf2dDevang Patel        if (isa<DbgInfoIntrinsic>(Inst))
312689fac02268929b756086753b4656d6dabc5cf2dDevang Patel          continue;
3132aa93efa0c983449e5464165e80ebd9c0fb5f6c1Dan Gohman        if (Inst == CI)
3144b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman          continue;
315f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        if (!L->makeLoopInvariant(Inst, Changed,
316f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman                                  Preheader ? Preheader->getTerminator() : 0)) {
3174b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman          AllInvariant = false;
3184b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman          break;
3194b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman        }
3204b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      }
3214b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      if (!AllInvariant) continue;
3224b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman
3234b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // The block has now been cleared of all instructions except for
3244b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // a comparison and a conditional branch. SimplifyCFG may be able
3254b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // to fold it now.
3264b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      if (!FoldBranchToCommonDest(BI)) continue;
3274b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman
3284b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // Success. The block is now dead, so remove it from the loop,
3294b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      // update the dominator tree and dominance frontier, and delete it.
330c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
331c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman      DEBUG(dbgs() << "LoopSimplify: Eliminating exiting block ";
332c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman            WriteAsOperand(dbgs(), ExitingBlock, false);
333c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman            dbgs() << "\n");
334c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
3354b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      assert(pred_begin(ExitingBlock) == pred_end(ExitingBlock));
3364b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      Changed = true;
337a1baee20c4b042eca1f182fc003f38ab52efc7a9Dan Gohman      LI->removeBlock(ExitingBlock);
3384b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman
3394b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      DominanceFrontier *DF = getAnalysisIfAvailable<DominanceFrontier>();
3404b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      DomTreeNode *Node = DT->getNode(ExitingBlock);
3414b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      const std::vector<DomTreeNodeBase<BasicBlock> *> &Children =
3424b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman        Node->getChildren();
3435184635eda68a0cdcd39c958ccc11ba1843bcc7bDan Gohman      while (!Children.empty()) {
3445184635eda68a0cdcd39c958ccc11ba1843bcc7bDan Gohman        DomTreeNode *Child = Children.front();
3455184635eda68a0cdcd39c958ccc11ba1843bcc7bDan Gohman        DT->changeImmediateDominator(Child, Node->getIDom());
3465184635eda68a0cdcd39c958ccc11ba1843bcc7bDan Gohman        if (DF) DF->changeImmediateDominator(Child->getBlock(),
3474b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman                                             Node->getIDom()->getBlock(),
3484b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman                                             DT);
3494b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      }
3504b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      DT->eraseNode(ExitingBlock);
3514b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      if (DF) DF->removeBlock(ExitingBlock);
3524b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman
3534b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      BI->getSuccessor(0)->removePredecessor(ExitingBlock);
3544b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      BI->getSuccessor(1)->removePredecessor(ExitingBlock);
3554b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman      ExitingBlock->eraseFromParent();
3564b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman    }
3574b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman  }
3584b35f83b91a1a313f0730c600e5178aaf7df98d6Dan Gohman
35938acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner  return Changed;
36038acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner}
36138acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner
362dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner/// InsertPreheaderForLoop - Once we discover that a loop doesn't have a
363dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner/// preheader, this method is called to insert one.  This method has two phases:
364dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner/// preheader insertion and analysis updating.
365dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner///
3660df6e09d43d6d733555a10d22572ddb0006e7d23Dan GohmanBasicBlock *LoopSimplify::InsertPreheaderForLoop(Loop *L) {
367dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner  BasicBlock *Header = L->getHeader();
368dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner
369dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner  // Compute the set of predecessors of the loop that are not in the loop.
37054b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner  SmallVector<BasicBlock*, 8> OutsideBlocks;
371dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner  for (pred_iterator PI = pred_begin(Header), PE = pred_end(Header);
3729672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif       PI != PE; ++PI) {
3739672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    BasicBlock *P = *PI;
3749672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    if (!L->contains(P)) {         // Coming in from outside the loop?
375f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      // If the loop is branched to from an indirect branch, we won't
376f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      // be able to fully transform the loop, because it prohibits
377f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      // edge splitting.
3789672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      if (isa<IndirectBrInst>(P->getTerminator())) return 0;
379f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
380f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      // Keep track of it.
3819672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      OutsideBlocks.push_back(P);
382f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    }
3839672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif  }
384fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
385c3984578bed8236f35825ca8aa30b3ed6cff60d5Chris Lattner  // Split out the loop pre-header.
386dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner  BasicBlock *NewBB =
38754b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner    SplitBlockPredecessors(Header, &OutsideBlocks[0], OutsideBlocks.size(),
38854b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                           ".preheader", this);
3899f879cfb0a93bf34818fb68e1dc209d47a7d24f3Chris Lattner
390c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman  DEBUG(dbgs() << "LoopSimplify: Creating pre-header ";
391c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman        WriteAsOperand(dbgs(), NewBB, false);
392c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman        dbgs() << "\n");
393c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
394120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // Make sure that NewBB is put someplace intelligent, which doesn't mess up
395120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // code layout too horribly.
396120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  PlaceSplitBlockCarefully(NewBB, OutsideBlocks, L);
3970df6e09d43d6d733555a10d22572ddb0006e7d23Dan Gohman
3980df6e09d43d6d733555a10d22572ddb0006e7d23Dan Gohman  return NewBB;
399dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner}
400dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner
401529b28da455a703d226a31a03400e6662ff569feChris Lattner/// RewriteLoopExitBlock - Ensure that the loop preheader dominates all exit
402529b28da455a703d226a31a03400e6662ff569feChris Lattner/// blocks.  This method is used to split exit blocks that have predecessors
403529b28da455a703d226a31a03400e6662ff569feChris Lattner/// outside of the loop.
40459fb87d469b9b38b0f4c1e31a2f34fa8f09b981dChris LattnerBasicBlock *LoopSimplify::RewriteLoopExitBlock(Loop *L, BasicBlock *Exit) {
40554b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner  SmallVector<BasicBlock*, 8> LoopBlocks;
4069672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif  for (pred_iterator I = pred_begin(Exit), E = pred_end(Exit); I != E; ++I) {
4079672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    BasicBlock *P = *I;
4089672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    if (L->contains(P)) {
409f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      // Don't do this if the loop is exited via an indirect branch.
4109672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      if (isa<IndirectBrInst>(P->getTerminator())) return 0;
411f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
4129672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      LoopBlocks.push_back(P);
413f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    }
4149672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif  }
415dbf3cd7952736b649b4d19badb73ec6c1f9be583Chris Lattner
4167e7ad49c23867a5de8e15adfd946fdfa4ba68902Chris Lattner  assert(!LoopBlocks.empty() && "No edges coming in from outside the loop?");
41754b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner  BasicBlock *NewBB = SplitBlockPredecessors(Exit, &LoopBlocks[0],
41854b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                                             LoopBlocks.size(), ".loopexit",
41954b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                                             this);
4207e7ad49c23867a5de8e15adfd946fdfa4ba68902Chris Lattner
421c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman  DEBUG(dbgs() << "LoopSimplify: Creating dedicated exit block ";
422c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman        WriteAsOperand(dbgs(), NewBB, false);
423c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman        dbgs() << "\n");
424c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
42559fb87d469b9b38b0f4c1e31a2f34fa8f09b981dChris Lattner  return NewBB;
4262ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner}
4272ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
428529b28da455a703d226a31a03400e6662ff569feChris Lattner/// AddBlockAndPredsToSet - Add the specified block, and all of its
429529b28da455a703d226a31a03400e6662ff569feChris Lattner/// predecessors, to the specified set, if it's not already in there.  Stop
430529b28da455a703d226a31a03400e6662ff569feChris Lattner/// predecessor traversal when we reach StopBlock.
43158d7fbf250659246fcca9417a91170a681b1850aDevang Patelstatic void AddBlockAndPredsToSet(BasicBlock *InputBB, BasicBlock *StopBlock,
432529b28da455a703d226a31a03400e6662ff569feChris Lattner                                  std::set<BasicBlock*> &Blocks) {
43358d7fbf250659246fcca9417a91170a681b1850aDevang Patel  std::vector<BasicBlock *> WorkList;
43458d7fbf250659246fcca9417a91170a681b1850aDevang Patel  WorkList.push_back(InputBB);
43558d7fbf250659246fcca9417a91170a681b1850aDevang Patel  do {
43658d7fbf250659246fcca9417a91170a681b1850aDevang Patel    BasicBlock *BB = WorkList.back(); WorkList.pop_back();
43758d7fbf250659246fcca9417a91170a681b1850aDevang Patel    if (Blocks.insert(BB).second && BB != StopBlock)
43858d7fbf250659246fcca9417a91170a681b1850aDevang Patel      // If BB is not already processed and it is not a stop block then
43958d7fbf250659246fcca9417a91170a681b1850aDevang Patel      // insert its predecessor in the work list
44058d7fbf250659246fcca9417a91170a681b1850aDevang Patel      for (pred_iterator I = pred_begin(BB), E = pred_end(BB); I != E; ++I) {
44158d7fbf250659246fcca9417a91170a681b1850aDevang Patel        BasicBlock *WBB = *I;
44258d7fbf250659246fcca9417a91170a681b1850aDevang Patel        WorkList.push_back(WBB);
44358d7fbf250659246fcca9417a91170a681b1850aDevang Patel      }
44458d7fbf250659246fcca9417a91170a681b1850aDevang Patel  } while(!WorkList.empty());
445529b28da455a703d226a31a03400e6662ff569feChris Lattner}
446529b28da455a703d226a31a03400e6662ff569feChris Lattner
4471f62f82b05563df9c83094608de24ea581014d1eChris Lattner/// FindPHIToPartitionLoops - The first part of loop-nestification is to find a
4481f62f82b05563df9c83094608de24ea581014d1eChris Lattner/// PHI node that tells us how to partition the loops.
449dba2413b2ecf4e781f457036a2eb0f103192e90dDevang Patelstatic PHINode *FindPHIToPartitionLoops(Loop *L, DominatorTree *DT,
450ad190145912facc6fbf2fbe58023bb238fbf2365Owen Anderson                                        AliasAnalysis *AA) {
451200a360ec66b4d016c17d6f8e3ea559b1fd07205Alkis Evlogimenos  for (BasicBlock::iterator I = L->getHeader()->begin(); isa<PHINode>(I); ) {
452200a360ec66b4d016c17d6f8e3ea559b1fd07205Alkis Evlogimenos    PHINode *PN = cast<PHINode>(I);
4531f62f82b05563df9c83094608de24ea581014d1eChris Lattner    ++I;
454bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman    if (Value *V = PN->hasConstantValue(DT)) {
455bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman      // This is a degenerate PHI already, don't modify it!
456bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman      PN->replaceAllUsesWith(V);
457bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman      if (AA) AA->deleteValue(PN);
458bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman      PN->eraseFromParent();
459bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman      continue;
460bccfc24c4e8092e1ee18746dd4cee01247728faaDan Gohman    }
461c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner
462c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner    // Scan this PHI node looking for a use of the PHI node by itself.
463c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner    for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i)
464c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner      if (PN->getIncomingValue(i) == PN &&
465c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner          L->contains(PN->getIncomingBlock(i)))
466c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner        // We found something tasty to remove.
467c30bda7540de573c887e00bb76ac78d85f56acd4Chris Lattner        return PN;
4681f62f82b05563df9c83094608de24ea581014d1eChris Lattner  }
4691f62f82b05563df9c83094608de24ea581014d1eChris Lattner  return 0;
4701f62f82b05563df9c83094608de24ea581014d1eChris Lattner}
4711f62f82b05563df9c83094608de24ea581014d1eChris Lattner
472120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner// PlaceSplitBlockCarefully - If the block isn't already, move the new block to
473120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner// right after some 'outside block' block.  This prevents the preheader from
474120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner// being placed inside the loop body, e.g. when the loop hasn't been rotated.
475120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattnervoid LoopSimplify::PlaceSplitBlockCarefully(BasicBlock *NewBB,
47654b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                                       SmallVectorImpl<BasicBlock*> &SplitPreds,
477120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner                                            Loop *L) {
478120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // Check to see if NewBB is already well placed.
479120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  Function::iterator BBI = NewBB; --BBI;
480120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  for (unsigned i = 0, e = SplitPreds.size(); i != e; ++i) {
481120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner    if (&*BBI == SplitPreds[i])
482120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner      return;
483120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  }
484120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner
485120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // If it isn't already after an outside block, move it after one.  This is
486120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // always good as it makes the uncond branch from the outside block into a
487120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // fall-through.
488120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner
489120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // Figure out *which* outside block to put this after.  Prefer an outside
490120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // block that neighbors a BB actually in the loop.
491120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  BasicBlock *FoundBB = 0;
492120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  for (unsigned i = 0, e = SplitPreds.size(); i != e; ++i) {
493120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner    Function::iterator BBI = SplitPreds[i];
494120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner    if (++BBI != NewBB->getParent()->end() &&
495120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner        L->contains(BBI)) {
496120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner      FoundBB = SplitPreds[i];
497120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner      break;
498120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner    }
499120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  }
500120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner
501120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // If our heuristic for a *good* bb to place this after doesn't find
502120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // anything, just pick something.  It's likely better than leaving it within
503120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // the loop.
504120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  if (!FoundBB)
505120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner    FoundBB = SplitPreds[0];
506120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  NewBB->moveAfter(FoundBB);
507120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner}
508120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner
509120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner
510529b28da455a703d226a31a03400e6662ff569feChris Lattner/// SeparateNestedLoop - If this loop has multiple backedges, try to pull one of
511529b28da455a703d226a31a03400e6662ff569feChris Lattner/// them out into a nested loop.  This is important for code that looks like
512529b28da455a703d226a31a03400e6662ff569feChris Lattner/// this:
513529b28da455a703d226a31a03400e6662ff569feChris Lattner///
514529b28da455a703d226a31a03400e6662ff569feChris Lattner///  Loop:
515529b28da455a703d226a31a03400e6662ff569feChris Lattner///     ...
516529b28da455a703d226a31a03400e6662ff569feChris Lattner///     br cond, Loop, Next
517529b28da455a703d226a31a03400e6662ff569feChris Lattner///     ...
518529b28da455a703d226a31a03400e6662ff569feChris Lattner///     br cond2, Loop, Out
519529b28da455a703d226a31a03400e6662ff569feChris Lattner///
520529b28da455a703d226a31a03400e6662ff569feChris Lattner/// To identify this common case, we look at the PHI nodes in the header of the
521529b28da455a703d226a31a03400e6662ff569feChris Lattner/// loop.  PHI nodes with unchanging values on one backedge correspond to values
522529b28da455a703d226a31a03400e6662ff569feChris Lattner/// that change in the "outer" loop, but not in the "inner" loop.
523529b28da455a703d226a31a03400e6662ff569feChris Lattner///
524529b28da455a703d226a31a03400e6662ff569feChris Lattner/// If we are able to separate out a loop, return the new outer loop that was
525529b28da455a703d226a31a03400e6662ff569feChris Lattner/// created.
526529b28da455a703d226a31a03400e6662ff569feChris Lattner///
527d84db1133345234738b646c70b907bf8a0983ac9Dan GohmanLoop *LoopSimplify::SeparateNestedLoop(Loop *L, LPPassManager &LPM) {
528dba2413b2ecf4e781f457036a2eb0f103192e90dDevang Patel  PHINode *PN = FindPHIToPartitionLoops(L, DT, AA);
5291f62f82b05563df9c83094608de24ea581014d1eChris Lattner  if (PN == 0) return 0;  // No known way to partition.
530529b28da455a703d226a31a03400e6662ff569feChris Lattner
5311f62f82b05563df9c83094608de24ea581014d1eChris Lattner  // Pull out all predecessors that have varying values in the loop.  This
5321f62f82b05563df9c83094608de24ea581014d1eChris Lattner  // handles the case when a PHI node has multiple instances of itself as
5331f62f82b05563df9c83094608de24ea581014d1eChris Lattner  // arguments.
53454b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner  SmallVector<BasicBlock*, 8> OuterLoopPreds;
5351f62f82b05563df9c83094608de24ea581014d1eChris Lattner  for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i)
5361f62f82b05563df9c83094608de24ea581014d1eChris Lattner    if (PN->getIncomingValue(i) != PN ||
537a58a04921deba911d6ead8d24f495cec234681c1Dan Gohman        !L->contains(PN->getIncomingBlock(i))) {
538a58a04921deba911d6ead8d24f495cec234681c1Dan Gohman      // We can't split indirectbr edges.
539a58a04921deba911d6ead8d24f495cec234681c1Dan Gohman      if (isa<IndirectBrInst>(PN->getIncomingBlock(i)->getTerminator()))
540a58a04921deba911d6ead8d24f495cec234681c1Dan Gohman        return 0;
541a58a04921deba911d6ead8d24f495cec234681c1Dan Gohman
5421f62f82b05563df9c83094608de24ea581014d1eChris Lattner      OuterLoopPreds.push_back(PN->getIncomingBlock(i));
543a58a04921deba911d6ead8d24f495cec234681c1Dan Gohman    }
544529b28da455a703d226a31a03400e6662ff569feChris Lattner
545c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman  DEBUG(dbgs() << "LoopSimplify: Splitting out a new outer loop\n");
546c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
547ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman  // If ScalarEvolution is around and knows anything about values in
548ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman  // this loop, tell it to forget them, because we're about to
549ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman  // substantially change it.
550ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman  if (SE)
551ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman    SE->forgetLoop(L);
552ffa75cdcf82ef2034249a313b9276eaa1bee6c43Dan Gohman
5534b66242c5498a99ed754f698d779243dd1e291e2Chris Lattner  BasicBlock *Header = L->getHeader();
55454b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner  BasicBlock *NewBB = SplitBlockPredecessors(Header, &OuterLoopPreds[0],
55554b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                                             OuterLoopPreds.size(),
55654b9c3ba2a5b0aa8fda817bcc72c370040cfb3f8Chris Lattner                                             ".outer", this);
557529b28da455a703d226a31a03400e6662ff569feChris Lattner
558120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // Make sure that NewBB is put someplace intelligent, which doesn't mess up
559120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  // code layout too horribly.
560120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner  PlaceSplitBlockCarefully(NewBB, OuterLoopPreds, L);
561120fce5540b34f81ee5773d30548ce7cc2b5f571Chris Lattner
562529b28da455a703d226a31a03400e6662ff569feChris Lattner  // Create the new outer loop.
563529b28da455a703d226a31a03400e6662ff569feChris Lattner  Loop *NewOuter = new Loop();
564529b28da455a703d226a31a03400e6662ff569feChris Lattner
565529b28da455a703d226a31a03400e6662ff569feChris Lattner  // Change the parent loop to use the outer loop as its child now.
566529b28da455a703d226a31a03400e6662ff569feChris Lattner  if (Loop *Parent = L->getParentLoop())
567529b28da455a703d226a31a03400e6662ff569feChris Lattner    Parent->replaceChildLoopWith(L, NewOuter);
568529b28da455a703d226a31a03400e6662ff569feChris Lattner  else
569c27e056d4fd7f6ecdd8e40eb92230be380c5c8c9Chris Lattner    LI->changeTopLevelLoop(L, NewOuter);
570529b28da455a703d226a31a03400e6662ff569feChris Lattner
571529b28da455a703d226a31a03400e6662ff569feChris Lattner  // L is now a subloop of our outer loop.
572529b28da455a703d226a31a03400e6662ff569feChris Lattner  NewOuter->addChildLoop(L);
573529b28da455a703d226a31a03400e6662ff569feChris Lattner
574d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  // Add the new loop to the pass manager queue.
575d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman  LPM.insertLoopIntoQueue(NewOuter);
576d84db1133345234738b646c70b907bf8a0983ac9Dan Gohman
5779b78763fce4cb418e7a2e672efb84bac25559b79Dan Gohman  for (Loop::block_iterator I = L->block_begin(), E = L->block_end();
5789b78763fce4cb418e7a2e672efb84bac25559b79Dan Gohman       I != E; ++I)
5799b78763fce4cb418e7a2e672efb84bac25559b79Dan Gohman    NewOuter->addBlockEntry(*I);
580529b28da455a703d226a31a03400e6662ff569feChris Lattner
5815c89b5240c90eb8171f999e5f06f815502d0321cDan Gohman  // Now reset the header in L, which had been moved by
5825c89b5240c90eb8171f999e5f06f815502d0321cDan Gohman  // SplitBlockPredecessors for the outer loop.
5835c89b5240c90eb8171f999e5f06f815502d0321cDan Gohman  L->moveToHeader(Header);
5845c89b5240c90eb8171f999e5f06f815502d0321cDan Gohman
585529b28da455a703d226a31a03400e6662ff569feChris Lattner  // Determine which blocks should stay in L and which should be moved out to
586529b28da455a703d226a31a03400e6662ff569feChris Lattner  // the Outer loop now.
587529b28da455a703d226a31a03400e6662ff569feChris Lattner  std::set<BasicBlock*> BlocksInL;
5889672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif  for (pred_iterator PI=pred_begin(Header), E = pred_end(Header); PI!=E; ++PI) {
5899672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    BasicBlock *P = *PI;
5909672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif    if (DT->dominates(Header, P))
5919672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif      AddBlockAndPredsToSet(P, Header, BlocksInL);
5929672414017e9d5d764a56e5c8c61b39163d2d5e5Gabor Greif  }
593529b28da455a703d226a31a03400e6662ff569feChris Lattner
594529b28da455a703d226a31a03400e6662ff569feChris Lattner  // Scan all of the loop children of L, moving them to OuterLoop if they are
595529b28da455a703d226a31a03400e6662ff569feChris Lattner  // not part of the inner loop.
596c08fa28897356be54fba724056c3aa91da8b3e39David Greene  const std::vector<Loop*> &SubLoops = L->getSubLoops();
597c08fa28897356be54fba724056c3aa91da8b3e39David Greene  for (size_t I = 0; I != SubLoops.size(); )
598c08fa28897356be54fba724056c3aa91da8b3e39David Greene    if (BlocksInL.count(SubLoops[I]->getHeader()))
599529b28da455a703d226a31a03400e6662ff569feChris Lattner      ++I;   // Loop remains in L
600529b28da455a703d226a31a03400e6662ff569feChris Lattner    else
601c08fa28897356be54fba724056c3aa91da8b3e39David Greene      NewOuter->addChildLoop(L->removeChildLoop(SubLoops.begin() + I));
602529b28da455a703d226a31a03400e6662ff569feChris Lattner
603529b28da455a703d226a31a03400e6662ff569feChris Lattner  // Now that we know which blocks are in L and which need to be moved to
604529b28da455a703d226a31a03400e6662ff569feChris Lattner  // OuterLoop, move any blocks that need it.
605529b28da455a703d226a31a03400e6662ff569feChris Lattner  for (unsigned i = 0; i != L->getBlocks().size(); ++i) {
606529b28da455a703d226a31a03400e6662ff569feChris Lattner    BasicBlock *BB = L->getBlocks()[i];
607529b28da455a703d226a31a03400e6662ff569feChris Lattner    if (!BlocksInL.count(BB)) {
608529b28da455a703d226a31a03400e6662ff569feChris Lattner      // Move this block to the parent, updating the exit blocks sets
609529b28da455a703d226a31a03400e6662ff569feChris Lattner      L->removeBlockFromLoop(BB);
610c27e056d4fd7f6ecdd8e40eb92230be380c5c8c9Chris Lattner      if ((*LI)[BB] == L)
611c27e056d4fd7f6ecdd8e40eb92230be380c5c8c9Chris Lattner        LI->changeLoopFor(BB, NewOuter);
612529b28da455a703d226a31a03400e6662ff569feChris Lattner      --i;
613529b28da455a703d226a31a03400e6662ff569feChris Lattner    }
614529b28da455a703d226a31a03400e6662ff569feChris Lattner  }
615529b28da455a703d226a31a03400e6662ff569feChris Lattner
616529b28da455a703d226a31a03400e6662ff569feChris Lattner  return NewOuter;
617529b28da455a703d226a31a03400e6662ff569feChris Lattner}
618529b28da455a703d226a31a03400e6662ff569feChris Lattner
619529b28da455a703d226a31a03400e6662ff569feChris Lattner
620529b28da455a703d226a31a03400e6662ff569feChris Lattner
6212ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner/// InsertUniqueBackedgeBlock - This method is called when the specified loop
6222ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner/// has more than one backedge in it.  If this occurs, revector all of these
6232ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner/// backedges to target a new basic block and have that block branch to the loop
6242ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner/// header.  This ensures that loops have exactly one backedge.
6252ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner///
626f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan GohmanBasicBlock *
627f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan GohmanLoopSimplify::InsertUniqueBackedgeBlock(Loop *L, BasicBlock *Preheader) {
6282ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  assert(L->getNumBackEdges() > 1 && "Must have > 1 backedge!");
6292ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
6302ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Get information about the loop
6312ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  BasicBlock *Header = L->getHeader();
6322ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  Function *F = Header->getParent();
6332ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
634f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // Unique backedge insertion currently depends on having a preheader.
635f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  if (!Preheader)
636f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    return 0;
637f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
6382ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Figure out which basic blocks contain back-edges to the loop header.
6392ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  std::vector<BasicBlock*> BackedgeBlocks;
640bf2eefdb0dac4e331ca26fa0792a1dfd420b06f6Gabor Greif  for (pred_iterator I = pred_begin(Header), E = pred_end(Header); I != E; ++I){
641bf2eefdb0dac4e331ca26fa0792a1dfd420b06f6Gabor Greif    BasicBlock *P = *I;
642c2f40066bbceb15e73e5c4df97d2d115f8a36e58Dan Gohman
643c2f40066bbceb15e73e5c4df97d2d115f8a36e58Dan Gohman    // Indirectbr edges cannot be split, so we must fail if we find one.
644c2f40066bbceb15e73e5c4df97d2d115f8a36e58Dan Gohman    if (isa<IndirectBrInst>(P->getTerminator()))
645c2f40066bbceb15e73e5c4df97d2d115f8a36e58Dan Gohman      return 0;
646c2f40066bbceb15e73e5c4df97d2d115f8a36e58Dan Gohman
647bf2eefdb0dac4e331ca26fa0792a1dfd420b06f6Gabor Greif    if (P != Preheader) BackedgeBlocks.push_back(P);
648bf2eefdb0dac4e331ca26fa0792a1dfd420b06f6Gabor Greif  }
6492ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
6502ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Create and insert the new backedge block...
6511d0be15f89cb5056e20e2d24faa8d6afb1573bcaOwen Anderson  BasicBlock *BEBlock = BasicBlock::Create(Header->getContext(),
6521d0be15f89cb5056e20e2d24faa8d6afb1573bcaOwen Anderson                                           Header->getName()+".backedge", F);
653051a950000e21935165db56695e35bade668193bGabor Greif  BranchInst *BETerminator = BranchInst::Create(Header, BEBlock);
6542ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
655c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman  DEBUG(dbgs() << "LoopSimplify: Inserting unique backedge block ";
656c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman        WriteAsOperand(dbgs(), BEBlock, false);
657c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman        dbgs() << "\n");
658c5e49c64d18eacdd72c80c04855df846df97f8a8Dan Gohman
6592ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Move the new backedge block to right after the last backedge block.
6602ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  Function::iterator InsertPos = BackedgeBlocks.back(); ++InsertPos;
6612ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  F->getBasicBlockList().splice(InsertPos, F->getBasicBlockList(), BEBlock);
662fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
6632ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Now that the block has been inserted into the function, create PHI nodes in
6642ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // the backedge block which correspond to any PHI nodes in the header block.
665200a360ec66b4d016c17d6f8e3ea559b1fd07205Alkis Evlogimenos  for (BasicBlock::iterator I = Header->begin(); isa<PHINode>(I); ++I) {
666200a360ec66b4d016c17d6f8e3ea559b1fd07205Alkis Evlogimenos    PHINode *PN = cast<PHINode>(I);
667051a950000e21935165db56695e35bade668193bGabor Greif    PHINode *NewPN = PHINode::Create(PN->getType(), PN->getName()+".be",
668051a950000e21935165db56695e35bade668193bGabor Greif                                     BETerminator);
6695551706b0f8e970720deea0bf6aa34116030d6beChris Lattner    NewPN->reserveOperandSpace(BackedgeBlocks.size());
670cec5b8831d4ee3d81990bf1af41ce1d4f4cf9704Chris Lattner    if (AA) AA->copyValue(PN, NewPN);
6712ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
6722ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // Loop over the PHI node, moving all entries except the one for the
6732ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // preheader over to the new PHI node.
6742ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    unsigned PreheaderIdx = ~0U;
6752ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    bool HasUniqueIncomingValue = true;
6762ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    Value *UniqueValue = 0;
6772ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
6782ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      BasicBlock *IBB = PN->getIncomingBlock(i);
6792ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      Value *IV = PN->getIncomingValue(i);
6802ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      if (IBB == Preheader) {
6812ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner        PreheaderIdx = i;
6822ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      } else {
6832ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner        NewPN->addIncoming(IV, IBB);
6842ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner        if (HasUniqueIncomingValue) {
6852ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner          if (UniqueValue == 0)
6862ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner            UniqueValue = IV;
6872ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner          else if (UniqueValue != IV)
6882ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner            HasUniqueIncomingValue = false;
6892ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner        }
6902ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      }
6912ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    }
692fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
6932ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // Delete all of the incoming values from the old PN except the preheader's
6942ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    assert(PreheaderIdx != ~0U && "PHI has no preheader entry??");
6952ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    if (PreheaderIdx != 0) {
6962ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      PN->setIncomingValue(0, PN->getIncomingValue(PreheaderIdx));
6972ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      PN->setIncomingBlock(0, PN->getIncomingBlock(PreheaderIdx));
6982ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    }
6995551706b0f8e970720deea0bf6aa34116030d6beChris Lattner    // Nuke all entries except the zero'th.
7005551706b0f8e970720deea0bf6aa34116030d6beChris Lattner    for (unsigned i = 0, e = PN->getNumIncomingValues()-1; i != e; ++i)
7015551706b0f8e970720deea0bf6aa34116030d6beChris Lattner      PN->removeIncomingValue(e-i, false);
7022ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
7032ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // Finally, add the newly constructed PHI node as the entry for the BEBlock.
7042ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    PN->addIncoming(NewPN, BEBlock);
7052ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
7062ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // As an optimization, if all incoming values in the new PhiNode (which is a
7072ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // subset of the incoming values of the old PHI node) have the same value,
7082ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    // eliminate the PHI Node.
7092ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    if (HasUniqueIncomingValue) {
7102ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      NewPN->replaceAllUsesWith(UniqueValue);
711cec5b8831d4ee3d81990bf1af41ce1d4f4cf9704Chris Lattner      if (AA) AA->deleteValue(NewPN);
7122ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      BEBlock->getInstList().erase(NewPN);
7132ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    }
7142ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  }
7152ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
7162ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Now that all of the PHI nodes have been inserted and adjusted, modify the
717280a6e607d8eb7401749a92db624a82de47da777Nick Lewycky  // backedge blocks to just to the BEBlock instead of the header.
7182ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  for (unsigned i = 0, e = BackedgeBlocks.size(); i != e; ++i) {
7192ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    TerminatorInst *TI = BackedgeBlocks[i]->getTerminator();
7202ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner    for (unsigned Op = 0, e = TI->getNumSuccessors(); Op != e; ++Op)
7212ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner      if (TI->getSuccessor(Op) == Header)
7222ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner        TI->setSuccessor(Op, BEBlock);
7232ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  }
7242ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
7252ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  //===--- Update all analyses which we must preserve now -----------------===//
7262ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
7272ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // Update Loop Information - we know that this block is now in the current
7282ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner  // loop and all parent loops.
729d735ee85dbab8e4f66f9ec157f19956e0d11ec7aOwen Anderson  L->addBasicBlockToLoop(BEBlock, LI->getBase());
7302ab6a7358e7788eae43b73a79e066322ef0a55d5Chris Lattner
7310e7f728ad1ac25b0ed450fe0f8b86a38d3c2a93aDevang Patel  // Update dominator information
7320e7f728ad1ac25b0ed450fe0f8b86a38d3c2a93aDevang Patel  DT->splitBlock(BEBlock);
7331465d61bdd36cfd6021036a527895f0dd358e97dDuncan Sands  if (DominanceFrontier *DF = getAnalysisIfAvailable<DominanceFrontier>())
7340e7f728ad1ac25b0ed450fe0f8b86a38d3c2a93aDevang Patel    DF->splitBlock(BEBlock);
735f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
736f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  return BEBlock;
737f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman}
738f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
739f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohmanvoid LoopSimplify::verifyAnalysis() const {
740f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // It used to be possible to just assert L->isLoopSimplifyForm(), however
741f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // with the introduction of indirectbr, there are now cases where it's
742f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // not possible to transform a loop as necessary. We can at least check
743f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // that there is an indirectbr near any time there's trouble.
744f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
745f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // Indirectbr can interfere with preheader and unique backedge insertion.
746f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  if (!L->getLoopPreheader() || !L->getLoopLatch()) {
747f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    bool HasIndBrPred = false;
748f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    for (pred_iterator PI = pred_begin(L->getHeader()),
749f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman         PE = pred_end(L->getHeader()); PI != PE; ++PI)
750f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      if (isa<IndirectBrInst>((*PI)->getTerminator())) {
751f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        HasIndBrPred = true;
752f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        break;
753f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      }
754f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    assert(HasIndBrPred &&
755f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman           "LoopSimplify has no excuse for missing loop header info!");
756f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  }
757f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman
758f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  // Indirectbr can interfere with exit block canonicalization.
759f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  if (!L->hasDedicatedExits()) {
760f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    bool HasIndBrExiting = false;
761f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    SmallVector<BasicBlock*, 8> ExitingBlocks;
762f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    L->getExitingBlocks(ExitingBlocks);
763f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    for (unsigned i = 0, e = ExitingBlocks.size(); i != e; ++i)
764f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      if (isa<IndirectBrInst>((ExitingBlocks[i])->getTerminator())) {
765f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        HasIndBrExiting = true;
766f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman        break;
767f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman      }
768f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman    assert(HasIndBrExiting &&
769f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman           "LoopSimplify has no excuse for missing exit block info!");
770f4e82d1f2e25f7cf8b7e9c3bd42b0e384139e07eDan Gohman  }
77138acf9e85d25f022309372c26d54ecb7c77840f2Chris Lattner}
772