1b0aa36f9SDavid Green //===----------------- LoopRotationUtils.cpp -----------------------------===//
2b0aa36f9SDavid Green //
3b0aa36f9SDavid Green //                     The LLVM Compiler Infrastructure
4b0aa36f9SDavid Green //
5b0aa36f9SDavid Green // This file is distributed under the University of Illinois Open Source
6b0aa36f9SDavid Green // License. See LICENSE.TXT for details.
7b0aa36f9SDavid Green //
8b0aa36f9SDavid Green //===----------------------------------------------------------------------===//
9b0aa36f9SDavid Green //
10b0aa36f9SDavid Green // This file provides utilities to convert a loop into a loop with bottom test.
11b0aa36f9SDavid Green //
12b0aa36f9SDavid Green //===----------------------------------------------------------------------===//
13b0aa36f9SDavid Green 
14b0aa36f9SDavid Green #include "llvm/Transforms/Utils/LoopRotationUtils.h"
15b0aa36f9SDavid Green #include "llvm/ADT/Statistic.h"
16b0aa36f9SDavid Green #include "llvm/Analysis/AliasAnalysis.h"
17b0aa36f9SDavid Green #include "llvm/Analysis/AssumptionCache.h"
18b0aa36f9SDavid Green #include "llvm/Analysis/BasicAliasAnalysis.h"
19b0aa36f9SDavid Green #include "llvm/Analysis/CodeMetrics.h"
20b0aa36f9SDavid Green #include "llvm/Analysis/GlobalsModRef.h"
21b0aa36f9SDavid Green #include "llvm/Analysis/InstructionSimplify.h"
22b0aa36f9SDavid Green #include "llvm/Analysis/LoopPass.h"
23b0aa36f9SDavid Green #include "llvm/Analysis/ScalarEvolution.h"
24b0aa36f9SDavid Green #include "llvm/Analysis/ScalarEvolutionAliasAnalysis.h"
25b0aa36f9SDavid Green #include "llvm/Analysis/TargetTransformInfo.h"
26b0aa36f9SDavid Green #include "llvm/Analysis/Utils/Local.h"
27b0aa36f9SDavid Green #include "llvm/Analysis/ValueTracking.h"
28b0aa36f9SDavid Green #include "llvm/IR/CFG.h"
29b0aa36f9SDavid Green #include "llvm/IR/DebugInfoMetadata.h"
30b0aa36f9SDavid Green #include "llvm/IR/Dominators.h"
31b0aa36f9SDavid Green #include "llvm/IR/Function.h"
32b0aa36f9SDavid Green #include "llvm/IR/IntrinsicInst.h"
33b0aa36f9SDavid Green #include "llvm/IR/Module.h"
34b0aa36f9SDavid Green #include "llvm/Support/CommandLine.h"
35b0aa36f9SDavid Green #include "llvm/Support/Debug.h"
36b0aa36f9SDavid Green #include "llvm/Support/raw_ostream.h"
37b0aa36f9SDavid Green #include "llvm/Transforms/Utils/BasicBlockUtils.h"
38b0aa36f9SDavid Green #include "llvm/Transforms/Utils/LoopUtils.h"
39b0aa36f9SDavid Green #include "llvm/Transforms/Utils/SSAUpdater.h"
40b0aa36f9SDavid Green #include "llvm/Transforms/Utils/ValueMapper.h"
41b0aa36f9SDavid Green using namespace llvm;
42b0aa36f9SDavid Green 
43b0aa36f9SDavid Green #define DEBUG_TYPE "loop-rotate"
44b0aa36f9SDavid Green 
45b0aa36f9SDavid Green STATISTIC(NumRotated, "Number of loops rotated");
46b0aa36f9SDavid Green 
47b0aa36f9SDavid Green namespace {
48b0aa36f9SDavid Green /// A simple loop rotation transformation.
49b0aa36f9SDavid Green class LoopRotate {
50b0aa36f9SDavid Green   const unsigned MaxHeaderSize;
51b0aa36f9SDavid Green   LoopInfo *LI;
52b0aa36f9SDavid Green   const TargetTransformInfo *TTI;
53b0aa36f9SDavid Green   AssumptionCache *AC;
54b0aa36f9SDavid Green   DominatorTree *DT;
55b0aa36f9SDavid Green   ScalarEvolution *SE;
56b0aa36f9SDavid Green   const SimplifyQuery &SQ;
57b0aa36f9SDavid Green 
58b0aa36f9SDavid Green public:
59b0aa36f9SDavid Green   LoopRotate(unsigned MaxHeaderSize, LoopInfo *LI,
60b0aa36f9SDavid Green              const TargetTransformInfo *TTI, AssumptionCache *AC,
61b0aa36f9SDavid Green              DominatorTree *DT, ScalarEvolution *SE, const SimplifyQuery &SQ)
62b0aa36f9SDavid Green       : MaxHeaderSize(MaxHeaderSize), LI(LI), TTI(TTI), AC(AC), DT(DT), SE(SE),
63b0aa36f9SDavid Green         SQ(SQ) {}
64b0aa36f9SDavid Green   bool processLoop(Loop *L);
65b0aa36f9SDavid Green 
66b0aa36f9SDavid Green private:
67b0aa36f9SDavid Green   bool rotateLoop(Loop *L, bool SimplifiedLatch);
68b0aa36f9SDavid Green   bool simplifyLoopLatch(Loop *L);
69b0aa36f9SDavid Green };
70b0aa36f9SDavid Green } // end anonymous namespace
71b0aa36f9SDavid Green 
72b0aa36f9SDavid Green /// RewriteUsesOfClonedInstructions - We just cloned the instructions from the
73b0aa36f9SDavid Green /// old header into the preheader.  If there were uses of the values produced by
74b0aa36f9SDavid Green /// these instruction that were outside of the loop, we have to insert PHI nodes
75b0aa36f9SDavid Green /// to merge the two values.  Do this now.
76b0aa36f9SDavid Green static void RewriteUsesOfClonedInstructions(BasicBlock *OrigHeader,
77b0aa36f9SDavid Green                                             BasicBlock *OrigPreheader,
78b0aa36f9SDavid Green                                             ValueToValueMapTy &ValueMap,
79b0aa36f9SDavid Green                                 SmallVectorImpl<PHINode*> *InsertedPHIs) {
80b0aa36f9SDavid Green   // Remove PHI node entries that are no longer live.
81b0aa36f9SDavid Green   BasicBlock::iterator I, E = OrigHeader->end();
82b0aa36f9SDavid Green   for (I = OrigHeader->begin(); PHINode *PN = dyn_cast<PHINode>(I); ++I)
83b0aa36f9SDavid Green     PN->removeIncomingValue(PN->getBasicBlockIndex(OrigPreheader));
84b0aa36f9SDavid Green 
85b0aa36f9SDavid Green   // Now fix up users of the instructions in OrigHeader, inserting PHI nodes
86b0aa36f9SDavid Green   // as necessary.
87b0aa36f9SDavid Green   SSAUpdater SSA(InsertedPHIs);
88b0aa36f9SDavid Green   for (I = OrigHeader->begin(); I != E; ++I) {
89b0aa36f9SDavid Green     Value *OrigHeaderVal = &*I;
90b0aa36f9SDavid Green 
91b0aa36f9SDavid Green     // If there are no uses of the value (e.g. because it returns void), there
92b0aa36f9SDavid Green     // is nothing to rewrite.
93b0aa36f9SDavid Green     if (OrigHeaderVal->use_empty())
94b0aa36f9SDavid Green       continue;
95b0aa36f9SDavid Green 
96b0aa36f9SDavid Green     Value *OrigPreHeaderVal = ValueMap.lookup(OrigHeaderVal);
97b0aa36f9SDavid Green 
98b0aa36f9SDavid Green     // The value now exits in two versions: the initial value in the preheader
99b0aa36f9SDavid Green     // and the loop "next" value in the original header.
100b0aa36f9SDavid Green     SSA.Initialize(OrigHeaderVal->getType(), OrigHeaderVal->getName());
101b0aa36f9SDavid Green     SSA.AddAvailableValue(OrigHeader, OrigHeaderVal);
102b0aa36f9SDavid Green     SSA.AddAvailableValue(OrigPreheader, OrigPreHeaderVal);
103b0aa36f9SDavid Green 
104b0aa36f9SDavid Green     // Visit each use of the OrigHeader instruction.
105b0aa36f9SDavid Green     for (Value::use_iterator UI = OrigHeaderVal->use_begin(),
106b0aa36f9SDavid Green                              UE = OrigHeaderVal->use_end();
107b0aa36f9SDavid Green          UI != UE;) {
108b0aa36f9SDavid Green       // Grab the use before incrementing the iterator.
109b0aa36f9SDavid Green       Use &U = *UI;
110b0aa36f9SDavid Green 
111b0aa36f9SDavid Green       // Increment the iterator before removing the use from the list.
112b0aa36f9SDavid Green       ++UI;
113b0aa36f9SDavid Green 
114b0aa36f9SDavid Green       // SSAUpdater can't handle a non-PHI use in the same block as an
115b0aa36f9SDavid Green       // earlier def. We can easily handle those cases manually.
116b0aa36f9SDavid Green       Instruction *UserInst = cast<Instruction>(U.getUser());
117b0aa36f9SDavid Green       if (!isa<PHINode>(UserInst)) {
118b0aa36f9SDavid Green         BasicBlock *UserBB = UserInst->getParent();
119b0aa36f9SDavid Green 
120b0aa36f9SDavid Green         // The original users in the OrigHeader are already using the
121b0aa36f9SDavid Green         // original definitions.
122b0aa36f9SDavid Green         if (UserBB == OrigHeader)
123b0aa36f9SDavid Green           continue;
124b0aa36f9SDavid Green 
125b0aa36f9SDavid Green         // Users in the OrigPreHeader need to use the value to which the
126b0aa36f9SDavid Green         // original definitions are mapped.
127b0aa36f9SDavid Green         if (UserBB == OrigPreheader) {
128b0aa36f9SDavid Green           U = OrigPreHeaderVal;
129b0aa36f9SDavid Green           continue;
130b0aa36f9SDavid Green         }
131b0aa36f9SDavid Green       }
132b0aa36f9SDavid Green 
133b0aa36f9SDavid Green       // Anything else can be handled by SSAUpdater.
134b0aa36f9SDavid Green       SSA.RewriteUse(U);
135b0aa36f9SDavid Green     }
136b0aa36f9SDavid Green 
137b0aa36f9SDavid Green     // Replace MetadataAsValue(ValueAsMetadata(OrigHeaderVal)) uses in debug
138b0aa36f9SDavid Green     // intrinsics.
139b0aa36f9SDavid Green     SmallVector<DbgValueInst *, 1> DbgValues;
140b0aa36f9SDavid Green     llvm::findDbgValues(DbgValues, OrigHeaderVal);
141b0aa36f9SDavid Green     for (auto &DbgValue : DbgValues) {
142b0aa36f9SDavid Green       // The original users in the OrigHeader are already using the original
143b0aa36f9SDavid Green       // definitions.
144b0aa36f9SDavid Green       BasicBlock *UserBB = DbgValue->getParent();
145b0aa36f9SDavid Green       if (UserBB == OrigHeader)
146b0aa36f9SDavid Green         continue;
147b0aa36f9SDavid Green 
148b0aa36f9SDavid Green       // Users in the OrigPreHeader need to use the value to which the
149b0aa36f9SDavid Green       // original definitions are mapped and anything else can be handled by
150b0aa36f9SDavid Green       // the SSAUpdater. To avoid adding PHINodes, check if the value is
151b0aa36f9SDavid Green       // available in UserBB, if not substitute undef.
152b0aa36f9SDavid Green       Value *NewVal;
153b0aa36f9SDavid Green       if (UserBB == OrigPreheader)
154b0aa36f9SDavid Green         NewVal = OrigPreHeaderVal;
155b0aa36f9SDavid Green       else if (SSA.HasValueForBlock(UserBB))
156b0aa36f9SDavid Green         NewVal = SSA.GetValueInMiddleOfBlock(UserBB);
157b0aa36f9SDavid Green       else
158b0aa36f9SDavid Green         NewVal = UndefValue::get(OrigHeaderVal->getType());
159b0aa36f9SDavid Green       DbgValue->setOperand(0,
160b0aa36f9SDavid Green                            MetadataAsValue::get(OrigHeaderVal->getContext(),
161b0aa36f9SDavid Green                                                 ValueAsMetadata::get(NewVal)));
162b0aa36f9SDavid Green     }
163b0aa36f9SDavid Green   }
164b0aa36f9SDavid Green }
165b0aa36f9SDavid Green 
166*f80ebc8dSDavid Green // Look for a phi which is only used outside the loop (via a LCSSA phi)
167*f80ebc8dSDavid Green // in the exit from the header. This means that rotating the loop can
168*f80ebc8dSDavid Green // remove the phi.
169*f80ebc8dSDavid Green static bool shouldRotateLoopExitingLatch(Loop *L) {
170*f80ebc8dSDavid Green   BasicBlock *Header = L->getHeader();
171*f80ebc8dSDavid Green   BasicBlock *HeaderExit = Header->getTerminator()->getSuccessor(0);
172*f80ebc8dSDavid Green   if (L->contains(HeaderExit))
173*f80ebc8dSDavid Green     HeaderExit = Header->getTerminator()->getSuccessor(1);
174*f80ebc8dSDavid Green 
175*f80ebc8dSDavid Green   for (auto &Phi : Header->phis()) {
176*f80ebc8dSDavid Green     // Look for uses of this phi in the loop/via exits other than the header.
177*f80ebc8dSDavid Green     if (llvm::any_of(Phi.users(), [HeaderExit](const User *U) {
178*f80ebc8dSDavid Green           return cast<Instruction>(U)->getParent() != HeaderExit;
179*f80ebc8dSDavid Green         }))
180*f80ebc8dSDavid Green       continue;
181*f80ebc8dSDavid Green     return true;
182*f80ebc8dSDavid Green   }
183*f80ebc8dSDavid Green 
184*f80ebc8dSDavid Green   return false;
185*f80ebc8dSDavid Green }
186*f80ebc8dSDavid Green 
187b0aa36f9SDavid Green /// Rotate loop LP. Return true if the loop is rotated.
188b0aa36f9SDavid Green ///
189b0aa36f9SDavid Green /// \param SimplifiedLatch is true if the latch was just folded into the final
190b0aa36f9SDavid Green /// loop exit. In this case we may want to rotate even though the new latch is
191b0aa36f9SDavid Green /// now an exiting branch. This rotation would have happened had the latch not
192b0aa36f9SDavid Green /// been simplified. However, if SimplifiedLatch is false, then we avoid
193b0aa36f9SDavid Green /// rotating loops in which the latch exits to avoid excessive or endless
194b0aa36f9SDavid Green /// rotation. LoopRotate should be repeatable and converge to a canonical
195b0aa36f9SDavid Green /// form. This property is satisfied because simplifying the loop latch can only
196b0aa36f9SDavid Green /// happen once across multiple invocations of the LoopRotate pass.
197b0aa36f9SDavid Green bool LoopRotate::rotateLoop(Loop *L, bool SimplifiedLatch) {
198b0aa36f9SDavid Green   // If the loop has only one block then there is not much to rotate.
199b0aa36f9SDavid Green   if (L->getBlocks().size() == 1)
200b0aa36f9SDavid Green     return false;
201b0aa36f9SDavid Green 
202b0aa36f9SDavid Green   BasicBlock *OrigHeader = L->getHeader();
203b0aa36f9SDavid Green   BasicBlock *OrigLatch = L->getLoopLatch();
204b0aa36f9SDavid Green 
205b0aa36f9SDavid Green   BranchInst *BI = dyn_cast<BranchInst>(OrigHeader->getTerminator());
206b0aa36f9SDavid Green   if (!BI || BI->isUnconditional())
207b0aa36f9SDavid Green     return false;
208b0aa36f9SDavid Green 
209b0aa36f9SDavid Green   // If the loop header is not one of the loop exiting blocks then
210b0aa36f9SDavid Green   // either this loop is already rotated or it is not
211b0aa36f9SDavid Green   // suitable for loop rotation transformations.
212b0aa36f9SDavid Green   if (!L->isLoopExiting(OrigHeader))
213b0aa36f9SDavid Green     return false;
214b0aa36f9SDavid Green 
215b0aa36f9SDavid Green   // If the loop latch already contains a branch that leaves the loop then the
216b0aa36f9SDavid Green   // loop is already rotated.
217b0aa36f9SDavid Green   if (!OrigLatch)
218b0aa36f9SDavid Green     return false;
219b0aa36f9SDavid Green 
220b0aa36f9SDavid Green   // Rotate if either the loop latch does *not* exit the loop, or if the loop
221*f80ebc8dSDavid Green   // latch was just simplified. Or if we think it will be profitable.
222*f80ebc8dSDavid Green   if (L->isLoopExiting(OrigLatch) && !SimplifiedLatch &&
223*f80ebc8dSDavid Green       !shouldRotateLoopExitingLatch(L))
224b0aa36f9SDavid Green     return false;
225b0aa36f9SDavid Green 
226b0aa36f9SDavid Green   // Check size of original header and reject loop if it is very big or we can't
227b0aa36f9SDavid Green   // duplicate blocks inside it.
228b0aa36f9SDavid Green   {
229b0aa36f9SDavid Green     SmallPtrSet<const Value *, 32> EphValues;
230b0aa36f9SDavid Green     CodeMetrics::collectEphemeralValues(L, AC, EphValues);
231b0aa36f9SDavid Green 
232b0aa36f9SDavid Green     CodeMetrics Metrics;
233b0aa36f9SDavid Green     Metrics.analyzeBasicBlock(OrigHeader, *TTI, EphValues);
234b0aa36f9SDavid Green     if (Metrics.notDuplicatable) {
235b0aa36f9SDavid Green       DEBUG(dbgs() << "LoopRotation: NOT rotating - contains non-duplicatable"
236b0aa36f9SDavid Green                    << " instructions: ";
237b0aa36f9SDavid Green             L->dump());
238b0aa36f9SDavid Green       return false;
239b0aa36f9SDavid Green     }
240b0aa36f9SDavid Green     if (Metrics.convergent) {
241b0aa36f9SDavid Green       DEBUG(dbgs() << "LoopRotation: NOT rotating - contains convergent "
242b0aa36f9SDavid Green                       "instructions: ";
243b0aa36f9SDavid Green             L->dump());
244b0aa36f9SDavid Green       return false;
245b0aa36f9SDavid Green     }
246b0aa36f9SDavid Green     if (Metrics.NumInsts > MaxHeaderSize)
247b0aa36f9SDavid Green       return false;
248b0aa36f9SDavid Green   }
249b0aa36f9SDavid Green 
250b0aa36f9SDavid Green   // Now, this loop is suitable for rotation.
251b0aa36f9SDavid Green   BasicBlock *OrigPreheader = L->getLoopPreheader();
252b0aa36f9SDavid Green 
253b0aa36f9SDavid Green   // If the loop could not be converted to canonical form, it must have an
254b0aa36f9SDavid Green   // indirectbr in it, just give up.
255b0aa36f9SDavid Green   if (!OrigPreheader || !L->hasDedicatedExits())
256b0aa36f9SDavid Green     return false;
257b0aa36f9SDavid Green 
258b0aa36f9SDavid Green   // Anything ScalarEvolution may know about this loop or the PHI nodes
259b0aa36f9SDavid Green   // in its header will soon be invalidated.
260b0aa36f9SDavid Green   if (SE)
261b0aa36f9SDavid Green     SE->forgetLoop(L);
262b0aa36f9SDavid Green 
263b0aa36f9SDavid Green   DEBUG(dbgs() << "LoopRotation: rotating "; L->dump());
264b0aa36f9SDavid Green 
265b0aa36f9SDavid Green   // Find new Loop header. NewHeader is a Header's one and only successor
266b0aa36f9SDavid Green   // that is inside loop.  Header's other successor is outside the
267b0aa36f9SDavid Green   // loop.  Otherwise loop is not suitable for rotation.
268b0aa36f9SDavid Green   BasicBlock *Exit = BI->getSuccessor(0);
269b0aa36f9SDavid Green   BasicBlock *NewHeader = BI->getSuccessor(1);
270b0aa36f9SDavid Green   if (L->contains(Exit))
271b0aa36f9SDavid Green     std::swap(Exit, NewHeader);
272b0aa36f9SDavid Green   assert(NewHeader && "Unable to determine new loop header");
273b0aa36f9SDavid Green   assert(L->contains(NewHeader) && !L->contains(Exit) &&
274b0aa36f9SDavid Green          "Unable to determine loop header and exit blocks");
275b0aa36f9SDavid Green 
276b0aa36f9SDavid Green   // This code assumes that the new header has exactly one predecessor.
277b0aa36f9SDavid Green   // Remove any single-entry PHI nodes in it.
278b0aa36f9SDavid Green   assert(NewHeader->getSinglePredecessor() &&
279b0aa36f9SDavid Green          "New header doesn't have one pred!");
280b0aa36f9SDavid Green   FoldSingleEntryPHINodes(NewHeader);
281b0aa36f9SDavid Green 
282b0aa36f9SDavid Green   // Begin by walking OrigHeader and populating ValueMap with an entry for
283b0aa36f9SDavid Green   // each Instruction.
284b0aa36f9SDavid Green   BasicBlock::iterator I = OrigHeader->begin(), E = OrigHeader->end();
285b0aa36f9SDavid Green   ValueToValueMapTy ValueMap;
286b0aa36f9SDavid Green 
287b0aa36f9SDavid Green   // For PHI nodes, the value available in OldPreHeader is just the
288b0aa36f9SDavid Green   // incoming value from OldPreHeader.
289b0aa36f9SDavid Green   for (; PHINode *PN = dyn_cast<PHINode>(I); ++I)
290b0aa36f9SDavid Green     ValueMap[PN] = PN->getIncomingValueForBlock(OrigPreheader);
291b0aa36f9SDavid Green 
292b0aa36f9SDavid Green   // For the rest of the instructions, either hoist to the OrigPreheader if
293b0aa36f9SDavid Green   // possible or create a clone in the OldPreHeader if not.
294b0aa36f9SDavid Green   TerminatorInst *LoopEntryBranch = OrigPreheader->getTerminator();
295b0aa36f9SDavid Green 
296b0aa36f9SDavid Green   // Record all debug intrinsics preceding LoopEntryBranch to avoid duplication.
297b0aa36f9SDavid Green   using DbgIntrinsicHash =
298b0aa36f9SDavid Green       std::pair<std::pair<Value *, DILocalVariable *>, DIExpression *>;
299b0aa36f9SDavid Green   auto makeHash = [](DbgInfoIntrinsic *D) -> DbgIntrinsicHash {
300b0aa36f9SDavid Green     return {{D->getVariableLocation(), D->getVariable()}, D->getExpression()};
301b0aa36f9SDavid Green   };
302b0aa36f9SDavid Green   SmallDenseSet<DbgIntrinsicHash, 8> DbgIntrinsics;
303b0aa36f9SDavid Green   for (auto I = std::next(OrigPreheader->rbegin()), E = OrigPreheader->rend();
304b0aa36f9SDavid Green        I != E; ++I) {
305b0aa36f9SDavid Green     if (auto *DII = dyn_cast<DbgInfoIntrinsic>(&*I))
306b0aa36f9SDavid Green       DbgIntrinsics.insert(makeHash(DII));
307b0aa36f9SDavid Green     else
308b0aa36f9SDavid Green       break;
309b0aa36f9SDavid Green   }
310b0aa36f9SDavid Green 
311b0aa36f9SDavid Green   while (I != E) {
312b0aa36f9SDavid Green     Instruction *Inst = &*I++;
313b0aa36f9SDavid Green 
314b0aa36f9SDavid Green     // If the instruction's operands are invariant and it doesn't read or write
315b0aa36f9SDavid Green     // memory, then it is safe to hoist.  Doing this doesn't change the order of
316b0aa36f9SDavid Green     // execution in the preheader, but does prevent the instruction from
317b0aa36f9SDavid Green     // executing in each iteration of the loop.  This means it is safe to hoist
318b0aa36f9SDavid Green     // something that might trap, but isn't safe to hoist something that reads
319b0aa36f9SDavid Green     // memory (without proving that the loop doesn't write).
320b0aa36f9SDavid Green     if (L->hasLoopInvariantOperands(Inst) && !Inst->mayReadFromMemory() &&
321b0aa36f9SDavid Green         !Inst->mayWriteToMemory() && !isa<TerminatorInst>(Inst) &&
322b0aa36f9SDavid Green         !isa<DbgInfoIntrinsic>(Inst) && !isa<AllocaInst>(Inst)) {
323b0aa36f9SDavid Green       Inst->moveBefore(LoopEntryBranch);
324b0aa36f9SDavid Green       continue;
325b0aa36f9SDavid Green     }
326b0aa36f9SDavid Green 
327b0aa36f9SDavid Green     // Otherwise, create a duplicate of the instruction.
328b0aa36f9SDavid Green     Instruction *C = Inst->clone();
329b0aa36f9SDavid Green 
330b0aa36f9SDavid Green     // Eagerly remap the operands of the instruction.
331b0aa36f9SDavid Green     RemapInstruction(C, ValueMap,
332b0aa36f9SDavid Green                      RF_NoModuleLevelChanges | RF_IgnoreMissingLocals);
333b0aa36f9SDavid Green 
334b0aa36f9SDavid Green     // Avoid inserting the same intrinsic twice.
335b0aa36f9SDavid Green     if (auto *DII = dyn_cast<DbgInfoIntrinsic>(C))
336b0aa36f9SDavid Green       if (DbgIntrinsics.count(makeHash(DII))) {
337b0aa36f9SDavid Green         C->deleteValue();
338b0aa36f9SDavid Green         continue;
339b0aa36f9SDavid Green       }
340b0aa36f9SDavid Green 
341b0aa36f9SDavid Green     // With the operands remapped, see if the instruction constant folds or is
342b0aa36f9SDavid Green     // otherwise simplifyable.  This commonly occurs because the entry from PHI
343b0aa36f9SDavid Green     // nodes allows icmps and other instructions to fold.
344b0aa36f9SDavid Green     Value *V = SimplifyInstruction(C, SQ);
345b0aa36f9SDavid Green     if (V && LI->replacementPreservesLCSSAForm(C, V)) {
346b0aa36f9SDavid Green       // If so, then delete the temporary instruction and stick the folded value
347b0aa36f9SDavid Green       // in the map.
348b0aa36f9SDavid Green       ValueMap[Inst] = V;
349b0aa36f9SDavid Green       if (!C->mayHaveSideEffects()) {
350b0aa36f9SDavid Green         C->deleteValue();
351b0aa36f9SDavid Green         C = nullptr;
352b0aa36f9SDavid Green       }
353b0aa36f9SDavid Green     } else {
354b0aa36f9SDavid Green       ValueMap[Inst] = C;
355b0aa36f9SDavid Green     }
356b0aa36f9SDavid Green     if (C) {
357b0aa36f9SDavid Green       // Otherwise, stick the new instruction into the new block!
358b0aa36f9SDavid Green       C->setName(Inst->getName());
359b0aa36f9SDavid Green       C->insertBefore(LoopEntryBranch);
360b0aa36f9SDavid Green 
361b0aa36f9SDavid Green       if (auto *II = dyn_cast<IntrinsicInst>(C))
362b0aa36f9SDavid Green         if (II->getIntrinsicID() == Intrinsic::assume)
363b0aa36f9SDavid Green           AC->registerAssumption(II);
364b0aa36f9SDavid Green     }
365b0aa36f9SDavid Green   }
366b0aa36f9SDavid Green 
367b0aa36f9SDavid Green   // Along with all the other instructions, we just cloned OrigHeader's
368b0aa36f9SDavid Green   // terminator into OrigPreHeader. Fix up the PHI nodes in each of OrigHeader's
369b0aa36f9SDavid Green   // successors by duplicating their incoming values for OrigHeader.
370b0aa36f9SDavid Green   TerminatorInst *TI = OrigHeader->getTerminator();
371b0aa36f9SDavid Green   for (BasicBlock *SuccBB : TI->successors())
372b0aa36f9SDavid Green     for (BasicBlock::iterator BI = SuccBB->begin();
373b0aa36f9SDavid Green          PHINode *PN = dyn_cast<PHINode>(BI); ++BI)
374b0aa36f9SDavid Green       PN->addIncoming(PN->getIncomingValueForBlock(OrigHeader), OrigPreheader);
375b0aa36f9SDavid Green 
376b0aa36f9SDavid Green   // Now that OrigPreHeader has a clone of OrigHeader's terminator, remove
377b0aa36f9SDavid Green   // OrigPreHeader's old terminator (the original branch into the loop), and
378b0aa36f9SDavid Green   // remove the corresponding incoming values from the PHI nodes in OrigHeader.
379b0aa36f9SDavid Green   LoopEntryBranch->eraseFromParent();
380b0aa36f9SDavid Green 
381b0aa36f9SDavid Green 
382b0aa36f9SDavid Green   SmallVector<PHINode*, 2> InsertedPHIs;
383b0aa36f9SDavid Green   // If there were any uses of instructions in the duplicated block outside the
384b0aa36f9SDavid Green   // loop, update them, inserting PHI nodes as required
385b0aa36f9SDavid Green   RewriteUsesOfClonedInstructions(OrigHeader, OrigPreheader, ValueMap,
386b0aa36f9SDavid Green                                   &InsertedPHIs);
387b0aa36f9SDavid Green 
388b0aa36f9SDavid Green   // Attach dbg.value intrinsics to the new phis if that phi uses a value that
389b0aa36f9SDavid Green   // previously had debug metadata attached. This keeps the debug info
390b0aa36f9SDavid Green   // up-to-date in the loop body.
391b0aa36f9SDavid Green   if (!InsertedPHIs.empty())
392b0aa36f9SDavid Green     insertDebugValuesForPHIs(OrigHeader, InsertedPHIs);
393b0aa36f9SDavid Green 
394b0aa36f9SDavid Green   // NewHeader is now the header of the loop.
395b0aa36f9SDavid Green   L->moveToHeader(NewHeader);
396b0aa36f9SDavid Green   assert(L->getHeader() == NewHeader && "Latch block is our new header");
397b0aa36f9SDavid Green 
398b0aa36f9SDavid Green   // Inform DT about changes to the CFG.
399b0aa36f9SDavid Green   if (DT) {
400b0aa36f9SDavid Green     // The OrigPreheader branches to the NewHeader and Exit now. Then, inform
401b0aa36f9SDavid Green     // the DT about the removed edge to the OrigHeader (that got removed).
402b0aa36f9SDavid Green     SmallVector<DominatorTree::UpdateType, 3> Updates;
403b0aa36f9SDavid Green     Updates.push_back({DominatorTree::Insert, OrigPreheader, Exit});
404b0aa36f9SDavid Green     Updates.push_back({DominatorTree::Insert, OrigPreheader, NewHeader});
405b0aa36f9SDavid Green     Updates.push_back({DominatorTree::Delete, OrigPreheader, OrigHeader});
406b0aa36f9SDavid Green     DT->applyUpdates(Updates);
407b0aa36f9SDavid Green   }
408b0aa36f9SDavid Green 
409b0aa36f9SDavid Green   // At this point, we've finished our major CFG changes.  As part of cloning
410b0aa36f9SDavid Green   // the loop into the preheader we've simplified instructions and the
411b0aa36f9SDavid Green   // duplicated conditional branch may now be branching on a constant.  If it is
412b0aa36f9SDavid Green   // branching on a constant and if that constant means that we enter the loop,
413b0aa36f9SDavid Green   // then we fold away the cond branch to an uncond branch.  This simplifies the
414b0aa36f9SDavid Green   // loop in cases important for nested loops, and it also means we don't have
415b0aa36f9SDavid Green   // to split as many edges.
416b0aa36f9SDavid Green   BranchInst *PHBI = cast<BranchInst>(OrigPreheader->getTerminator());
417b0aa36f9SDavid Green   assert(PHBI->isConditional() && "Should be clone of BI condbr!");
418b0aa36f9SDavid Green   if (!isa<ConstantInt>(PHBI->getCondition()) ||
419b0aa36f9SDavid Green       PHBI->getSuccessor(cast<ConstantInt>(PHBI->getCondition())->isZero()) !=
420b0aa36f9SDavid Green           NewHeader) {
421b0aa36f9SDavid Green     // The conditional branch can't be folded, handle the general case.
422b0aa36f9SDavid Green     // Split edges as necessary to preserve LoopSimplify form.
423b0aa36f9SDavid Green 
424b0aa36f9SDavid Green     // Right now OrigPreHeader has two successors, NewHeader and ExitBlock, and
425b0aa36f9SDavid Green     // thus is not a preheader anymore.
426b0aa36f9SDavid Green     // Split the edge to form a real preheader.
427b0aa36f9SDavid Green     BasicBlock *NewPH = SplitCriticalEdge(
428b0aa36f9SDavid Green         OrigPreheader, NewHeader,
429b0aa36f9SDavid Green         CriticalEdgeSplittingOptions(DT, LI).setPreserveLCSSA());
430b0aa36f9SDavid Green     NewPH->setName(NewHeader->getName() + ".lr.ph");
431b0aa36f9SDavid Green 
432b0aa36f9SDavid Green     // Preserve canonical loop form, which means that 'Exit' should have only
433b0aa36f9SDavid Green     // one predecessor. Note that Exit could be an exit block for multiple
434b0aa36f9SDavid Green     // nested loops, causing both of the edges to now be critical and need to
435b0aa36f9SDavid Green     // be split.
436b0aa36f9SDavid Green     SmallVector<BasicBlock *, 4> ExitPreds(pred_begin(Exit), pred_end(Exit));
437b0aa36f9SDavid Green     bool SplitLatchEdge = false;
438b0aa36f9SDavid Green     for (BasicBlock *ExitPred : ExitPreds) {
439b0aa36f9SDavid Green       // We only need to split loop exit edges.
440b0aa36f9SDavid Green       Loop *PredLoop = LI->getLoopFor(ExitPred);
441b0aa36f9SDavid Green       if (!PredLoop || PredLoop->contains(Exit))
442b0aa36f9SDavid Green         continue;
443b0aa36f9SDavid Green       if (isa<IndirectBrInst>(ExitPred->getTerminator()))
444b0aa36f9SDavid Green         continue;
445b0aa36f9SDavid Green       SplitLatchEdge |= L->getLoopLatch() == ExitPred;
446b0aa36f9SDavid Green       BasicBlock *ExitSplit = SplitCriticalEdge(
447b0aa36f9SDavid Green           ExitPred, Exit,
448b0aa36f9SDavid Green           CriticalEdgeSplittingOptions(DT, LI).setPreserveLCSSA());
449b0aa36f9SDavid Green       ExitSplit->moveBefore(Exit);
450b0aa36f9SDavid Green     }
451b0aa36f9SDavid Green     assert(SplitLatchEdge &&
452b0aa36f9SDavid Green            "Despite splitting all preds, failed to split latch exit?");
453b0aa36f9SDavid Green   } else {
454b0aa36f9SDavid Green     // We can fold the conditional branch in the preheader, this makes things
455b0aa36f9SDavid Green     // simpler. The first step is to remove the extra edge to the Exit block.
456b0aa36f9SDavid Green     Exit->removePredecessor(OrigPreheader, true /*preserve LCSSA*/);
457b0aa36f9SDavid Green     BranchInst *NewBI = BranchInst::Create(NewHeader, PHBI);
458b0aa36f9SDavid Green     NewBI->setDebugLoc(PHBI->getDebugLoc());
459b0aa36f9SDavid Green     PHBI->eraseFromParent();
460b0aa36f9SDavid Green 
461b0aa36f9SDavid Green     // With our CFG finalized, update DomTree if it is available.
462b0aa36f9SDavid Green     if (DT) DT->deleteEdge(OrigPreheader, Exit);
463b0aa36f9SDavid Green   }
464b0aa36f9SDavid Green 
465b0aa36f9SDavid Green   assert(L->getLoopPreheader() && "Invalid loop preheader after loop rotation");
466b0aa36f9SDavid Green   assert(L->getLoopLatch() && "Invalid loop latch after loop rotation");
467b0aa36f9SDavid Green 
468b0aa36f9SDavid Green   // Now that the CFG and DomTree are in a consistent state again, try to merge
469b0aa36f9SDavid Green   // the OrigHeader block into OrigLatch.  This will succeed if they are
470b0aa36f9SDavid Green   // connected by an unconditional branch.  This is just a cleanup so the
471b0aa36f9SDavid Green   // emitted code isn't too gross in this common case.
472b0aa36f9SDavid Green   MergeBlockIntoPredecessor(OrigHeader, DT, LI);
473b0aa36f9SDavid Green 
474b0aa36f9SDavid Green   DEBUG(dbgs() << "LoopRotation: into "; L->dump());
475b0aa36f9SDavid Green 
476b0aa36f9SDavid Green   ++NumRotated;
477b0aa36f9SDavid Green   return true;
478b0aa36f9SDavid Green }
479b0aa36f9SDavid Green 
480b0aa36f9SDavid Green /// Determine whether the instructions in this range may be safely and cheaply
481b0aa36f9SDavid Green /// speculated. This is not an important enough situation to develop complex
482b0aa36f9SDavid Green /// heuristics. We handle a single arithmetic instruction along with any type
483b0aa36f9SDavid Green /// conversions.
484b0aa36f9SDavid Green static bool shouldSpeculateInstrs(BasicBlock::iterator Begin,
485b0aa36f9SDavid Green                                   BasicBlock::iterator End, Loop *L) {
486b0aa36f9SDavid Green   bool seenIncrement = false;
487b0aa36f9SDavid Green   bool MultiExitLoop = false;
488b0aa36f9SDavid Green 
489b0aa36f9SDavid Green   if (!L->getExitingBlock())
490b0aa36f9SDavid Green     MultiExitLoop = true;
491b0aa36f9SDavid Green 
492b0aa36f9SDavid Green   for (BasicBlock::iterator I = Begin; I != End; ++I) {
493b0aa36f9SDavid Green 
494b0aa36f9SDavid Green     if (!isSafeToSpeculativelyExecute(&*I))
495b0aa36f9SDavid Green       return false;
496b0aa36f9SDavid Green 
497b0aa36f9SDavid Green     if (isa<DbgInfoIntrinsic>(I))
498b0aa36f9SDavid Green       continue;
499b0aa36f9SDavid Green 
500b0aa36f9SDavid Green     switch (I->getOpcode()) {
501b0aa36f9SDavid Green     default:
502b0aa36f9SDavid Green       return false;
503b0aa36f9SDavid Green     case Instruction::GetElementPtr:
504b0aa36f9SDavid Green       // GEPs are cheap if all indices are constant.
505b0aa36f9SDavid Green       if (!cast<GEPOperator>(I)->hasAllConstantIndices())
506b0aa36f9SDavid Green         return false;
507b0aa36f9SDavid Green       // fall-thru to increment case
508b0aa36f9SDavid Green       LLVM_FALLTHROUGH;
509b0aa36f9SDavid Green     case Instruction::Add:
510b0aa36f9SDavid Green     case Instruction::Sub:
511b0aa36f9SDavid Green     case Instruction::And:
512b0aa36f9SDavid Green     case Instruction::Or:
513b0aa36f9SDavid Green     case Instruction::Xor:
514b0aa36f9SDavid Green     case Instruction::Shl:
515b0aa36f9SDavid Green     case Instruction::LShr:
516b0aa36f9SDavid Green     case Instruction::AShr: {
517b0aa36f9SDavid Green       Value *IVOpnd =
518b0aa36f9SDavid Green           !isa<Constant>(I->getOperand(0))
519b0aa36f9SDavid Green               ? I->getOperand(0)
520b0aa36f9SDavid Green               : !isa<Constant>(I->getOperand(1)) ? I->getOperand(1) : nullptr;
521b0aa36f9SDavid Green       if (!IVOpnd)
522b0aa36f9SDavid Green         return false;
523b0aa36f9SDavid Green 
524b0aa36f9SDavid Green       // If increment operand is used outside of the loop, this speculation
525b0aa36f9SDavid Green       // could cause extra live range interference.
526b0aa36f9SDavid Green       if (MultiExitLoop) {
527b0aa36f9SDavid Green         for (User *UseI : IVOpnd->users()) {
528b0aa36f9SDavid Green           auto *UserInst = cast<Instruction>(UseI);
529b0aa36f9SDavid Green           if (!L->contains(UserInst))
530b0aa36f9SDavid Green             return false;
531b0aa36f9SDavid Green         }
532b0aa36f9SDavid Green       }
533b0aa36f9SDavid Green 
534b0aa36f9SDavid Green       if (seenIncrement)
535b0aa36f9SDavid Green         return false;
536b0aa36f9SDavid Green       seenIncrement = true;
537b0aa36f9SDavid Green       break;
538b0aa36f9SDavid Green     }
539b0aa36f9SDavid Green     case Instruction::Trunc:
540b0aa36f9SDavid Green     case Instruction::ZExt:
541b0aa36f9SDavid Green     case Instruction::SExt:
542b0aa36f9SDavid Green       // ignore type conversions
543b0aa36f9SDavid Green       break;
544b0aa36f9SDavid Green     }
545b0aa36f9SDavid Green   }
546b0aa36f9SDavid Green   return true;
547b0aa36f9SDavid Green }
548b0aa36f9SDavid Green 
549b0aa36f9SDavid Green /// Fold the loop tail into the loop exit by speculating the loop tail
550b0aa36f9SDavid Green /// instructions. Typically, this is a single post-increment. In the case of a
551b0aa36f9SDavid Green /// simple 2-block loop, hoisting the increment can be much better than
552b0aa36f9SDavid Green /// duplicating the entire loop header. In the case of loops with early exits,
553b0aa36f9SDavid Green /// rotation will not work anyway, but simplifyLoopLatch will put the loop in
554b0aa36f9SDavid Green /// canonical form so downstream passes can handle it.
555b0aa36f9SDavid Green ///
556b0aa36f9SDavid Green /// I don't believe this invalidates SCEV.
557b0aa36f9SDavid Green bool LoopRotate::simplifyLoopLatch(Loop *L) {
558b0aa36f9SDavid Green   BasicBlock *Latch = L->getLoopLatch();
559b0aa36f9SDavid Green   if (!Latch || Latch->hasAddressTaken())
560b0aa36f9SDavid Green     return false;
561b0aa36f9SDavid Green 
562b0aa36f9SDavid Green   BranchInst *Jmp = dyn_cast<BranchInst>(Latch->getTerminator());
563b0aa36f9SDavid Green   if (!Jmp || !Jmp->isUnconditional())
564b0aa36f9SDavid Green     return false;
565b0aa36f9SDavid Green 
566b0aa36f9SDavid Green   BasicBlock *LastExit = Latch->getSinglePredecessor();
567b0aa36f9SDavid Green   if (!LastExit || !L->isLoopExiting(LastExit))
568b0aa36f9SDavid Green     return false;
569b0aa36f9SDavid Green 
570b0aa36f9SDavid Green   BranchInst *BI = dyn_cast<BranchInst>(LastExit->getTerminator());
571b0aa36f9SDavid Green   if (!BI)
572b0aa36f9SDavid Green     return false;
573b0aa36f9SDavid Green 
574b0aa36f9SDavid Green   if (!shouldSpeculateInstrs(Latch->begin(), Jmp->getIterator(), L))
575b0aa36f9SDavid Green     return false;
576b0aa36f9SDavid Green 
577b0aa36f9SDavid Green   DEBUG(dbgs() << "Folding loop latch " << Latch->getName() << " into "
578b0aa36f9SDavid Green                << LastExit->getName() << "\n");
579b0aa36f9SDavid Green 
580b0aa36f9SDavid Green   // Hoist the instructions from Latch into LastExit.
581b0aa36f9SDavid Green   LastExit->getInstList().splice(BI->getIterator(), Latch->getInstList(),
582b0aa36f9SDavid Green                                  Latch->begin(), Jmp->getIterator());
583b0aa36f9SDavid Green 
584b0aa36f9SDavid Green   unsigned FallThruPath = BI->getSuccessor(0) == Latch ? 0 : 1;
585b0aa36f9SDavid Green   BasicBlock *Header = Jmp->getSuccessor(0);
586b0aa36f9SDavid Green   assert(Header == L->getHeader() && "expected a backward branch");
587b0aa36f9SDavid Green 
588b0aa36f9SDavid Green   // Remove Latch from the CFG so that LastExit becomes the new Latch.
589b0aa36f9SDavid Green   BI->setSuccessor(FallThruPath, Header);
590b0aa36f9SDavid Green   Latch->replaceSuccessorsPhiUsesWith(LastExit);
591b0aa36f9SDavid Green   Jmp->eraseFromParent();
592b0aa36f9SDavid Green 
593b0aa36f9SDavid Green   // Nuke the Latch block.
594b0aa36f9SDavid Green   assert(Latch->empty() && "unable to evacuate Latch");
595b0aa36f9SDavid Green   LI->removeBlock(Latch);
596b0aa36f9SDavid Green   if (DT)
597b0aa36f9SDavid Green     DT->eraseNode(Latch);
598b0aa36f9SDavid Green   Latch->eraseFromParent();
599b0aa36f9SDavid Green   return true;
600b0aa36f9SDavid Green }
601b0aa36f9SDavid Green 
602b0aa36f9SDavid Green /// Rotate \c L, and return true if any modification was made.
603b0aa36f9SDavid Green bool LoopRotate::processLoop(Loop *L) {
604b0aa36f9SDavid Green   // Save the loop metadata.
605b0aa36f9SDavid Green   MDNode *LoopMD = L->getLoopID();
606b0aa36f9SDavid Green 
607b0aa36f9SDavid Green   // Simplify the loop latch before attempting to rotate the header
608b0aa36f9SDavid Green   // upward. Rotation may not be needed if the loop tail can be folded into the
609b0aa36f9SDavid Green   // loop exit.
610b0aa36f9SDavid Green   bool SimplifiedLatch = simplifyLoopLatch(L);
611b0aa36f9SDavid Green 
612b0aa36f9SDavid Green   bool MadeChange = rotateLoop(L, SimplifiedLatch);
613b0aa36f9SDavid Green   assert((!MadeChange || L->isLoopExiting(L->getLoopLatch())) &&
614b0aa36f9SDavid Green          "Loop latch should be exiting after loop-rotate.");
615b0aa36f9SDavid Green 
616b0aa36f9SDavid Green   // Restore the loop metadata.
617b0aa36f9SDavid Green   // NB! We presume LoopRotation DOESN'T ADD its own metadata.
618b0aa36f9SDavid Green   if ((MadeChange || SimplifiedLatch) && LoopMD)
619b0aa36f9SDavid Green     L->setLoopID(LoopMD);
620b0aa36f9SDavid Green 
621b0aa36f9SDavid Green   return MadeChange || SimplifiedLatch;
622b0aa36f9SDavid Green }
623b0aa36f9SDavid Green 
624b0aa36f9SDavid Green 
625b0aa36f9SDavid Green /// The utility to convert a loop into a loop with bottom test.
626b0aa36f9SDavid Green bool llvm::LoopRotation(Loop *L, unsigned MaxHeaderSize, LoopInfo *LI,
627b0aa36f9SDavid Green                         const TargetTransformInfo *TTI, AssumptionCache *AC,
628b0aa36f9SDavid Green                         DominatorTree *DT, ScalarEvolution *SE,
629b0aa36f9SDavid Green                         const SimplifyQuery &SQ) {
630b0aa36f9SDavid Green   LoopRotate LR(MaxHeaderSize, LI, TTI, AC, DT, SE, SQ);
631b0aa36f9SDavid Green 
632b0aa36f9SDavid Green   return LR.processLoop(L);
633b0aa36f9SDavid Green }
634