blob: 44dde1b51cfcb911fe2f070e918a9856618cb99e [file] [log] [blame]
Dan Gohman3dc2d922008-05-14 00:24:14 +00001//===-- UnrollLoop.cpp - Loop unrolling utilities -------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements some loop unrolling utilities. It does not define any
11// actual pass or policy, but provides a single function to perform loop
12// unrolling.
13//
Dan Gohman3dc2d922008-05-14 00:24:14 +000014// The process of unrolling can produce extraneous basic blocks linked with
15// unconditional branches. This will be corrected in the future.
Chris Lattnerdfcfcb42011-01-11 08:00:40 +000016//
Dan Gohman3dc2d922008-05-14 00:24:14 +000017//===----------------------------------------------------------------------===//
18
Dan Gohman3dc2d922008-05-14 00:24:14 +000019#include "llvm/Transforms/Utils/UnrollLoop.h"
Mark Heffernan675d4012014-07-10 23:30:06 +000020#include "llvm/ADT/SmallPtrSet.h"
Dan Gohman3dc2d922008-05-14 00:24:14 +000021#include "llvm/ADT/Statistic.h"
Chandler Carruth66b31302015-01-04 12:03:27 +000022#include "llvm/Analysis/AssumptionCache.h"
Duncan Sands433c1672010-11-23 20:26:33 +000023#include "llvm/Analysis/InstructionSimplify.h"
Andrew Trickb72bbe22011-08-10 00:28:10 +000024#include "llvm/Analysis/LoopIterator.h"
Dan Gohman3dc2d922008-05-14 00:24:14 +000025#include "llvm/Analysis/LoopPass.h"
Dan Gohmana7908ae2010-07-26 18:02:06 +000026#include "llvm/Analysis/ScalarEvolution.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000027#include "llvm/IR/BasicBlock.h"
Hal Finkela995f922014-07-10 14:41:31 +000028#include "llvm/IR/DataLayout.h"
Diego Novillo7f8af8b2014-05-22 14:19:46 +000029#include "llvm/IR/DiagnosticInfo.h"
Benjamin Kramer799003b2015-03-23 19:32:43 +000030#include "llvm/IR/Dominators.h"
Diego Novillo34fc8a72014-04-29 14:27:31 +000031#include "llvm/IR/LLVMContext.h"
Dan Gohman3dc2d922008-05-14 00:24:14 +000032#include "llvm/Support/Debug.h"
Daniel Dunbar0dd5e1e2009-07-25 00:23:56 +000033#include "llvm/Support/raw_ostream.h"
Chris Lattnerdc3f6f22008-12-03 19:44:02 +000034#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dan Gohman3dc2d922008-05-14 00:24:14 +000035#include "llvm/Transforms/Utils/Cloning.h"
36#include "llvm/Transforms/Utils/Local.h"
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +000037#include "llvm/Transforms/Utils/LoopUtils.h"
Andrew Trick4d0040b2011-08-10 04:29:49 +000038#include "llvm/Transforms/Utils/SimplifyIndVar.h"
Dan Gohman3dc2d922008-05-14 00:24:14 +000039using namespace llvm;
40
Chandler Carruth964daaa2014-04-22 02:55:47 +000041#define DEBUG_TYPE "loop-unroll"
42
Chris Lattnerdc3f6f22008-12-03 19:44:02 +000043// TODO: Should these be here or in LoopUnroll?
Dan Gohman3dc2d922008-05-14 00:24:14 +000044STATISTIC(NumCompletelyUnrolled, "Number of loops completely unrolled");
Chris Lattnerdfcfcb42011-01-11 08:00:40 +000045STATISTIC(NumUnrolled, "Number of loops unrolled (completely or otherwise)");
Dan Gohman3dc2d922008-05-14 00:24:14 +000046
47/// RemapInstruction - Convert the instruction operands from referencing the
Devang Patelb8f11de2010-06-23 23:55:51 +000048/// current values into those specified by VMap.
Dan Gohman3dc2d922008-05-14 00:24:14 +000049static inline void RemapInstruction(Instruction *I,
Rafael Espindola229e38f2010-10-13 01:36:30 +000050 ValueToValueMapTy &VMap) {
Dan Gohman3dc2d922008-05-14 00:24:14 +000051 for (unsigned op = 0, E = I->getNumOperands(); op != E; ++op) {
52 Value *Op = I->getOperand(op);
Rafael Espindola229e38f2010-10-13 01:36:30 +000053 ValueToValueMapTy::iterator It = VMap.find(Op);
Devang Patelb8f11de2010-06-23 23:55:51 +000054 if (It != VMap.end())
Dan Gohmanfa8969f2009-10-31 14:46:50 +000055 I->setOperand(op, It->second);
Dan Gohman3dc2d922008-05-14 00:24:14 +000056 }
Jay Foad61ea0e42011-06-23 09:09:15 +000057
58 if (PHINode *PN = dyn_cast<PHINode>(I)) {
59 for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
60 ValueToValueMapTy::iterator It = VMap.find(PN->getIncomingBlock(i));
61 if (It != VMap.end())
62 PN->setIncomingBlock(i, cast<BasicBlock>(It->second));
63 }
64 }
Dan Gohman3dc2d922008-05-14 00:24:14 +000065}
66
Dan Gohman2d02ff82009-10-31 17:33:01 +000067/// FoldBlockIntoPredecessor - Folds a basic block into its predecessor if it
68/// only has one predecessor, and that predecessor only has one successor.
Mark Heffernan675d4012014-07-10 23:30:06 +000069/// The LoopInfo Analysis that is passed will be kept consistent. If folding is
70/// successful references to the containing loop must be removed from
71/// ScalarEvolution by calling ScalarEvolution::forgetLoop because SE may have
72/// references to the eliminated BB. The argument ForgottenLoops contains a set
73/// of loops that have already been forgotten to prevent redundant, expensive
74/// calls to ScalarEvolution::forgetLoop. Returns the new combined block.
75static BasicBlock *
76FoldBlockIntoPredecessor(BasicBlock *BB, LoopInfo* LI, LPPassManager *LPM,
77 SmallPtrSetImpl<Loop *> &ForgottenLoops) {
Dan Gohman2d02ff82009-10-31 17:33:01 +000078 // Merge basic blocks into their predecessor if there is only one distinct
79 // pred, and if there is only one distinct successor of the predecessor, and
80 // if there are no PHI nodes.
81 BasicBlock *OnlyPred = BB->getSinglePredecessor();
Craig Topperf40110f2014-04-25 05:29:35 +000082 if (!OnlyPred) return nullptr;
Dan Gohman2d02ff82009-10-31 17:33:01 +000083
84 if (OnlyPred->getTerminator()->getNumSuccessors() != 1)
Craig Topperf40110f2014-04-25 05:29:35 +000085 return nullptr;
Dan Gohman2d02ff82009-10-31 17:33:01 +000086
David Greene627f40a2010-01-05 01:26:41 +000087 DEBUG(dbgs() << "Merging: " << *BB << "into: " << *OnlyPred);
Dan Gohman2d02ff82009-10-31 17:33:01 +000088
89 // Resolve any PHI nodes at the start of the block. They are all
90 // guaranteed to have exactly one entry if they exist, unless there are
91 // multiple duplicate (but guaranteed to be equal) entries for the
92 // incoming edges. This occurs when there are multiple edges from
93 // OnlyPred to OnlySucc.
94 FoldSingleEntryPHINodes(BB);
95
96 // Delete the unconditional branch from the predecessor...
97 OnlyPred->getInstList().pop_back();
98
Dan Gohman2d02ff82009-10-31 17:33:01 +000099 // Make all PHI nodes that referred to BB now refer to Pred as their
100 // source...
101 BB->replaceAllUsesWith(OnlyPred);
102
Jay Foad61ea0e42011-06-23 09:09:15 +0000103 // Move all definitions in the successor to the predecessor...
104 OnlyPred->getInstList().splice(OnlyPred->end(), BB->getInstList());
105
NAKAMURA Takumif9c83392013-11-17 18:05:34 +0000106 // OldName will be valid until erased.
Jakub Staszak86a74922013-11-13 20:09:11 +0000107 StringRef OldName = BB->getName();
Dan Gohman2d02ff82009-10-31 17:33:01 +0000108
109 // Erase basic block from the function...
Andrew Trickbf69d032011-08-03 18:32:11 +0000110
111 // ScalarEvolution holds references to loop exit blocks.
Andrew Tricka6fb9102012-06-05 17:51:05 +0000112 if (LPM) {
Chandler Carruth2f1fd162015-08-17 02:08:17 +0000113 if (auto *SEWP =
114 LPM->getAnalysisIfAvailable<ScalarEvolutionWrapperPass>()) {
Mark Heffernan675d4012014-07-10 23:30:06 +0000115 if (Loop *L = LI->getLoopFor(BB)) {
David Blaikie70573dc2014-11-19 07:49:26 +0000116 if (ForgottenLoops.insert(L).second)
Chandler Carruth2f1fd162015-08-17 02:08:17 +0000117 SEWP->getSE().forgetLoop(L);
Mark Heffernan675d4012014-07-10 23:30:06 +0000118 }
Andrew Tricka6fb9102012-06-05 17:51:05 +0000119 }
Andrew Trickbf69d032011-08-03 18:32:11 +0000120 }
Dan Gohman2d02ff82009-10-31 17:33:01 +0000121 LI->removeBlock(BB);
Dan Gohman2d02ff82009-10-31 17:33:01 +0000122
123 // Inherit predecessor's name if it exists...
124 if (!OldName.empty() && !OnlyPred->hasName())
125 OnlyPred->setName(OldName);
126
NAKAMURA Takumif9c83392013-11-17 18:05:34 +0000127 BB->eraseFromParent();
128
Dan Gohman2d02ff82009-10-31 17:33:01 +0000129 return OnlyPred;
130}
131
Dan Gohman3dc2d922008-05-14 00:24:14 +0000132/// Unroll the given loop by Count. The loop must be in LCSSA form. Returns true
Chris Lattner4a14fbc2011-02-18 04:25:21 +0000133/// if unrolling was successful, or false if the loop was unmodified. Unrolling
Dan Gohman3dc2d922008-05-14 00:24:14 +0000134/// can only fail when the loop's latch block is not terminated by a conditional
135/// branch instruction. However, if the trip count (and multiple) are not known,
136/// loop unrolling will mostly produce more code that is no faster.
137///
Andrew Trick990f7712011-07-25 22:17:47 +0000138/// TripCount is generally defined as the number of times the loop header
139/// executes. UnrollLoop relaxes the definition to permit early exits: here
140/// TripCount is the iteration on which control exits LatchBlock if no early
141/// exits were taken. Note that UnrollLoop assumes that the loop counter test
142/// terminates LatchBlock in order to remove unnecesssary instances of the
143/// test. In other words, control may exit the loop prior to TripCount
144/// iterations via an early branch, but control may not exit the loop from the
145/// LatchBlock's terminator prior to TripCount iterations.
146///
147/// Similarly, TripMultiple divides the number of times that the LatchBlock may
148/// execute without exiting the loop.
149///
Sanjoy Dase178f462015-04-14 03:20:38 +0000150/// If AllowRuntime is true then UnrollLoop will consider unrolling loops that
151/// have a runtime (i.e. not compile time constant) trip count. Unrolling these
152/// loops require a unroll "prologue" that runs "RuntimeTripCount % Count"
153/// iterations before branching into the unrolled loop. UnrollLoop will not
154/// runtime-unroll the loop if computing RuntimeTripCount will be expensive and
155/// AllowExpensiveTripCount is false.
156///
Dan Gohman3dc2d922008-05-14 00:24:14 +0000157/// The LoopInfo Analysis that is passed will be kept consistent.
158///
159/// If a LoopPassManager is passed in, and the loop is fully removed, it will be
160/// removed from the LoopPassManager as well. LPM can also be NULL.
Andrew Trick4d0040b2011-08-10 04:29:49 +0000161///
162/// This utility preserves LoopInfo. If DominatorTree or ScalarEvolution are
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +0000163/// available from the Pass it must also preserve those analyses.
Andrew Trick1cabe542011-07-23 00:33:05 +0000164bool llvm::UnrollLoop(Loop *L, unsigned Count, unsigned TripCount,
Sanjoy Dase178f462015-04-14 03:20:38 +0000165 bool AllowRuntime, bool AllowExpensiveTripCount,
166 unsigned TripMultiple, LoopInfo *LI, Pass *PP,
167 LPPassManager *LPM, AssumptionCache *AC) {
Dan Gohman415c64e2009-11-05 19:44:06 +0000168 BasicBlock *Preheader = L->getLoopPreheader();
169 if (!Preheader) {
David Greene627f40a2010-01-05 01:26:41 +0000170 DEBUG(dbgs() << " Can't unroll; loop preheader-insertion failed.\n");
Dan Gohman415c64e2009-11-05 19:44:06 +0000171 return false;
172 }
173
Dan Gohman3dc2d922008-05-14 00:24:14 +0000174 BasicBlock *LatchBlock = L->getLoopLatch();
Dan Gohman415c64e2009-11-05 19:44:06 +0000175 if (!LatchBlock) {
David Greene627f40a2010-01-05 01:26:41 +0000176 DEBUG(dbgs() << " Can't unroll; loop exit-block-insertion failed.\n");
Dan Gohman415c64e2009-11-05 19:44:06 +0000177 return false;
178 }
179
Andrew Trick4442bfe2012-04-10 05:14:42 +0000180 // Loops with indirectbr cannot be cloned.
181 if (!L->isSafeToClone()) {
182 DEBUG(dbgs() << " Can't unroll; Loop body cannot be cloned.\n");
183 return false;
184 }
185
Dan Gohman415c64e2009-11-05 19:44:06 +0000186 BasicBlock *Header = L->getHeader();
Dan Gohman3dc2d922008-05-14 00:24:14 +0000187 BranchInst *BI = dyn_cast<BranchInst>(LatchBlock->getTerminator());
Andrew Trick279e7a62011-07-23 00:29:16 +0000188
Dan Gohman3dc2d922008-05-14 00:24:14 +0000189 if (!BI || BI->isUnconditional()) {
190 // The loop-rotate pass can be helpful to avoid this in many cases.
David Greene627f40a2010-01-05 01:26:41 +0000191 DEBUG(dbgs() <<
Chris Lattnerb25de3f2009-08-23 04:37:46 +0000192 " Can't unroll; loop not terminated by a conditional branch.\n");
Dan Gohman3dc2d922008-05-14 00:24:14 +0000193 return false;
194 }
Andrew Trick279e7a62011-07-23 00:29:16 +0000195
Chris Lattner4a14fbc2011-02-18 04:25:21 +0000196 if (Header->hasAddressTaken()) {
197 // The loop-rotate pass can be helpful to avoid this in many cases.
198 DEBUG(dbgs() <<
199 " Won't unroll loop: address of header block is taken.\n");
200 return false;
201 }
Dan Gohman3dc2d922008-05-14 00:24:14 +0000202
Dan Gohman3dc2d922008-05-14 00:24:14 +0000203 if (TripCount != 0)
David Greene627f40a2010-01-05 01:26:41 +0000204 DEBUG(dbgs() << " Trip Count = " << TripCount << "\n");
Dan Gohman3dc2d922008-05-14 00:24:14 +0000205 if (TripMultiple != 1)
David Greene627f40a2010-01-05 01:26:41 +0000206 DEBUG(dbgs() << " Trip Multiple = " << TripMultiple << "\n");
Dan Gohman3dc2d922008-05-14 00:24:14 +0000207
208 // Effectively "DCE" unrolled iterations that are beyond the tripcount
209 // and will never be executed.
210 if (TripCount != 0 && Count > TripCount)
211 Count = TripCount;
212
Andrew Trickca3417e2011-12-16 02:03:48 +0000213 // Don't enter the unroll code if there is nothing to do. This way we don't
214 // need to support "partial unrolling by 1".
215 if (TripCount == 0 && Count < 2)
216 return false;
217
Dan Gohman3dc2d922008-05-14 00:24:14 +0000218 assert(Count > 0);
219 assert(TripMultiple > 0);
220 assert(TripCount == 0 || TripCount % TripMultiple == 0);
221
222 // Are we eliminating the loop control altogether?
223 bool CompletelyUnroll = Count == TripCount;
Michael Zolotukhin78760ee2015-12-09 18:20:28 +0000224 SmallVector<BasicBlock *, 4> ExitBlocks;
225 L->getExitBlocks(ExitBlocks);
226 Loop *ParentL = L->getParentLoop();
227 bool AllExitsAreInsideParentLoop = !ParentL ||
228 std::all_of(ExitBlocks.begin(), ExitBlocks.end(),
229 [&](BasicBlock *BB) { return ParentL->contains(BB); });
Dan Gohman3dc2d922008-05-14 00:24:14 +0000230
Andrew Trickd04d15292011-12-09 06:19:40 +0000231 // We assume a run-time trip count if the compiler cannot
232 // figure out the loop trip count and the unroll-runtime
233 // flag is specified.
234 bool RuntimeTripCount = (TripCount == 0 && Count > 0 && AllowRuntime);
235
Sanjoy Dase178f462015-04-14 03:20:38 +0000236 if (RuntimeTripCount &&
237 !UnrollRuntimeLoopProlog(L, Count, AllowExpensiveTripCount, LI, LPM))
Andrew Trickd04d15292011-12-09 06:19:40 +0000238 return false;
239
240 // Notify ScalarEvolution that the loop will be substantially changed,
241 // if not outright eliminated.
Chandler Carruth2f1fd162015-08-17 02:08:17 +0000242 auto *SEWP =
243 PP ? PP->getAnalysisIfAvailable<ScalarEvolutionWrapperPass>() : nullptr;
244 ScalarEvolution *SE = SEWP ? &SEWP->getSE() : nullptr;
Duncan P. N. Exon Smith9b4d37e2014-10-07 21:12:44 +0000245 if (SE)
246 SE->forgetLoop(L);
Andrew Trickd04d15292011-12-09 06:19:40 +0000247
Dan Gohman3dc2d922008-05-14 00:24:14 +0000248 // If we know the trip count, we know the multiple...
249 unsigned BreakoutTrip = 0;
250 if (TripCount != 0) {
251 BreakoutTrip = TripCount % Count;
252 TripMultiple = 0;
253 } else {
254 // Figure out what multiple to use.
255 BreakoutTrip = TripMultiple =
256 (unsigned)GreatestCommonDivisor64(Count, TripMultiple);
257 }
258
Diego Novillo34fc8a72014-04-29 14:27:31 +0000259 // Report the unrolling decision.
260 DebugLoc LoopLoc = L->getStartLoc();
261 Function *F = Header->getParent();
262 LLVMContext &Ctx = F->getContext();
263
Dan Gohman3dc2d922008-05-14 00:24:14 +0000264 if (CompletelyUnroll) {
David Greene627f40a2010-01-05 01:26:41 +0000265 DEBUG(dbgs() << "COMPLETELY UNROLLING loop %" << Header->getName()
Daniel Dunbar0dd5e1e2009-07-25 00:23:56 +0000266 << " with trip count " << TripCount << "!\n");
Diego Novillo7f8af8b2014-05-22 14:19:46 +0000267 emitOptimizationRemark(Ctx, DEBUG_TYPE, *F, LoopLoc,
268 Twine("completely unrolled loop with ") +
269 Twine(TripCount) + " iterations");
Dan Gohman3dc2d922008-05-14 00:24:14 +0000270 } else {
Benjamin Kramercccdadc2014-07-08 14:55:06 +0000271 auto EmitDiag = [&](const Twine &T) {
272 emitOptimizationRemark(Ctx, DEBUG_TYPE, *F, LoopLoc,
273 "unrolled loop by a factor of " + Twine(Count) +
274 T);
275 };
276
David Greene627f40a2010-01-05 01:26:41 +0000277 DEBUG(dbgs() << "UNROLLING loop %" << Header->getName()
Daniel Dunbar0dd5e1e2009-07-25 00:23:56 +0000278 << " by " << Count);
Dan Gohman3dc2d922008-05-14 00:24:14 +0000279 if (TripMultiple == 0 || BreakoutTrip != TripMultiple) {
David Greene627f40a2010-01-05 01:26:41 +0000280 DEBUG(dbgs() << " with a breakout at trip " << BreakoutTrip);
Benjamin Kramercccdadc2014-07-08 14:55:06 +0000281 EmitDiag(" with a breakout at trip " + Twine(BreakoutTrip));
Dan Gohman3dc2d922008-05-14 00:24:14 +0000282 } else if (TripMultiple != 1) {
David Greene627f40a2010-01-05 01:26:41 +0000283 DEBUG(dbgs() << " with " << TripMultiple << " trips per branch");
Benjamin Kramercccdadc2014-07-08 14:55:06 +0000284 EmitDiag(" with " + Twine(TripMultiple) + " trips per branch");
Andrew Trickd04d15292011-12-09 06:19:40 +0000285 } else if (RuntimeTripCount) {
286 DEBUG(dbgs() << " with run-time trip count");
Benjamin Kramercccdadc2014-07-08 14:55:06 +0000287 EmitDiag(" with run-time trip count");
Dan Gohman3dc2d922008-05-14 00:24:14 +0000288 }
David Greene627f40a2010-01-05 01:26:41 +0000289 DEBUG(dbgs() << "!\n");
Dan Gohman3dc2d922008-05-14 00:24:14 +0000290 }
291
Dan Gohman04c8bd72008-06-24 20:44:42 +0000292 bool ContinueOnTrue = L->contains(BI->getSuccessor(0));
Dan Gohman3dc2d922008-05-14 00:24:14 +0000293 BasicBlock *LoopExit = BI->getSuccessor(ContinueOnTrue);
294
295 // For the first iteration of the loop, we should use the precloned values for
296 // PHI nodes. Insert associations now.
Devang Patel21766432010-04-20 22:24:18 +0000297 ValueToValueMapTy LastValueMap;
Dan Gohman04c8bd72008-06-24 20:44:42 +0000298 std::vector<PHINode*> OrigPHINode;
Dan Gohman3dc2d922008-05-14 00:24:14 +0000299 for (BasicBlock::iterator I = Header->begin(); isa<PHINode>(I); ++I) {
Andrew Trick5e0ee1c2011-08-09 03:11:29 +0000300 OrigPHINode.push_back(cast<PHINode>(I));
Dan Gohman3dc2d922008-05-14 00:24:14 +0000301 }
302
303 std::vector<BasicBlock*> Headers;
304 std::vector<BasicBlock*> Latches;
305 Headers.push_back(Header);
306 Latches.push_back(LatchBlock);
307
Andrew Trickb72bbe22011-08-10 00:28:10 +0000308 // The current on-the-fly SSA update requires blocks to be processed in
309 // reverse postorder so that LastValueMap contains the correct value at each
310 // exit.
311 LoopBlocksDFS DFS(L);
Andrew Trick78b40c32011-08-10 01:59:05 +0000312 DFS.perform(LI);
313
Andrew Trickb72bbe22011-08-10 00:28:10 +0000314 // Stash the DFS iterators before adding blocks to the loop.
315 LoopBlocksDFS::RPOIterator BlockBegin = DFS.beginRPO();
316 LoopBlocksDFS::RPOIterator BlockEnd = DFS.endRPO();
317
Dan Gohman3dc2d922008-05-14 00:24:14 +0000318 for (unsigned It = 1; It != Count; ++It) {
Dan Gohman3dc2d922008-05-14 00:24:14 +0000319 std::vector<BasicBlock*> NewBlocks;
Duncan P. N. Exon Smithc46cfcb2014-10-07 21:19:00 +0000320 SmallDenseMap<const Loop *, Loop *, 4> NewLoops;
321 NewLoops[L] = L;
Andrew Trick279e7a62011-07-23 00:29:16 +0000322
Andrew Trickb72bbe22011-08-10 00:28:10 +0000323 for (LoopBlocksDFS::RPOIterator BB = BlockBegin; BB != BlockEnd; ++BB) {
Devang Patelb8f11de2010-06-23 23:55:51 +0000324 ValueToValueMapTy VMap;
325 BasicBlock *New = CloneBasicBlock(*BB, VMap, "." + Twine(It));
Dan Gohman04c8bd72008-06-24 20:44:42 +0000326 Header->getParent()->getBasicBlockList().push_back(New);
Dan Gohman3dc2d922008-05-14 00:24:14 +0000327
Duncan P. N. Exon Smithc46cfcb2014-10-07 21:19:00 +0000328 // Tell LI about New.
329 if (*BB == Header) {
330 assert(LI->getLoopFor(*BB) == L && "Header should not be in a sub-loop");
Chandler Carruth691addc2015-01-18 01:25:51 +0000331 L->addBasicBlockToLoop(New, *LI);
Duncan P. N. Exon Smithc46cfcb2014-10-07 21:19:00 +0000332 } else {
333 // Figure out which loop New is in.
334 const Loop *OldLoop = LI->getLoopFor(*BB);
335 assert(OldLoop && "Should (at least) be in the loop being unrolled!");
336
337 Loop *&NewLoop = NewLoops[OldLoop];
338 if (!NewLoop) {
339 // Found a new sub-loop.
340 assert(*BB == OldLoop->getHeader() &&
341 "Header should be first in RPO");
342
343 Loop *NewLoopParent = NewLoops.lookup(OldLoop->getParentLoop());
344 assert(NewLoopParent &&
345 "Expected parent loop before sub-loop in RPO");
346 NewLoop = new Loop;
347 NewLoopParent->addChildLoop(NewLoop);
348
349 // Forget the old loop, since its inputs may have changed.
350 if (SE)
351 SE->forgetLoop(OldLoop);
352 }
Chandler Carruth691addc2015-01-18 01:25:51 +0000353 NewLoop->addBasicBlockToLoop(New, *LI);
Duncan P. N. Exon Smithc46cfcb2014-10-07 21:19:00 +0000354 }
Duncan P. N. Exon Smithe5d7d972014-10-06 22:05:02 +0000355
Dan Gohman04c8bd72008-06-24 20:44:42 +0000356 if (*BB == Header)
Duncan P. N. Exon Smith0bbf5412014-10-06 22:04:59 +0000357 // Loop over all of the PHI nodes in the block, changing them to use
358 // the incoming values from the previous block.
Dan Gohman04c8bd72008-06-24 20:44:42 +0000359 for (unsigned i = 0, e = OrigPHINode.size(); i != e; ++i) {
Devang Patelb8f11de2010-06-23 23:55:51 +0000360 PHINode *NewPHI = cast<PHINode>(VMap[OrigPHINode[i]]);
Dan Gohman3dc2d922008-05-14 00:24:14 +0000361 Value *InVal = NewPHI->getIncomingValueForBlock(LatchBlock);
362 if (Instruction *InValI = dyn_cast<Instruction>(InVal))
Dan Gohman18fa5682009-12-18 01:24:09 +0000363 if (It > 1 && L->contains(InValI))
Dan Gohman3dc2d922008-05-14 00:24:14 +0000364 InVal = LastValueMap[InValI];
Devang Patelb8f11de2010-06-23 23:55:51 +0000365 VMap[OrigPHINode[i]] = InVal;
Dan Gohman3dc2d922008-05-14 00:24:14 +0000366 New->getInstList().erase(NewPHI);
367 }
368
369 // Update our running map of newest clones
Dan Gohman04c8bd72008-06-24 20:44:42 +0000370 LastValueMap[*BB] = New;
Devang Patelb8f11de2010-06-23 23:55:51 +0000371 for (ValueToValueMapTy::iterator VI = VMap.begin(), VE = VMap.end();
Dan Gohman3dc2d922008-05-14 00:24:14 +0000372 VI != VE; ++VI)
373 LastValueMap[VI->first] = VI->second;
374
Andrew Trickb72bbe22011-08-10 00:28:10 +0000375 // Add phi entries for newly created values to all exit blocks.
Duncan P. N. Exon Smith6c990152014-07-21 17:06:51 +0000376 for (succ_iterator SI = succ_begin(*BB), SE = succ_end(*BB);
377 SI != SE; ++SI) {
378 if (L->contains(*SI))
Andrew Trickb72bbe22011-08-10 00:28:10 +0000379 continue;
Duncan P. N. Exon Smith6c990152014-07-21 17:06:51 +0000380 for (BasicBlock::iterator BBI = (*SI)->begin();
Andrew Trickb72bbe22011-08-10 00:28:10 +0000381 PHINode *phi = dyn_cast<PHINode>(BBI); ++BBI) {
382 Value *Incoming = phi->getIncomingValueForBlock(*BB);
383 ValueToValueMapTy::iterator It = LastValueMap.find(Incoming);
384 if (It != LastValueMap.end())
385 Incoming = It->second;
386 phi->addIncoming(Incoming, New);
387 }
388 }
Dan Gohman04c8bd72008-06-24 20:44:42 +0000389 // Keep track of new headers and latches as we create them, so that
390 // we can insert the proper branches later.
391 if (*BB == Header)
392 Headers.push_back(New);
Andrew Trickb72bbe22011-08-10 00:28:10 +0000393 if (*BB == LatchBlock)
Dan Gohman04c8bd72008-06-24 20:44:42 +0000394 Latches.push_back(New);
395
Dan Gohman04c8bd72008-06-24 20:44:42 +0000396 NewBlocks.push_back(New);
Dan Gohman3dc2d922008-05-14 00:24:14 +0000397 }
Andrew Trick279e7a62011-07-23 00:29:16 +0000398
Dan Gohman3dc2d922008-05-14 00:24:14 +0000399 // Remap all instructions in the most recent iteration
Dan Gohman04c8bd72008-06-24 20:44:42 +0000400 for (unsigned i = 0; i < NewBlocks.size(); ++i)
Dan Gohman3dc2d922008-05-14 00:24:14 +0000401 for (BasicBlock::iterator I = NewBlocks[i]->begin(),
402 E = NewBlocks[i]->end(); I != E; ++I)
Duncan P. N. Exon Smith5b4c8372015-10-13 02:39:05 +0000403 ::RemapInstruction(&*I, LastValueMap);
Dan Gohman3dc2d922008-05-14 00:24:14 +0000404 }
Andrew Trick279e7a62011-07-23 00:29:16 +0000405
Andrew Trickb72bbe22011-08-10 00:28:10 +0000406 // Loop over the PHI nodes in the original block, setting incoming values.
407 for (unsigned i = 0, e = OrigPHINode.size(); i != e; ++i) {
408 PHINode *PN = OrigPHINode[i];
409 if (CompletelyUnroll) {
Dan Gohman04c8bd72008-06-24 20:44:42 +0000410 PN->replaceAllUsesWith(PN->getIncomingValueForBlock(Preheader));
411 Header->getInstList().erase(PN);
412 }
Andrew Trickb72bbe22011-08-10 00:28:10 +0000413 else if (Count > 1) {
414 Value *InVal = PN->removeIncomingValue(LatchBlock, false);
415 // If this value was defined in the loop, take the value defined by the
416 // last iteration of the loop.
417 if (Instruction *InValI = dyn_cast<Instruction>(InVal)) {
418 if (L->contains(InValI))
419 InVal = LastValueMap[InVal];
420 }
421 assert(Latches.back() == LastValueMap[LatchBlock] && "bad last latch");
422 PN->addIncoming(InVal, Latches.back());
423 }
Dan Gohman04c8bd72008-06-24 20:44:42 +0000424 }
Dan Gohman3dc2d922008-05-14 00:24:14 +0000425
426 // Now that all the basic blocks for the unrolled iterations are in place,
427 // set up the branches to connect them.
Dan Gohman04c8bd72008-06-24 20:44:42 +0000428 for (unsigned i = 0, e = Latches.size(); i != e; ++i) {
Dan Gohman3dc2d922008-05-14 00:24:14 +0000429 // The original branch was replicated in each unrolled iteration.
Dan Gohman04c8bd72008-06-24 20:44:42 +0000430 BranchInst *Term = cast<BranchInst>(Latches[i]->getTerminator());
Dan Gohman3dc2d922008-05-14 00:24:14 +0000431
432 // The branch destination.
Dan Gohman04c8bd72008-06-24 20:44:42 +0000433 unsigned j = (i + 1) % e;
434 BasicBlock *Dest = Headers[j];
Dan Gohman3dc2d922008-05-14 00:24:14 +0000435 bool NeedConditional = true;
436
Andrew Trickd04d15292011-12-09 06:19:40 +0000437 if (RuntimeTripCount && j != 0) {
438 NeedConditional = false;
439 }
440
Dan Gohman04c8bd72008-06-24 20:44:42 +0000441 // For a complete unroll, make the last iteration end with a branch
442 // to the exit block.
Michael Zolotukhind56ee062015-09-23 23:12:43 +0000443 if (CompletelyUnroll) {
444 if (j == 0)
445 Dest = LoopExit;
Dan Gohman3dc2d922008-05-14 00:24:14 +0000446 NeedConditional = false;
447 }
448
449 // If we know the trip count or a multiple of it, we can safely use an
450 // unconditional branch for some iterations.
Dan Gohman04c8bd72008-06-24 20:44:42 +0000451 if (j != BreakoutTrip && (TripMultiple == 0 || j % TripMultiple != 0)) {
Dan Gohman3dc2d922008-05-14 00:24:14 +0000452 NeedConditional = false;
453 }
454
455 if (NeedConditional) {
456 // Update the conditional branch's successor for the following
457 // iteration.
458 Term->setSuccessor(!ContinueOnTrue, Dest);
459 } else {
Andrew Trickb72bbe22011-08-10 00:28:10 +0000460 // Remove phi operands at this loop exit
461 if (Dest != LoopExit) {
462 BasicBlock *BB = Latches[i];
Duncan P. N. Exon Smith6c990152014-07-21 17:06:51 +0000463 for (succ_iterator SI = succ_begin(BB), SE = succ_end(BB);
464 SI != SE; ++SI) {
465 if (*SI == Headers[i])
Andrew Trickb72bbe22011-08-10 00:28:10 +0000466 continue;
Duncan P. N. Exon Smith6c990152014-07-21 17:06:51 +0000467 for (BasicBlock::iterator BBI = (*SI)->begin();
Andrew Trickb72bbe22011-08-10 00:28:10 +0000468 PHINode *Phi = dyn_cast<PHINode>(BBI); ++BBI) {
469 Phi->removeIncomingValue(BB, false);
470 }
471 }
472 }
Jay Foad89afb432011-01-07 20:25:56 +0000473 // Replace the conditional branch with an unconditional one.
474 BranchInst::Create(Dest, Term);
475 Term->eraseFromParent();
Jay Foada97a2c92011-06-21 10:33:19 +0000476 }
477 }
478
Jay Foad61ea0e42011-06-23 09:09:15 +0000479 // Merge adjacent basic blocks, if possible.
Mark Heffernan675d4012014-07-10 23:30:06 +0000480 SmallPtrSet<Loop *, 4> ForgottenLoops;
Jay Foad61ea0e42011-06-23 09:09:15 +0000481 for (unsigned i = 0, e = Latches.size(); i != e; ++i) {
482 BranchInst *Term = cast<BranchInst>(Latches[i]->getTerminator());
483 if (Term->isUnconditional()) {
484 BasicBlock *Dest = Term->getSuccessor(0);
Mark Heffernan675d4012014-07-10 23:30:06 +0000485 if (BasicBlock *Fold = FoldBlockIntoPredecessor(Dest, LI, LPM,
486 ForgottenLoops))
Jay Foad61ea0e42011-06-23 09:09:15 +0000487 std::replace(Latches.begin(), Latches.end(), Dest, Fold);
488 }
489 }
Andrew Trick279e7a62011-07-23 00:29:16 +0000490
Hal Finkel74c2f352014-09-07 12:44:26 +0000491 // FIXME: We could register any cloned assumptions instead of clearing the
492 // whole function's cache.
Chandler Carruth66b31302015-01-04 12:03:27 +0000493 AC->clear();
Hal Finkel74c2f352014-09-07 12:44:26 +0000494
Craig Topperf40110f2014-04-25 05:29:35 +0000495 DominatorTree *DT = nullptr;
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +0000496 if (PP) {
Andrew Trickd29cd732012-05-08 02:52:09 +0000497 // FIXME: Reconstruct dom info, because it is not preserved properly.
498 // Incrementally updating domtree after loop unrolling would be easy.
Chandler Carruth73523022014-01-13 13:07:17 +0000499 if (DominatorTreeWrapperPass *DTWP =
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +0000500 PP->getAnalysisIfAvailable<DominatorTreeWrapperPass>()) {
501 DT = &DTWP->getDomTree();
502 DT->recalculate(*L->getHeader()->getParent());
503 }
Andrew Trick4d0040b2011-08-10 04:29:49 +0000504
Andrew Trickd29cd732012-05-08 02:52:09 +0000505 // Simplify any new induction variables in the partially unrolled loop.
Andrew Trickd29cd732012-05-08 02:52:09 +0000506 if (SE && !CompletelyUnroll) {
507 SmallVector<WeakVH, 16> DeadInsts;
Sanjoy Das5c8bead2015-10-06 21:44:49 +0000508 simplifyLoopIVs(L, SE, DT, LPM, DeadInsts);
Andrew Trick4d0040b2011-08-10 04:29:49 +0000509
Andrew Trickd29cd732012-05-08 02:52:09 +0000510 // Aggressively clean up dead instructions that simplifyLoopIVs already
511 // identified. Any remaining should be cleaned up below.
512 while (!DeadInsts.empty())
513 if (Instruction *Inst =
514 dyn_cast_or_null<Instruction>(&*DeadInsts.pop_back_val()))
515 RecursivelyDeleteTriviallyDeadInstructions(Inst);
516 }
Andrew Trick4d0040b2011-08-10 04:29:49 +0000517 }
Dan Gohman04c8bd72008-06-24 20:44:42 +0000518 // At this point, the code is well formed. We now do a quick sweep over the
519 // inserted code, doing constant propagation and dead code elimination as we
520 // go.
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000521 const DataLayout &DL = Header->getModule()->getDataLayout();
Dan Gohman04c8bd72008-06-24 20:44:42 +0000522 const std::vector<BasicBlock*> &NewLoopBlocks = L->getBlocks();
523 for (std::vector<BasicBlock*>::const_iterator BB = NewLoopBlocks.begin(),
524 BBE = NewLoopBlocks.end(); BB != BBE; ++BB)
525 for (BasicBlock::iterator I = (*BB)->begin(), E = (*BB)->end(); I != E; ) {
Duncan P. N. Exon Smith5b4c8372015-10-13 02:39:05 +0000526 Instruction *Inst = &*I++;
Dan Gohman3dc2d922008-05-14 00:24:14 +0000527
528 if (isInstructionTriviallyDead(Inst))
Dan Gohman04c8bd72008-06-24 20:44:42 +0000529 (*BB)->getInstList().erase(Inst);
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000530 else if (Value *V = SimplifyInstruction(Inst, DL))
Duncan Sands433c1672010-11-23 20:26:33 +0000531 if (LI->replacementPreservesLCSSAForm(Inst, V)) {
532 Inst->replaceAllUsesWith(V);
533 (*BB)->getInstList().erase(Inst);
534 }
Dan Gohman3dc2d922008-05-14 00:24:14 +0000535 }
Dan Gohman3dc2d922008-05-14 00:24:14 +0000536
537 NumCompletelyUnrolled += CompletelyUnroll;
538 ++NumUnrolled;
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +0000539
540 Loop *OuterL = L->getParentLoop();
Dan Gohman3dc2d922008-05-14 00:24:14 +0000541 // Remove the loop from the LoopPassManager if it's completely removed.
Craig Topperf40110f2014-04-25 05:29:35 +0000542 if (CompletelyUnroll && LPM != nullptr)
Dan Gohman3dc2d922008-05-14 00:24:14 +0000543 LPM->deleteLoopFromQueue(L);
544
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +0000545 // If we have a pass and a DominatorTree we should re-simplify impacted loops
546 // to ensure subsequent analyses can rely on this form. We want to simplify
547 // at least one layer outside of the loop that was unrolled so that any
548 // changes to the parent loop exposed by the unrolling are considered.
549 if (PP && DT) {
550 if (!OuterL && !CompletelyUnroll)
551 OuterL = L;
Chandler Carruthd84f7762014-01-28 01:25:38 +0000552 if (OuterL) {
Michael Zolotukhin78760ee2015-12-09 18:20:28 +0000553 bool Simplified = simplifyLoop(OuterL, DT, LI, PP, SE, AC);
Dinesh Dwivedid266cb12014-05-29 06:47:23 +0000554
555 // LCSSA must be performed on the outermost affected loop. The unrolled
556 // loop's last loop latch is guaranteed to be in the outermost loop after
557 // deleteLoopFromQueue updates LoopInfo.
558 Loop *LatchLoop = LI->getLoopFor(Latches.back());
559 if (!OuterL->contains(LatchLoop))
560 while (OuterL->getParentLoop() != LatchLoop)
561 OuterL = OuterL->getParentLoop();
562
Michael Zolotukhin78760ee2015-12-09 18:20:28 +0000563 if (CompletelyUnroll && (!AllExitsAreInsideParentLoop || Simplified))
564 formLCSSARecursively(*OuterL, *DT, LI, SE);
565 else
566 assert(OuterL->isLCSSAForm(*DT) &&
567 "Loops should be in LCSSA form after loop-unroll.");
Chandler Carruthd84f7762014-01-28 01:25:38 +0000568 }
Chandler Carruthaa7fa5e2014-01-23 11:23:19 +0000569 }
570
Dan Gohman3dc2d922008-05-14 00:24:14 +0000571 return true;
572}
Jingyue Wu0220df02015-02-01 02:27:45 +0000573
574/// Given an llvm.loop loop id metadata node, returns the loop hint metadata
575/// node with the given name (for example, "llvm.loop.unroll.count"). If no
576/// such metadata node exists, then nullptr is returned.
Jingyue Wu49a766e2015-02-02 20:41:11 +0000577MDNode *llvm::GetUnrollMetadata(MDNode *LoopID, StringRef Name) {
Jingyue Wu0220df02015-02-01 02:27:45 +0000578 // First operand should refer to the loop id itself.
579 assert(LoopID->getNumOperands() > 0 && "requires at least one operand");
Jingyue Wu49a766e2015-02-02 20:41:11 +0000580 assert(LoopID->getOperand(0) == LoopID && "invalid loop id");
Jingyue Wu0220df02015-02-01 02:27:45 +0000581
582 for (unsigned i = 1, e = LoopID->getNumOperands(); i < e; ++i) {
Jingyue Wu49a766e2015-02-02 20:41:11 +0000583 MDNode *MD = dyn_cast<MDNode>(LoopID->getOperand(i));
Jingyue Wu0220df02015-02-01 02:27:45 +0000584 if (!MD)
585 continue;
586
Jingyue Wu49a766e2015-02-02 20:41:11 +0000587 MDString *S = dyn_cast<MDString>(MD->getOperand(0));
Jingyue Wu0220df02015-02-01 02:27:45 +0000588 if (!S)
589 continue;
590
591 if (Name.equals(S->getString()))
592 return MD;
593 }
594 return nullptr;
595}