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