1 //===- LoopPass.cpp - Loop Pass and Loop Pass Manager ---------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file was developed by Devang Patel and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file implements LoopPass and LPPassManager. All loop optimization
11 // and transformation passes are derived from LoopPass. LPPassManager is
12 // responsible for managing LoopPasses.
14 //===----------------------------------------------------------------------===//
16 #include "llvm/Analysis/LoopPass.h"
17 #include "llvm/Analysis/ScalarEvolutionExpander.h"
20 //===----------------------------------------------------------------------===//
24 char LPPassManager::ID = 0;
25 /// LPPassManager manages FPPassManagers and CalLGraphSCCPasses.
27 LPPassManager::LPPassManager(int Depth)
28 : FunctionPass((intptr_t)&ID), PMDataManager(Depth) {
35 /// Delete loop from the loop queue and loop hierarcy (LoopInfo).
36 void LPPassManager::deleteLoopFromQueue(Loop *L) {
38 if (Loop *ParentLoop = L->getParentLoop()) { // Not a top-level loop.
39 // Reparent all of the blocks in this loop. Since BBLoop had a parent,
40 // they are now all in it.
41 for (Loop::block_iterator I = L->block_begin(), E = L->block_end();
43 if (LI->getLoopFor(*I) == L) // Don't change blocks in subloops.
44 LI->changeLoopFor(*I, ParentLoop);
46 // Remove the loop from its parent loop.
47 for (Loop::iterator I = ParentLoop->begin(), E = ParentLoop->end();;
49 assert(I != E && "Couldn't find loop");
51 ParentLoop->removeChildLoop(I);
56 // Move all subloops into the parent loop.
57 while (L->begin() != L->end())
58 ParentLoop->addChildLoop(L->removeChildLoop(L->end()-1));
60 // Reparent all of the blocks in this loop. Since BBLoop had no parent,
61 // they no longer in a loop at all.
63 for (unsigned i = 0; i != L->getBlocks().size(); ++i) {
64 // Don't change blocks in subloops.
65 if (LI->getLoopFor(L->getBlocks()[i]) == L) {
66 LI->removeBlock(L->getBlocks()[i]);
71 // Remove the loop from the top-level LoopInfo object.
72 for (LoopInfo::iterator I = LI->begin(), E = LI->end();; ++I) {
73 assert(I != E && "Couldn't find loop");
80 // Move all of the subloops to the top-level.
81 while (L->begin() != L->end())
82 LI->addTopLevelLoop(L->removeChildLoop(L->end()-1));
87 // If L is current loop then skip rest of the passes and let
88 // runOnFunction remove L from LQ. Otherwise, remove L from LQ now
89 // and continue applying other passes on CurrentLoop.
90 if (CurrentLoop == L) {
95 for (std::deque<Loop *>::iterator I = LQ.begin(),
96 E = LQ.end(); I != E; ++I) {
104 // Inset loop into loop nest (LoopInfo) and loop queue (LQ).
105 void LPPassManager::insertLoop(Loop *L, Loop *ParentLoop) {
107 assert (CurrentLoop != L && "Cannot insert CurrentLoop");
109 // Insert into loop nest
111 ParentLoop->addChildLoop(L);
113 LI->addTopLevelLoop(L);
115 // Insert L into loop queue
116 if (L == CurrentLoop)
118 else if (!ParentLoop)
119 // This is top level loop.
122 // Insert L after ParentLoop
123 for (std::deque<Loop *>::iterator I = LQ.begin(),
124 E = LQ.end(); I != E; ++I) {
125 if (*I == ParentLoop) {
126 // deque does not support insert after.
135 // Reoptimize this loop. LPPassManager will re-insert this loop into the
136 // queue. This allows LoopPass to change loop nest for the loop. This
137 // utility may send LPPassManager into infinite loops so use caution.
138 void LPPassManager::redoLoop(Loop *L) {
139 assert (CurrentLoop == L && "Can redo only CurrentLoop");
143 // Recurse through all subloops and all loops into LQ.
144 static void addLoopIntoQueue(Loop *L, std::deque<Loop *> &LQ) {
146 for (Loop::iterator I = L->begin(), E = L->end(); I != E; ++I)
147 addLoopIntoQueue(*I, LQ);
150 /// Pass Manager itself does not invalidate any analysis info.
151 void LPPassManager::getAnalysisUsage(AnalysisUsage &Info) const {
152 // LPPassManager needs LoopInfo. In the long term LoopInfo class will
153 // become part of LPPassManager.
154 Info.addRequired<LoopInfo>();
155 // Used by IndVar doInitialization.
156 Info.addRequired<ScalarEvolution>();
157 Info.setPreservesAll();
160 /// verifyLoopInfo - Verify loop nest.
161 void LPPassManager::verifyLoopInfo() {
162 assert (LI && "Loop Info is missing");
164 for (LoopInfo::iterator I = LI->begin(), E = LI->end(); I != E; ++I) {
166 assert (L->getHeader() && "Loop header is missing");
167 assert (L->getLoopPreheader() && "Loop preheader is missing");
168 assert (L->getLoopLatch() && "Loop latch is missing");
172 /// run - Execute all of the passes scheduled for execution. Keep track of
173 /// whether any of the passes modifies the function, and if so, return true.
174 bool LPPassManager::runOnFunction(Function &F) {
175 LI = &getAnalysis<LoopInfo>();
176 bool Changed = false;
178 // Populate Loop Queue
179 for (LoopInfo::iterator I = LI->begin(), E = LI->end(); I != E; ++I)
180 addLoopIntoQueue(*I, LQ);
183 for (std::deque<Loop *>::const_iterator I = LQ.begin(), E = LQ.end();
186 for (unsigned Index = 0; Index < getNumContainedPasses(); ++Index) {
187 Pass *P = getContainedPass(Index);
188 LoopPass *LP = dynamic_cast<LoopPass *>(P);
190 Changed |= LP->doInitialization(L, *this);
195 while (!LQ.empty()) {
197 CurrentLoop = LQ.back();
198 skipThisLoop = false;
199 redoThisLoop = false;
201 // Run all passes on current SCC
202 for (unsigned Index = 0; Index < getNumContainedPasses(); ++Index) {
204 Pass *P = getContainedPass(Index);
205 AnalysisUsage AnUsage;
206 P->getAnalysisUsage(AnUsage);
208 dumpPassInfo(P, EXECUTION_MSG, ON_LOOP_MSG, "");
209 dumpAnalysisSetInfo("Required", P, AnUsage.getRequiredSet());
211 initializeAnalysisImpl(P);
214 LoopPass *LP = dynamic_cast<LoopPass *>(P);
215 assert (LP && "Invalid LPPassManager member");
216 LP->runOnLoop(CurrentLoop, *this);
221 dumpPassInfo(P, MODIFICATION_MSG, ON_LOOP_MSG, "");
222 dumpAnalysisSetInfo("Preserved", P, AnUsage.getPreservedSet());
224 removeNotPreservedAnalysis(P);
225 recordAvailableAnalysis(P);
226 removeDeadPasses(P, "", ON_LOOP_MSG);
229 // Do not run other passes on this loop.
233 // Pop the loop from queue after running all passes.
237 LQ.push_back(CurrentLoop);
241 for (unsigned Index = 0; Index < getNumContainedPasses(); ++Index) {
242 Pass *P = getContainedPass(Index);
243 LoopPass *LP = dynamic_cast <LoopPass *>(P);
245 Changed |= LP->doFinalization();
252 //===----------------------------------------------------------------------===//
255 // Check if this pass is suitable for the current LPPassManager, if
256 // available. This pass P is not suitable for a LPPassManager if P
257 // is not preserving higher level analysis info used by other
258 // LPPassManager passes. In such case, pop LPPassManager from the
259 // stack. This will force assignPassManager() to create new
260 // LPPassManger as expected.
261 void LoopPass::preparePassManager(PMStack &PMS) {
263 // Find LPPassManager
264 while (!PMS.empty()) {
265 if (PMS.top()->getPassManagerType() > PMT_LoopPassManager)
271 LPPassManager *LPPM = dynamic_cast<LPPassManager *>(PMS.top());
273 // If this pass is destroying high level information that is used
274 // by other passes that are managed by LPM then do not insert
275 // this pass in current LPM. Use new LPPassManager.
276 if (LPPM && !LPPM->preserveHigherLevelAnalysis(this))
280 /// Assign pass manager to manage this pass.
281 void LoopPass::assignPassManager(PMStack &PMS,
282 PassManagerType PreferredType) {
283 // Find LPPassManager
284 while (!PMS.empty()) {
285 if (PMS.top()->getPassManagerType() > PMT_LoopPassManager)
291 LPPassManager *LPPM = dynamic_cast<LPPassManager *>(PMS.top());
293 // Create new Loop Pass Manager if it does not exist.
296 assert (!PMS.empty() && "Unable to create Loop Pass Manager");
297 PMDataManager *PMD = PMS.top();
299 // [1] Create new Call Graph Pass Manager
300 LPPM = new LPPassManager(PMD->getDepth() + 1);
301 LPPM->populateInheritedAnalysis(PMS);
303 // [2] Set up new manager's top level manager
304 PMTopLevelManager *TPM = PMD->getTopLevelManager();
305 TPM->addIndirectPassManager(LPPM);
307 // [3] Assign manager to manage this new manager. This may create
308 // and push new managers into PMS
309 Pass *P = dynamic_cast<Pass *>(LPPM);
310 TPM->schedulePass(P);
312 // [4] Push new manager into PMS