| //===----------------- LoopRotationUtils.cpp -----------------------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file provides utilities to convert a loop into a loop with bottom test. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "llvm/Transforms/Utils/LoopRotationUtils.h" |
| #include "llvm/ADT/Statistic.h" |
| #include "llvm/Analysis/AliasAnalysis.h" |
| #include "llvm/Analysis/AssumptionCache.h" |
| #include "llvm/Analysis/BasicAliasAnalysis.h" |
| #include "llvm/Analysis/CodeMetrics.h" |
| #include "llvm/Analysis/GlobalsModRef.h" |
| #include "llvm/Analysis/InstructionSimplify.h" |
| #include "llvm/Analysis/LoopPass.h" |
| #include "llvm/Analysis/ScalarEvolution.h" |
| #include "llvm/Analysis/ScalarEvolutionAliasAnalysis.h" |
| #include "llvm/Analysis/TargetTransformInfo.h" |
| #include "llvm/Transforms/Utils/Local.h" |
| #include "llvm/Analysis/ValueTracking.h" |
| #include "llvm/IR/CFG.h" |
| #include "llvm/IR/DebugInfoMetadata.h" |
| #include "llvm/IR/Dominators.h" |
| #include "llvm/IR/Function.h" |
| #include "llvm/IR/IntrinsicInst.h" |
| #include "llvm/IR/Module.h" |
| #include "llvm/Support/CommandLine.h" |
| #include "llvm/Support/Debug.h" |
| #include "llvm/Support/raw_ostream.h" |
| #include "llvm/Transforms/Utils/BasicBlockUtils.h" |
| #include "llvm/Transforms/Utils/LoopUtils.h" |
| #include "llvm/Transforms/Utils/SSAUpdater.h" |
| #include "llvm/Transforms/Utils/ValueMapper.h" |
| using namespace llvm; |
| |
| #define DEBUG_TYPE "loop-rotate" |
| |
| STATISTIC(NumRotated, "Number of loops rotated"); |
| |
| namespace { |
| /// A simple loop rotation transformation. |
| class LoopRotate { |
| const unsigned MaxHeaderSize; |
| LoopInfo *LI; |
| const TargetTransformInfo *TTI; |
| AssumptionCache *AC; |
| DominatorTree *DT; |
| ScalarEvolution *SE; |
| const SimplifyQuery &SQ; |
| bool RotationOnly; |
| bool IsUtilMode; |
| |
| public: |
| LoopRotate(unsigned MaxHeaderSize, LoopInfo *LI, |
| const TargetTransformInfo *TTI, AssumptionCache *AC, |
| DominatorTree *DT, ScalarEvolution *SE, const SimplifyQuery &SQ, |
| bool RotationOnly, bool IsUtilMode) |
| : MaxHeaderSize(MaxHeaderSize), LI(LI), TTI(TTI), AC(AC), DT(DT), SE(SE), |
| SQ(SQ), RotationOnly(RotationOnly), IsUtilMode(IsUtilMode) {} |
| bool processLoop(Loop *L); |
| |
| private: |
| bool rotateLoop(Loop *L, bool SimplifiedLatch); |
| bool simplifyLoopLatch(Loop *L); |
| }; |
| } // end anonymous namespace |
| |
| /// RewriteUsesOfClonedInstructions - We just cloned the instructions from the |
| /// old header into the preheader. If there were uses of the values produced by |
| /// these instruction that were outside of the loop, we have to insert PHI nodes |
| /// to merge the two values. Do this now. |
| static void RewriteUsesOfClonedInstructions(BasicBlock *OrigHeader, |
| BasicBlock *OrigPreheader, |
| ValueToValueMapTy &ValueMap, |
| SmallVectorImpl<PHINode*> *InsertedPHIs) { |
| // Remove PHI node entries that are no longer live. |
| BasicBlock::iterator I, E = OrigHeader->end(); |
| for (I = OrigHeader->begin(); PHINode *PN = dyn_cast<PHINode>(I); ++I) |
| PN->removeIncomingValue(PN->getBasicBlockIndex(OrigPreheader)); |
| |
| // Now fix up users of the instructions in OrigHeader, inserting PHI nodes |
| // as necessary. |
| SSAUpdater SSA(InsertedPHIs); |
| for (I = OrigHeader->begin(); I != E; ++I) { |
| Value *OrigHeaderVal = &*I; |
| |
| // If there are no uses of the value (e.g. because it returns void), there |
| // is nothing to rewrite. |
| if (OrigHeaderVal->use_empty()) |
| continue; |
| |
| Value *OrigPreHeaderVal = ValueMap.lookup(OrigHeaderVal); |
| |
| // The value now exits in two versions: the initial value in the preheader |
| // and the loop "next" value in the original header. |
| SSA.Initialize(OrigHeaderVal->getType(), OrigHeaderVal->getName()); |
| SSA.AddAvailableValue(OrigHeader, OrigHeaderVal); |
| SSA.AddAvailableValue(OrigPreheader, OrigPreHeaderVal); |
| |
| // Visit each use of the OrigHeader instruction. |
| for (Value::use_iterator UI = OrigHeaderVal->use_begin(), |
| UE = OrigHeaderVal->use_end(); |
| UI != UE;) { |
| // Grab the use before incrementing the iterator. |
| Use &U = *UI; |
| |
| // Increment the iterator before removing the use from the list. |
| ++UI; |
| |
| // SSAUpdater can't handle a non-PHI use in the same block as an |
| // earlier def. We can easily handle those cases manually. |
| Instruction *UserInst = cast<Instruction>(U.getUser()); |
| if (!isa<PHINode>(UserInst)) { |
| BasicBlock *UserBB = UserInst->getParent(); |
| |
| // The original users in the OrigHeader are already using the |
| // original definitions. |
| if (UserBB == OrigHeader) |
| continue; |
| |
| // Users in the OrigPreHeader need to use the value to which the |
| // original definitions are mapped. |
| if (UserBB == OrigPreheader) { |
| U = OrigPreHeaderVal; |
| continue; |
| } |
| } |
| |
| // Anything else can be handled by SSAUpdater. |
| SSA.RewriteUse(U); |
| } |
| |
| // Replace MetadataAsValue(ValueAsMetadata(OrigHeaderVal)) uses in debug |
| // intrinsics. |
| SmallVector<DbgValueInst *, 1> DbgValues; |
| llvm::findDbgValues(DbgValues, OrigHeaderVal); |
| for (auto &DbgValue : DbgValues) { |
| // The original users in the OrigHeader are already using the original |
| // definitions. |
| BasicBlock *UserBB = DbgValue->getParent(); |
| if (UserBB == OrigHeader) |
| continue; |
| |
| // Users in the OrigPreHeader need to use the value to which the |
| // original definitions are mapped and anything else can be handled by |
| // the SSAUpdater. To avoid adding PHINodes, check if the value is |
| // available in UserBB, if not substitute undef. |
| Value *NewVal; |
| if (UserBB == OrigPreheader) |
| NewVal = OrigPreHeaderVal; |
| else if (SSA.HasValueForBlock(UserBB)) |
| NewVal = SSA.GetValueInMiddleOfBlock(UserBB); |
| else |
| NewVal = UndefValue::get(OrigHeaderVal->getType()); |
| DbgValue->setOperand(0, |
| MetadataAsValue::get(OrigHeaderVal->getContext(), |
| ValueAsMetadata::get(NewVal))); |
| } |
| } |
| } |
| |
| // Look for a phi which is only used outside the loop (via a LCSSA phi) |
| // in the exit from the header. This means that rotating the loop can |
| // remove the phi. |
| static bool shouldRotateLoopExitingLatch(Loop *L) { |
| BasicBlock *Header = L->getHeader(); |
| BasicBlock *HeaderExit = Header->getTerminator()->getSuccessor(0); |
| if (L->contains(HeaderExit)) |
| HeaderExit = Header->getTerminator()->getSuccessor(1); |
| |
| for (auto &Phi : Header->phis()) { |
| // Look for uses of this phi in the loop/via exits other than the header. |
| if (llvm::any_of(Phi.users(), [HeaderExit](const User *U) { |
| return cast<Instruction>(U)->getParent() != HeaderExit; |
| })) |
| continue; |
| return true; |
| } |
| |
| return false; |
| } |
| |
| /// Rotate loop LP. Return true if the loop is rotated. |
| /// |
| /// \param SimplifiedLatch is true if the latch was just folded into the final |
| /// loop exit. In this case we may want to rotate even though the new latch is |
| /// now an exiting branch. This rotation would have happened had the latch not |
| /// been simplified. However, if SimplifiedLatch is false, then we avoid |
| /// rotating loops in which the latch exits to avoid excessive or endless |
| /// rotation. LoopRotate should be repeatable and converge to a canonical |
| /// form. This property is satisfied because simplifying the loop latch can only |
| /// happen once across multiple invocations of the LoopRotate pass. |
| bool LoopRotate::rotateLoop(Loop *L, bool SimplifiedLatch) { |
| // If the loop has only one block then there is not much to rotate. |
| if (L->getBlocks().size() == 1) |
| return false; |
| |
| BasicBlock *OrigHeader = L->getHeader(); |
| BasicBlock *OrigLatch = L->getLoopLatch(); |
| |
| BranchInst *BI = dyn_cast<BranchInst>(OrigHeader->getTerminator()); |
| if (!BI || BI->isUnconditional()) |
| return false; |
| |
| // If the loop header is not one of the loop exiting blocks then |
| // either this loop is already rotated or it is not |
| // suitable for loop rotation transformations. |
| if (!L->isLoopExiting(OrigHeader)) |
| return false; |
| |
| // If the loop latch already contains a branch that leaves the loop then the |
| // loop is already rotated. |
| if (!OrigLatch) |
| return false; |
| |
| // Rotate if either the loop latch does *not* exit the loop, or if the loop |
| // latch was just simplified. Or if we think it will be profitable. |
| if (L->isLoopExiting(OrigLatch) && !SimplifiedLatch && IsUtilMode == false && |
| !shouldRotateLoopExitingLatch(L)) |
| return false; |
| |
| // Check size of original header and reject loop if it is very big or we can't |
| // duplicate blocks inside it. |
| { |
| SmallPtrSet<const Value *, 32> EphValues; |
| CodeMetrics::collectEphemeralValues(L, AC, EphValues); |
| |
| CodeMetrics Metrics; |
| Metrics.analyzeBasicBlock(OrigHeader, *TTI, EphValues); |
| if (Metrics.notDuplicatable) { |
| LLVM_DEBUG( |
| dbgs() << "LoopRotation: NOT rotating - contains non-duplicatable" |
| << " instructions: "; |
| L->dump()); |
| return false; |
| } |
| if (Metrics.convergent) { |
| LLVM_DEBUG(dbgs() << "LoopRotation: NOT rotating - contains convergent " |
| "instructions: "; |
| L->dump()); |
| return false; |
| } |
| if (Metrics.NumInsts > MaxHeaderSize) |
| return false; |
| } |
| |
| // Now, this loop is suitable for rotation. |
| BasicBlock *OrigPreheader = L->getLoopPreheader(); |
| |
| // If the loop could not be converted to canonical form, it must have an |
| // indirectbr in it, just give up. |
| if (!OrigPreheader || !L->hasDedicatedExits()) |
| return false; |
| |
| // Anything ScalarEvolution may know about this loop or the PHI nodes |
| // in its header will soon be invalidated. We should also invalidate |
| // all outer loops because insertion and deletion of blocks that happens |
| // during the rotation may violate invariants related to backedge taken |
| // infos in them. |
| if (SE) |
| SE->forgetTopmostLoop(L); |
| |
| LLVM_DEBUG(dbgs() << "LoopRotation: rotating "; L->dump()); |
| |
| // Find new Loop header. NewHeader is a Header's one and only successor |
| // that is inside loop. Header's other successor is outside the |
| // loop. Otherwise loop is not suitable for rotation. |
| BasicBlock *Exit = BI->getSuccessor(0); |
| BasicBlock *NewHeader = BI->getSuccessor(1); |
| if (L->contains(Exit)) |
| std::swap(Exit, NewHeader); |
| assert(NewHeader && "Unable to determine new loop header"); |
| assert(L->contains(NewHeader) && !L->contains(Exit) && |
| "Unable to determine loop header and exit blocks"); |
| |
| // This code assumes that the new header has exactly one predecessor. |
| // Remove any single-entry PHI nodes in it. |
| assert(NewHeader->getSinglePredecessor() && |
| "New header doesn't have one pred!"); |
| FoldSingleEntryPHINodes(NewHeader); |
| |
| // Begin by walking OrigHeader and populating ValueMap with an entry for |
| // each Instruction. |
| BasicBlock::iterator I = OrigHeader->begin(), E = OrigHeader->end(); |
| ValueToValueMapTy ValueMap; |
| |
| // For PHI nodes, the value available in OldPreHeader is just the |
| // incoming value from OldPreHeader. |
| for (; PHINode *PN = dyn_cast<PHINode>(I); ++I) |
| ValueMap[PN] = PN->getIncomingValueForBlock(OrigPreheader); |
| |
| // For the rest of the instructions, either hoist to the OrigPreheader if |
| // possible or create a clone in the OldPreHeader if not. |
| TerminatorInst *LoopEntryBranch = OrigPreheader->getTerminator(); |
| |
| // Record all debug intrinsics preceding LoopEntryBranch to avoid duplication. |
| using DbgIntrinsicHash = |
| std::pair<std::pair<Value *, DILocalVariable *>, DIExpression *>; |
| auto makeHash = [](DbgInfoIntrinsic *D) -> DbgIntrinsicHash { |
| return {{D->getVariableLocation(), D->getVariable()}, D->getExpression()}; |
| }; |
| SmallDenseSet<DbgIntrinsicHash, 8> DbgIntrinsics; |
| for (auto I = std::next(OrigPreheader->rbegin()), E = OrigPreheader->rend(); |
| I != E; ++I) { |
| if (auto *DII = dyn_cast<DbgInfoIntrinsic>(&*I)) |
| DbgIntrinsics.insert(makeHash(DII)); |
| else |
| break; |
| } |
| |
| while (I != E) { |
| Instruction *Inst = &*I++; |
| |
| // If the instruction's operands are invariant and it doesn't read or write |
| // memory, then it is safe to hoist. Doing this doesn't change the order of |
| // execution in the preheader, but does prevent the instruction from |
| // executing in each iteration of the loop. This means it is safe to hoist |
| // something that might trap, but isn't safe to hoist something that reads |
| // memory (without proving that the loop doesn't write). |
| if (L->hasLoopInvariantOperands(Inst) && !Inst->mayReadFromMemory() && |
| !Inst->mayWriteToMemory() && !isa<TerminatorInst>(Inst) && |
| !isa<DbgInfoIntrinsic>(Inst) && !isa<AllocaInst>(Inst)) { |
| Inst->moveBefore(LoopEntryBranch); |
| continue; |
| } |
| |
| // Otherwise, create a duplicate of the instruction. |
| Instruction *C = Inst->clone(); |
| |
| // Eagerly remap the operands of the instruction. |
| RemapInstruction(C, ValueMap, |
| RF_NoModuleLevelChanges | RF_IgnoreMissingLocals); |
| |
| // Avoid inserting the same intrinsic twice. |
| if (auto *DII = dyn_cast<DbgInfoIntrinsic>(C)) |
| if (DbgIntrinsics.count(makeHash(DII))) { |
| C->deleteValue(); |
| continue; |
| } |
| |
| // With the operands remapped, see if the instruction constant folds or is |
| // otherwise simplifyable. This commonly occurs because the entry from PHI |
| // nodes allows icmps and other instructions to fold. |
| Value *V = SimplifyInstruction(C, SQ); |
| if (V && LI->replacementPreservesLCSSAForm(C, V)) { |
| // If so, then delete the temporary instruction and stick the folded value |
| // in the map. |
| ValueMap[Inst] = V; |
| if (!C->mayHaveSideEffects()) { |
| C->deleteValue(); |
| C = nullptr; |
| } |
| } else { |
| ValueMap[Inst] = C; |
| } |
| if (C) { |
| // Otherwise, stick the new instruction into the new block! |
| C->setName(Inst->getName()); |
| C->insertBefore(LoopEntryBranch); |
| |
| if (auto *II = dyn_cast<IntrinsicInst>(C)) |
| if (II->getIntrinsicID() == Intrinsic::assume) |
| AC->registerAssumption(II); |
| } |
| } |
| |
| // Along with all the other instructions, we just cloned OrigHeader's |
| // terminator into OrigPreHeader. Fix up the PHI nodes in each of OrigHeader's |
| // successors by duplicating their incoming values for OrigHeader. |
| TerminatorInst *TI = OrigHeader->getTerminator(); |
| for (BasicBlock *SuccBB : TI->successors()) |
| for (BasicBlock::iterator BI = SuccBB->begin(); |
| PHINode *PN = dyn_cast<PHINode>(BI); ++BI) |
| PN->addIncoming(PN->getIncomingValueForBlock(OrigHeader), OrigPreheader); |
| |
| // Now that OrigPreHeader has a clone of OrigHeader's terminator, remove |
| // OrigPreHeader's old terminator (the original branch into the loop), and |
| // remove the corresponding incoming values from the PHI nodes in OrigHeader. |
| LoopEntryBranch->eraseFromParent(); |
| |
| |
| SmallVector<PHINode*, 2> InsertedPHIs; |
| // If there were any uses of instructions in the duplicated block outside the |
| // loop, update them, inserting PHI nodes as required |
| RewriteUsesOfClonedInstructions(OrigHeader, OrigPreheader, ValueMap, |
| &InsertedPHIs); |
| |
| // Attach dbg.value intrinsics to the new phis if that phi uses a value that |
| // previously had debug metadata attached. This keeps the debug info |
| // up-to-date in the loop body. |
| if (!InsertedPHIs.empty()) |
| insertDebugValuesForPHIs(OrigHeader, InsertedPHIs); |
| |
| // NewHeader is now the header of the loop. |
| L->moveToHeader(NewHeader); |
| assert(L->getHeader() == NewHeader && "Latch block is our new header"); |
| |
| // Inform DT about changes to the CFG. |
| if (DT) { |
| // The OrigPreheader branches to the NewHeader and Exit now. Then, inform |
| // the DT about the removed edge to the OrigHeader (that got removed). |
| SmallVector<DominatorTree::UpdateType, 3> Updates; |
| Updates.push_back({DominatorTree::Insert, OrigPreheader, Exit}); |
| Updates.push_back({DominatorTree::Insert, OrigPreheader, NewHeader}); |
| Updates.push_back({DominatorTree::Delete, OrigPreheader, OrigHeader}); |
| DT->applyUpdates(Updates); |
| } |
| |
| // At this point, we've finished our major CFG changes. As part of cloning |
| // the loop into the preheader we've simplified instructions and the |
| // duplicated conditional branch may now be branching on a constant. If it is |
| // branching on a constant and if that constant means that we enter the loop, |
| // then we fold away the cond branch to an uncond branch. This simplifies the |
| // loop in cases important for nested loops, and it also means we don't have |
| // to split as many edges. |
| BranchInst *PHBI = cast<BranchInst>(OrigPreheader->getTerminator()); |
| assert(PHBI->isConditional() && "Should be clone of BI condbr!"); |
| if (!isa<ConstantInt>(PHBI->getCondition()) || |
| PHBI->getSuccessor(cast<ConstantInt>(PHBI->getCondition())->isZero()) != |
| NewHeader) { |
| // The conditional branch can't be folded, handle the general case. |
| // Split edges as necessary to preserve LoopSimplify form. |
| |
| // Right now OrigPreHeader has two successors, NewHeader and ExitBlock, and |
| // thus is not a preheader anymore. |
| // Split the edge to form a real preheader. |
| BasicBlock *NewPH = SplitCriticalEdge( |
| OrigPreheader, NewHeader, |
| CriticalEdgeSplittingOptions(DT, LI).setPreserveLCSSA()); |
| NewPH->setName(NewHeader->getName() + ".lr.ph"); |
| |
| // Preserve canonical loop form, which means that 'Exit' should have only |
| // one predecessor. Note that Exit could be an exit block for multiple |
| // nested loops, causing both of the edges to now be critical and need to |
| // be split. |
| SmallVector<BasicBlock *, 4> ExitPreds(pred_begin(Exit), pred_end(Exit)); |
| bool SplitLatchEdge = false; |
| for (BasicBlock *ExitPred : ExitPreds) { |
| // We only need to split loop exit edges. |
| Loop *PredLoop = LI->getLoopFor(ExitPred); |
| if (!PredLoop || PredLoop->contains(Exit)) |
| continue; |
| if (isa<IndirectBrInst>(ExitPred->getTerminator())) |
| continue; |
| SplitLatchEdge |= L->getLoopLatch() == ExitPred; |
| BasicBlock *ExitSplit = SplitCriticalEdge( |
| ExitPred, Exit, |
| CriticalEdgeSplittingOptions(DT, LI).setPreserveLCSSA()); |
| ExitSplit->moveBefore(Exit); |
| } |
| assert(SplitLatchEdge && |
| "Despite splitting all preds, failed to split latch exit?"); |
| } else { |
| // We can fold the conditional branch in the preheader, this makes things |
| // simpler. The first step is to remove the extra edge to the Exit block. |
| Exit->removePredecessor(OrigPreheader, true /*preserve LCSSA*/); |
| BranchInst *NewBI = BranchInst::Create(NewHeader, PHBI); |
| NewBI->setDebugLoc(PHBI->getDebugLoc()); |
| PHBI->eraseFromParent(); |
| |
| // With our CFG finalized, update DomTree if it is available. |
| if (DT) DT->deleteEdge(OrigPreheader, Exit); |
| } |
| |
| assert(L->getLoopPreheader() && "Invalid loop preheader after loop rotation"); |
| assert(L->getLoopLatch() && "Invalid loop latch after loop rotation"); |
| |
| // Now that the CFG and DomTree are in a consistent state again, try to merge |
| // the OrigHeader block into OrigLatch. This will succeed if they are |
| // connected by an unconditional branch. This is just a cleanup so the |
| // emitted code isn't too gross in this common case. |
| MergeBlockIntoPredecessor(OrigHeader, DT, LI); |
| |
| LLVM_DEBUG(dbgs() << "LoopRotation: into "; L->dump()); |
| |
| ++NumRotated; |
| return true; |
| } |
| |
| /// Determine whether the instructions in this range may be safely and cheaply |
| /// speculated. This is not an important enough situation to develop complex |
| /// heuristics. We handle a single arithmetic instruction along with any type |
| /// conversions. |
| static bool shouldSpeculateInstrs(BasicBlock::iterator Begin, |
| BasicBlock::iterator End, Loop *L) { |
| bool seenIncrement = false; |
| bool MultiExitLoop = false; |
| |
| if (!L->getExitingBlock()) |
| MultiExitLoop = true; |
| |
| for (BasicBlock::iterator I = Begin; I != End; ++I) { |
| |
| if (!isSafeToSpeculativelyExecute(&*I)) |
| return false; |
| |
| if (isa<DbgInfoIntrinsic>(I)) |
| continue; |
| |
| switch (I->getOpcode()) { |
| default: |
| return false; |
| case Instruction::GetElementPtr: |
| // GEPs are cheap if all indices are constant. |
| if (!cast<GEPOperator>(I)->hasAllConstantIndices()) |
| return false; |
| // fall-thru to increment case |
| LLVM_FALLTHROUGH; |
| case Instruction::Add: |
| case Instruction::Sub: |
| case Instruction::And: |
| case Instruction::Or: |
| case Instruction::Xor: |
| case Instruction::Shl: |
| case Instruction::LShr: |
| case Instruction::AShr: { |
| Value *IVOpnd = |
| !isa<Constant>(I->getOperand(0)) |
| ? I->getOperand(0) |
| : !isa<Constant>(I->getOperand(1)) ? I->getOperand(1) : nullptr; |
| if (!IVOpnd) |
| return false; |
| |
| // If increment operand is used outside of the loop, this speculation |
| // could cause extra live range interference. |
| if (MultiExitLoop) { |
| for (User *UseI : IVOpnd->users()) { |
| auto *UserInst = cast<Instruction>(UseI); |
| if (!L->contains(UserInst)) |
| return false; |
| } |
| } |
| |
| if (seenIncrement) |
| return false; |
| seenIncrement = true; |
| break; |
| } |
| case Instruction::Trunc: |
| case Instruction::ZExt: |
| case Instruction::SExt: |
| // ignore type conversions |
| break; |
| } |
| } |
| return true; |
| } |
| |
| /// Fold the loop tail into the loop exit by speculating the loop tail |
| /// instructions. Typically, this is a single post-increment. In the case of a |
| /// simple 2-block loop, hoisting the increment can be much better than |
| /// duplicating the entire loop header. In the case of loops with early exits, |
| /// rotation will not work anyway, but simplifyLoopLatch will put the loop in |
| /// canonical form so downstream passes can handle it. |
| /// |
| /// I don't believe this invalidates SCEV. |
| bool LoopRotate::simplifyLoopLatch(Loop *L) { |
| BasicBlock *Latch = L->getLoopLatch(); |
| if (!Latch || Latch->hasAddressTaken()) |
| return false; |
| |
| BranchInst *Jmp = dyn_cast<BranchInst>(Latch->getTerminator()); |
| if (!Jmp || !Jmp->isUnconditional()) |
| return false; |
| |
| BasicBlock *LastExit = Latch->getSinglePredecessor(); |
| if (!LastExit || !L->isLoopExiting(LastExit)) |
| return false; |
| |
| BranchInst *BI = dyn_cast<BranchInst>(LastExit->getTerminator()); |
| if (!BI) |
| return false; |
| |
| if (!shouldSpeculateInstrs(Latch->begin(), Jmp->getIterator(), L)) |
| return false; |
| |
| LLVM_DEBUG(dbgs() << "Folding loop latch " << Latch->getName() << " into " |
| << LastExit->getName() << "\n"); |
| |
| // Hoist the instructions from Latch into LastExit. |
| LastExit->getInstList().splice(BI->getIterator(), Latch->getInstList(), |
| Latch->begin(), Jmp->getIterator()); |
| |
| unsigned FallThruPath = BI->getSuccessor(0) == Latch ? 0 : 1; |
| BasicBlock *Header = Jmp->getSuccessor(0); |
| assert(Header == L->getHeader() && "expected a backward branch"); |
| |
| // Remove Latch from the CFG so that LastExit becomes the new Latch. |
| BI->setSuccessor(FallThruPath, Header); |
| Latch->replaceSuccessorsPhiUsesWith(LastExit); |
| Jmp->eraseFromParent(); |
| |
| // Nuke the Latch block. |
| assert(Latch->empty() && "unable to evacuate Latch"); |
| LI->removeBlock(Latch); |
| if (DT) |
| DT->eraseNode(Latch); |
| Latch->eraseFromParent(); |
| return true; |
| } |
| |
| /// Rotate \c L, and return true if any modification was made. |
| bool LoopRotate::processLoop(Loop *L) { |
| // Save the loop metadata. |
| MDNode *LoopMD = L->getLoopID(); |
| |
| bool SimplifiedLatch = false; |
| |
| // Simplify the loop latch before attempting to rotate the header |
| // upward. Rotation may not be needed if the loop tail can be folded into the |
| // loop exit. |
| if (!RotationOnly) |
| SimplifiedLatch = simplifyLoopLatch(L); |
| |
| bool MadeChange = rotateLoop(L, SimplifiedLatch); |
| assert((!MadeChange || L->isLoopExiting(L->getLoopLatch())) && |
| "Loop latch should be exiting after loop-rotate."); |
| |
| // Restore the loop metadata. |
| // NB! We presume LoopRotation DOESN'T ADD its own metadata. |
| if ((MadeChange || SimplifiedLatch) && LoopMD) |
| L->setLoopID(LoopMD); |
| |
| return MadeChange || SimplifiedLatch; |
| } |
| |
| |
| /// The utility to convert a loop into a loop with bottom test. |
| bool llvm::LoopRotation(Loop *L, LoopInfo *LI, const TargetTransformInfo *TTI, |
| AssumptionCache *AC, DominatorTree *DT, |
| ScalarEvolution *SE, const SimplifyQuery &SQ, |
| bool RotationOnly = true, |
| unsigned Threshold = unsigned(-1), |
| bool IsUtilMode = true) { |
| LoopRotate LR(Threshold, LI, TTI, AC, DT, SE, SQ, RotationOnly, IsUtilMode); |
| |
| return LR.processLoop(L); |
| } |