1 //===-- HexagonHardwareLoops.cpp - Identify and generate hardware loops ---===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This pass identifies loops where we can generate the Hexagon hardware 11 // loop instruction. The hardware loop can perform loop branches with a 12 // zero-cycle overhead. 13 // 14 // The pattern that defines the induction variable can changed depending on 15 // prior optimizations. For example, the IndVarSimplify phase run by 'opt' 16 // normalizes induction variables, and the Loop Strength Reduction pass 17 // run by 'llc' may also make changes to the induction variable. 18 // The pattern detected by this phase is due to running Strength Reduction. 19 // 20 // Criteria for hardware loops: 21 // - Countable loops (w/ ind. var for a trip count) 22 // - Assumes loops are normalized by IndVarSimplify 23 // - Try inner-most loops first 24 // - No function calls in loops. 25 // 26 //===----------------------------------------------------------------------===// 27 28 #include "llvm/ADT/SmallSet.h" 29 #include "Hexagon.h" 30 #include "HexagonSubtarget.h" 31 #include "llvm/ADT/Statistic.h" 32 #include "llvm/CodeGen/MachineDominators.h" 33 #include "llvm/CodeGen/MachineFunction.h" 34 #include "llvm/CodeGen/MachineFunctionPass.h" 35 #include "llvm/CodeGen/MachineInstrBuilder.h" 36 #include "llvm/CodeGen/MachineLoopInfo.h" 37 #include "llvm/CodeGen/MachineRegisterInfo.h" 38 #include "llvm/PassSupport.h" 39 #include "llvm/Support/CommandLine.h" 40 #include "llvm/Support/Debug.h" 41 #include "llvm/Support/raw_ostream.h" 42 #include "llvm/Target/TargetInstrInfo.h" 43 #include <algorithm> 44 #include <vector> 45 46 using namespace llvm; 47 48 #define DEBUG_TYPE "hwloops" 49 50 #ifndef NDEBUG 51 static cl::opt<int> HWLoopLimit("hexagon-max-hwloop", cl::Hidden, cl::init(-1)); 52 53 // Option to create preheader only for a specific function. 54 static cl::opt<std::string> PHFn("hexagon-hwloop-phfn", cl::Hidden, 55 cl::init("")); 56 #endif 57 58 // Option to create a preheader if one doesn't exist. 59 static cl::opt<bool> HWCreatePreheader("hexagon-hwloop-preheader", 60 cl::Hidden, cl::init(true), 61 cl::desc("Add a preheader to a hardware loop if one doesn't exist")); 62 63 STATISTIC(NumHWLoops, "Number of loops converted to hardware loops"); 64 65 namespace llvm { 66 FunctionPass *createHexagonHardwareLoops(); 67 void initializeHexagonHardwareLoopsPass(PassRegistry&); 68 } 69 70 namespace { 71 class CountValue; 72 struct HexagonHardwareLoops : public MachineFunctionPass { 73 MachineLoopInfo *MLI; 74 MachineRegisterInfo *MRI; 75 MachineDominatorTree *MDT; 76 const HexagonInstrInfo *TII; 77 #ifndef NDEBUG 78 static int Counter; 79 #endif 80 81 public: 82 static char ID; 83 84 HexagonHardwareLoops() : MachineFunctionPass(ID) { 85 initializeHexagonHardwareLoopsPass(*PassRegistry::getPassRegistry()); 86 } 87 88 bool runOnMachineFunction(MachineFunction &MF) override; 89 90 const char *getPassName() const override { return "Hexagon Hardware Loops"; } 91 92 void getAnalysisUsage(AnalysisUsage &AU) const override { 93 AU.addRequired<MachineDominatorTree>(); 94 AU.addRequired<MachineLoopInfo>(); 95 MachineFunctionPass::getAnalysisUsage(AU); 96 } 97 98 private: 99 typedef std::map<unsigned, MachineInstr *> LoopFeederMap; 100 101 /// Kinds of comparisons in the compare instructions. 102 struct Comparison { 103 enum Kind { 104 EQ = 0x01, 105 NE = 0x02, 106 L = 0x04, 107 G = 0x08, 108 U = 0x40, 109 LTs = L, 110 LEs = L | EQ, 111 GTs = G, 112 GEs = G | EQ, 113 LTu = L | U, 114 LEu = L | EQ | U, 115 GTu = G | U, 116 GEu = G | EQ | U 117 }; 118 119 static Kind getSwappedComparison(Kind Cmp) { 120 assert ((!((Cmp & L) && (Cmp & G))) && "Malformed comparison operator"); 121 if ((Cmp & L) || (Cmp & G)) 122 return (Kind)(Cmp ^ (L|G)); 123 return Cmp; 124 } 125 126 static Kind getNegatedComparison(Kind Cmp) { 127 if ((Cmp & L) || (Cmp & G)) 128 return (Kind)((Cmp ^ (L | G)) ^ EQ); 129 if ((Cmp & NE) || (Cmp & EQ)) 130 return (Kind)(Cmp ^ (EQ | NE)); 131 return (Kind)0; 132 } 133 134 static bool isSigned(Kind Cmp) { 135 return (Cmp & (L | G) && !(Cmp & U)); 136 } 137 138 static bool isUnsigned(Kind Cmp) { 139 return (Cmp & U); 140 } 141 142 }; 143 144 /// \brief Find the register that contains the loop controlling 145 /// induction variable. 146 /// If successful, it will return true and set the \p Reg, \p IVBump 147 /// and \p IVOp arguments. Otherwise it will return false. 148 /// The returned induction register is the register R that follows the 149 /// following induction pattern: 150 /// loop: 151 /// R = phi ..., [ R.next, LatchBlock ] 152 /// R.next = R + #bump 153 /// if (R.next < #N) goto loop 154 /// IVBump is the immediate value added to R, and IVOp is the instruction 155 /// "R.next = R + #bump". 156 bool findInductionRegister(MachineLoop *L, unsigned &Reg, 157 int64_t &IVBump, MachineInstr *&IVOp) const; 158 159 /// \brief Return the comparison kind for the specified opcode. 160 Comparison::Kind getComparisonKind(unsigned CondOpc, 161 MachineOperand *InitialValue, 162 const MachineOperand *Endvalue, 163 int64_t IVBump) const; 164 165 /// \brief Analyze the statements in a loop to determine if the loop 166 /// has a computable trip count and, if so, return a value that represents 167 /// the trip count expression. 168 CountValue *getLoopTripCount(MachineLoop *L, 169 SmallVectorImpl<MachineInstr *> &OldInsts); 170 171 /// \brief Return the expression that represents the number of times 172 /// a loop iterates. The function takes the operands that represent the 173 /// loop start value, loop end value, and induction value. Based upon 174 /// these operands, the function attempts to compute the trip count. 175 /// If the trip count is not directly available (as an immediate value, 176 /// or a register), the function will attempt to insert computation of it 177 /// to the loop's preheader. 178 CountValue *computeCount(MachineLoop *Loop, const MachineOperand *Start, 179 const MachineOperand *End, unsigned IVReg, 180 int64_t IVBump, Comparison::Kind Cmp) const; 181 182 /// \brief Return true if the instruction is not valid within a hardware 183 /// loop. 184 bool isInvalidLoopOperation(const MachineInstr *MI, 185 bool IsInnerHWLoop) const; 186 187 /// \brief Return true if the loop contains an instruction that inhibits 188 /// using the hardware loop. 189 bool containsInvalidInstruction(MachineLoop *L, bool IsInnerHWLoop) const; 190 191 /// \brief Given a loop, check if we can convert it to a hardware loop. 192 /// If so, then perform the conversion and return true. 193 bool convertToHardwareLoop(MachineLoop *L, bool &L0used, bool &L1used); 194 195 /// \brief Return true if the instruction is now dead. 196 bool isDead(const MachineInstr *MI, 197 SmallVectorImpl<MachineInstr *> &DeadPhis) const; 198 199 /// \brief Remove the instruction if it is now dead. 200 void removeIfDead(MachineInstr *MI); 201 202 /// \brief Make sure that the "bump" instruction executes before the 203 /// compare. We need that for the IV fixup, so that the compare 204 /// instruction would not use a bumped value that has not yet been 205 /// defined. If the instructions are out of order, try to reorder them. 206 bool orderBumpCompare(MachineInstr *BumpI, MachineInstr *CmpI); 207 208 /// \brief Return true if MO and MI pair is visited only once. If visited 209 /// more than once, this indicates there is recursion. In such a case, 210 /// return false. 211 bool isLoopFeeder(MachineLoop *L, MachineBasicBlock *A, MachineInstr *MI, 212 const MachineOperand *MO, 213 LoopFeederMap &LoopFeederPhi) const; 214 215 /// \brief Return true if the Phi may generate a value that may underflow, 216 /// or may wrap. 217 bool phiMayWrapOrUnderflow(MachineInstr *Phi, const MachineOperand *EndVal, 218 MachineBasicBlock *MBB, MachineLoop *L, 219 LoopFeederMap &LoopFeederPhi) const; 220 221 /// \brief Return true if the induction variable may underflow an unsigned 222 /// value in the first iteration. 223 bool loopCountMayWrapOrUnderFlow(const MachineOperand *InitVal, 224 const MachineOperand *EndVal, 225 MachineBasicBlock *MBB, MachineLoop *L, 226 LoopFeederMap &LoopFeederPhi) const; 227 228 /// \brief Check if the given operand has a compile-time known constant 229 /// value. Return true if yes, and false otherwise. When returning true, set 230 /// Val to the corresponding constant value. 231 bool checkForImmediate(const MachineOperand &MO, int64_t &Val) const; 232 233 /// \brief Check if the operand has a compile-time known constant value. 234 bool isImmediate(const MachineOperand &MO) const { 235 int64_t V; 236 return checkForImmediate(MO, V); 237 } 238 239 /// \brief Return the immediate for the specified operand. 240 int64_t getImmediate(const MachineOperand &MO) const { 241 int64_t V; 242 if (!checkForImmediate(MO, V)) 243 llvm_unreachable("Invalid operand"); 244 return V; 245 } 246 247 /// \brief Reset the given machine operand to now refer to a new immediate 248 /// value. Assumes that the operand was already referencing an immediate 249 /// value, either directly, or via a register. 250 void setImmediate(MachineOperand &MO, int64_t Val); 251 252 /// \brief Fix the data flow of the induction varible. 253 /// The desired flow is: phi ---> bump -+-> comparison-in-latch. 254 /// | 255 /// +-> back to phi 256 /// where "bump" is the increment of the induction variable: 257 /// iv = iv + #const. 258 /// Due to some prior code transformations, the actual flow may look 259 /// like this: 260 /// phi -+-> bump ---> back to phi 261 /// | 262 /// +-> comparison-in-latch (against upper_bound-bump), 263 /// i.e. the comparison that controls the loop execution may be using 264 /// the value of the induction variable from before the increment. 265 /// 266 /// Return true if the loop's flow is the desired one (i.e. it's 267 /// either been fixed, or no fixing was necessary). 268 /// Otherwise, return false. This can happen if the induction variable 269 /// couldn't be identified, or if the value in the latch's comparison 270 /// cannot be adjusted to reflect the post-bump value. 271 bool fixupInductionVariable(MachineLoop *L); 272 273 /// \brief Given a loop, if it does not have a preheader, create one. 274 /// Return the block that is the preheader. 275 MachineBasicBlock *createPreheaderForLoop(MachineLoop *L); 276 }; 277 278 char HexagonHardwareLoops::ID = 0; 279 #ifndef NDEBUG 280 int HexagonHardwareLoops::Counter = 0; 281 #endif 282 283 /// \brief Abstraction for a trip count of a loop. A smaller version 284 /// of the MachineOperand class without the concerns of changing the 285 /// operand representation. 286 class CountValue { 287 public: 288 enum CountValueType { 289 CV_Register, 290 CV_Immediate 291 }; 292 private: 293 CountValueType Kind; 294 union Values { 295 struct { 296 unsigned Reg; 297 unsigned Sub; 298 } R; 299 unsigned ImmVal; 300 } Contents; 301 302 public: 303 explicit CountValue(CountValueType t, unsigned v, unsigned u = 0) { 304 Kind = t; 305 if (Kind == CV_Register) { 306 Contents.R.Reg = v; 307 Contents.R.Sub = u; 308 } else { 309 Contents.ImmVal = v; 310 } 311 } 312 bool isReg() const { return Kind == CV_Register; } 313 bool isImm() const { return Kind == CV_Immediate; } 314 315 unsigned getReg() const { 316 assert(isReg() && "Wrong CountValue accessor"); 317 return Contents.R.Reg; 318 } 319 unsigned getSubReg() const { 320 assert(isReg() && "Wrong CountValue accessor"); 321 return Contents.R.Sub; 322 } 323 unsigned getImm() const { 324 assert(isImm() && "Wrong CountValue accessor"); 325 return Contents.ImmVal; 326 } 327 328 void print(raw_ostream &OS, const TargetRegisterInfo *TRI = nullptr) const { 329 if (isReg()) { OS << PrintReg(Contents.R.Reg, TRI, Contents.R.Sub); } 330 if (isImm()) { OS << Contents.ImmVal; } 331 } 332 }; 333 } // end anonymous namespace 334 335 336 INITIALIZE_PASS_BEGIN(HexagonHardwareLoops, "hwloops", 337 "Hexagon Hardware Loops", false, false) 338 INITIALIZE_PASS_DEPENDENCY(MachineDominatorTree) 339 INITIALIZE_PASS_DEPENDENCY(MachineLoopInfo) 340 INITIALIZE_PASS_END(HexagonHardwareLoops, "hwloops", 341 "Hexagon Hardware Loops", false, false) 342 343 FunctionPass *llvm::createHexagonHardwareLoops() { 344 return new HexagonHardwareLoops(); 345 } 346 347 bool HexagonHardwareLoops::runOnMachineFunction(MachineFunction &MF) { 348 DEBUG(dbgs() << "********* Hexagon Hardware Loops *********\n"); 349 if (skipFunction(*MF.getFunction())) 350 return false; 351 352 bool Changed = false; 353 354 MLI = &getAnalysis<MachineLoopInfo>(); 355 MRI = &MF.getRegInfo(); 356 MDT = &getAnalysis<MachineDominatorTree>(); 357 TII = MF.getSubtarget<HexagonSubtarget>().getInstrInfo(); 358 359 for (auto &L : *MLI) 360 if (!L->getParentLoop()) { 361 bool L0Used = false; 362 bool L1Used = false; 363 Changed |= convertToHardwareLoop(L, L0Used, L1Used); 364 } 365 366 return Changed; 367 } 368 369 /// \brief Return the latch block if it's one of the exiting blocks. Otherwise, 370 /// return the exiting block. Return 'null' when multiple exiting blocks are 371 /// present. 372 static MachineBasicBlock* getExitingBlock(MachineLoop *L) { 373 if (MachineBasicBlock *Latch = L->getLoopLatch()) { 374 if (L->isLoopExiting(Latch)) 375 return Latch; 376 else 377 return L->getExitingBlock(); 378 } 379 return nullptr; 380 } 381 382 bool HexagonHardwareLoops::findInductionRegister(MachineLoop *L, 383 unsigned &Reg, 384 int64_t &IVBump, 385 MachineInstr *&IVOp 386 ) const { 387 MachineBasicBlock *Header = L->getHeader(); 388 MachineBasicBlock *Preheader = L->getLoopPreheader(); 389 MachineBasicBlock *Latch = L->getLoopLatch(); 390 MachineBasicBlock *ExitingBlock = getExitingBlock(L); 391 if (!Header || !Preheader || !Latch || !ExitingBlock) 392 return false; 393 394 // This pair represents an induction register together with an immediate 395 // value that will be added to it in each loop iteration. 396 typedef std::pair<unsigned,int64_t> RegisterBump; 397 398 // Mapping: R.next -> (R, bump), where R, R.next and bump are derived 399 // from an induction operation 400 // R.next = R + bump 401 // where bump is an immediate value. 402 typedef std::map<unsigned,RegisterBump> InductionMap; 403 404 InductionMap IndMap; 405 406 typedef MachineBasicBlock::instr_iterator instr_iterator; 407 for (instr_iterator I = Header->instr_begin(), E = Header->instr_end(); 408 I != E && I->isPHI(); ++I) { 409 MachineInstr *Phi = &*I; 410 411 // Have a PHI instruction. Get the operand that corresponds to the 412 // latch block, and see if is a result of an addition of form "reg+imm", 413 // where the "reg" is defined by the PHI node we are looking at. 414 for (unsigned i = 1, n = Phi->getNumOperands(); i < n; i += 2) { 415 if (Phi->getOperand(i+1).getMBB() != Latch) 416 continue; 417 418 unsigned PhiOpReg = Phi->getOperand(i).getReg(); 419 MachineInstr *DI = MRI->getVRegDef(PhiOpReg); 420 unsigned UpdOpc = DI->getOpcode(); 421 bool isAdd = (UpdOpc == Hexagon::A2_addi || UpdOpc == Hexagon::A2_addp); 422 423 if (isAdd) { 424 // If the register operand to the add is the PHI we're looking at, this 425 // meets the induction pattern. 426 unsigned IndReg = DI->getOperand(1).getReg(); 427 MachineOperand &Opnd2 = DI->getOperand(2); 428 int64_t V; 429 if (MRI->getVRegDef(IndReg) == Phi && checkForImmediate(Opnd2, V)) { 430 unsigned UpdReg = DI->getOperand(0).getReg(); 431 IndMap.insert(std::make_pair(UpdReg, std::make_pair(IndReg, V))); 432 } 433 } 434 } // for (i) 435 } // for (instr) 436 437 SmallVector<MachineOperand,2> Cond; 438 MachineBasicBlock *TB = nullptr, *FB = nullptr; 439 bool NotAnalyzed = TII->AnalyzeBranch(*ExitingBlock, TB, FB, Cond, false); 440 if (NotAnalyzed) 441 return false; 442 443 unsigned PredR, PredPos, PredRegFlags; 444 if (!TII->getPredReg(Cond, PredR, PredPos, PredRegFlags)) 445 return false; 446 447 MachineInstr *PredI = MRI->getVRegDef(PredR); 448 if (!PredI->isCompare()) 449 return false; 450 451 unsigned CmpReg1 = 0, CmpReg2 = 0; 452 int CmpImm = 0, CmpMask = 0; 453 bool CmpAnalyzed = TII->analyzeCompare(PredI, CmpReg1, CmpReg2, 454 CmpMask, CmpImm); 455 // Fail if the compare was not analyzed, or it's not comparing a register 456 // with an immediate value. Not checking the mask here, since we handle 457 // the individual compare opcodes (including A4_cmpb*) later on. 458 if (!CmpAnalyzed) 459 return false; 460 461 // Exactly one of the input registers to the comparison should be among 462 // the induction registers. 463 InductionMap::iterator IndMapEnd = IndMap.end(); 464 InductionMap::iterator F = IndMapEnd; 465 if (CmpReg1 != 0) { 466 InductionMap::iterator F1 = IndMap.find(CmpReg1); 467 if (F1 != IndMapEnd) 468 F = F1; 469 } 470 if (CmpReg2 != 0) { 471 InductionMap::iterator F2 = IndMap.find(CmpReg2); 472 if (F2 != IndMapEnd) { 473 if (F != IndMapEnd) 474 return false; 475 F = F2; 476 } 477 } 478 if (F == IndMapEnd) 479 return false; 480 481 Reg = F->second.first; 482 IVBump = F->second.second; 483 IVOp = MRI->getVRegDef(F->first); 484 return true; 485 } 486 487 // Return the comparison kind for the specified opcode. 488 HexagonHardwareLoops::Comparison::Kind 489 HexagonHardwareLoops::getComparisonKind(unsigned CondOpc, 490 MachineOperand *InitialValue, 491 const MachineOperand *EndValue, 492 int64_t IVBump) const { 493 Comparison::Kind Cmp = (Comparison::Kind)0; 494 switch (CondOpc) { 495 case Hexagon::C2_cmpeqi: 496 case Hexagon::C2_cmpeq: 497 case Hexagon::C2_cmpeqp: 498 Cmp = Comparison::EQ; 499 break; 500 case Hexagon::C4_cmpneq: 501 case Hexagon::C4_cmpneqi: 502 Cmp = Comparison::NE; 503 break; 504 case Hexagon::C4_cmplte: 505 Cmp = Comparison::LEs; 506 break; 507 case Hexagon::C4_cmplteu: 508 Cmp = Comparison::LEu; 509 break; 510 case Hexagon::C2_cmpgtui: 511 case Hexagon::C2_cmpgtu: 512 case Hexagon::C2_cmpgtup: 513 Cmp = Comparison::GTu; 514 break; 515 case Hexagon::C2_cmpgti: 516 case Hexagon::C2_cmpgt: 517 case Hexagon::C2_cmpgtp: 518 Cmp = Comparison::GTs; 519 break; 520 default: 521 return (Comparison::Kind)0; 522 } 523 return Cmp; 524 } 525 526 /// \brief Analyze the statements in a loop to determine if the loop has 527 /// a computable trip count and, if so, return a value that represents 528 /// the trip count expression. 529 /// 530 /// This function iterates over the phi nodes in the loop to check for 531 /// induction variable patterns that are used in the calculation for 532 /// the number of time the loop is executed. 533 CountValue *HexagonHardwareLoops::getLoopTripCount(MachineLoop *L, 534 SmallVectorImpl<MachineInstr *> &OldInsts) { 535 MachineBasicBlock *TopMBB = L->getTopBlock(); 536 MachineBasicBlock::pred_iterator PI = TopMBB->pred_begin(); 537 assert(PI != TopMBB->pred_end() && 538 "Loop must have more than one incoming edge!"); 539 MachineBasicBlock *Backedge = *PI++; 540 if (PI == TopMBB->pred_end()) // dead loop? 541 return nullptr; 542 MachineBasicBlock *Incoming = *PI++; 543 if (PI != TopMBB->pred_end()) // multiple backedges? 544 return nullptr; 545 546 // Make sure there is one incoming and one backedge and determine which 547 // is which. 548 if (L->contains(Incoming)) { 549 if (L->contains(Backedge)) 550 return nullptr; 551 std::swap(Incoming, Backedge); 552 } else if (!L->contains(Backedge)) 553 return nullptr; 554 555 // Look for the cmp instruction to determine if we can get a useful trip 556 // count. The trip count can be either a register or an immediate. The 557 // location of the value depends upon the type (reg or imm). 558 MachineBasicBlock *ExitingBlock = getExitingBlock(L); 559 if (!ExitingBlock) 560 return nullptr; 561 562 unsigned IVReg = 0; 563 int64_t IVBump = 0; 564 MachineInstr *IVOp; 565 bool FoundIV = findInductionRegister(L, IVReg, IVBump, IVOp); 566 if (!FoundIV) 567 return nullptr; 568 569 MachineBasicBlock *Preheader = L->getLoopPreheader(); 570 571 MachineOperand *InitialValue = nullptr; 572 MachineInstr *IV_Phi = MRI->getVRegDef(IVReg); 573 MachineBasicBlock *Latch = L->getLoopLatch(); 574 for (unsigned i = 1, n = IV_Phi->getNumOperands(); i < n; i += 2) { 575 MachineBasicBlock *MBB = IV_Phi->getOperand(i+1).getMBB(); 576 if (MBB == Preheader) 577 InitialValue = &IV_Phi->getOperand(i); 578 else if (MBB == Latch) 579 IVReg = IV_Phi->getOperand(i).getReg(); // Want IV reg after bump. 580 } 581 if (!InitialValue) 582 return nullptr; 583 584 SmallVector<MachineOperand,2> Cond; 585 MachineBasicBlock *TB = nullptr, *FB = nullptr; 586 bool NotAnalyzed = TII->AnalyzeBranch(*ExitingBlock, TB, FB, Cond, false); 587 if (NotAnalyzed) 588 return nullptr; 589 590 MachineBasicBlock *Header = L->getHeader(); 591 // TB must be non-null. If FB is also non-null, one of them must be 592 // the header. Otherwise, branch to TB could be exiting the loop, and 593 // the fall through can go to the header. 594 assert (TB && "Exit block without a branch?"); 595 if (ExitingBlock != Latch && (TB == Latch || FB == Latch)) { 596 MachineBasicBlock *LTB = 0, *LFB = 0; 597 SmallVector<MachineOperand,2> LCond; 598 bool NotAnalyzed = TII->AnalyzeBranch(*Latch, LTB, LFB, LCond, false); 599 if (NotAnalyzed) 600 return nullptr; 601 if (TB == Latch) 602 TB = (LTB == Header) ? LTB : LFB; 603 else 604 FB = (LTB == Header) ? LTB: LFB; 605 } 606 assert ((!FB || TB == Header || FB == Header) && "Branches not to header?"); 607 if (!TB || (FB && TB != Header && FB != Header)) 608 return nullptr; 609 610 // Branches of form "if (!P) ..." cause HexagonInstrInfo::AnalyzeBranch 611 // to put imm(0), followed by P in the vector Cond. 612 // If TB is not the header, it means that the "not-taken" path must lead 613 // to the header. 614 bool Negated = TII->predOpcodeHasNot(Cond) ^ (TB != Header); 615 unsigned PredReg, PredPos, PredRegFlags; 616 if (!TII->getPredReg(Cond, PredReg, PredPos, PredRegFlags)) 617 return nullptr; 618 MachineInstr *CondI = MRI->getVRegDef(PredReg); 619 unsigned CondOpc = CondI->getOpcode(); 620 621 unsigned CmpReg1 = 0, CmpReg2 = 0; 622 int Mask = 0, ImmValue = 0; 623 bool AnalyzedCmp = TII->analyzeCompare(CondI, CmpReg1, CmpReg2, 624 Mask, ImmValue); 625 if (!AnalyzedCmp) 626 return nullptr; 627 628 // The comparison operator type determines how we compute the loop 629 // trip count. 630 OldInsts.push_back(CondI); 631 OldInsts.push_back(IVOp); 632 633 // Sadly, the following code gets information based on the position 634 // of the operands in the compare instruction. This has to be done 635 // this way, because the comparisons check for a specific relationship 636 // between the operands (e.g. is-less-than), rather than to find out 637 // what relationship the operands are in (as on PPC). 638 Comparison::Kind Cmp; 639 bool isSwapped = false; 640 const MachineOperand &Op1 = CondI->getOperand(1); 641 const MachineOperand &Op2 = CondI->getOperand(2); 642 const MachineOperand *EndValue = nullptr; 643 644 if (Op1.isReg()) { 645 if (Op2.isImm() || Op1.getReg() == IVReg) 646 EndValue = &Op2; 647 else { 648 EndValue = &Op1; 649 isSwapped = true; 650 } 651 } 652 653 if (!EndValue) 654 return nullptr; 655 656 Cmp = getComparisonKind(CondOpc, InitialValue, EndValue, IVBump); 657 if (!Cmp) 658 return nullptr; 659 if (Negated) 660 Cmp = Comparison::getNegatedComparison(Cmp); 661 if (isSwapped) 662 Cmp = Comparison::getSwappedComparison(Cmp); 663 664 if (InitialValue->isReg()) { 665 unsigned R = InitialValue->getReg(); 666 MachineBasicBlock *DefBB = MRI->getVRegDef(R)->getParent(); 667 if (!MDT->properlyDominates(DefBB, Header)) 668 return nullptr; 669 OldInsts.push_back(MRI->getVRegDef(R)); 670 } 671 if (EndValue->isReg()) { 672 unsigned R = EndValue->getReg(); 673 MachineBasicBlock *DefBB = MRI->getVRegDef(R)->getParent(); 674 if (!MDT->properlyDominates(DefBB, Header)) 675 return nullptr; 676 OldInsts.push_back(MRI->getVRegDef(R)); 677 } 678 679 return computeCount(L, InitialValue, EndValue, IVReg, IVBump, Cmp); 680 } 681 682 /// \brief Helper function that returns the expression that represents the 683 /// number of times a loop iterates. The function takes the operands that 684 /// represent the loop start value, loop end value, and induction value. 685 /// Based upon these operands, the function attempts to compute the trip count. 686 CountValue *HexagonHardwareLoops::computeCount(MachineLoop *Loop, 687 const MachineOperand *Start, 688 const MachineOperand *End, 689 unsigned IVReg, 690 int64_t IVBump, 691 Comparison::Kind Cmp) const { 692 // Cannot handle comparison EQ, i.e. while (A == B). 693 if (Cmp == Comparison::EQ) 694 return nullptr; 695 696 // Check if either the start or end values are an assignment of an immediate. 697 // If so, use the immediate value rather than the register. 698 if (Start->isReg()) { 699 const MachineInstr *StartValInstr = MRI->getVRegDef(Start->getReg()); 700 if (StartValInstr && (StartValInstr->getOpcode() == Hexagon::A2_tfrsi || 701 StartValInstr->getOpcode() == Hexagon::A2_tfrpi)) 702 Start = &StartValInstr->getOperand(1); 703 } 704 if (End->isReg()) { 705 const MachineInstr *EndValInstr = MRI->getVRegDef(End->getReg()); 706 if (EndValInstr && (EndValInstr->getOpcode() == Hexagon::A2_tfrsi || 707 EndValInstr->getOpcode() == Hexagon::A2_tfrpi)) 708 End = &EndValInstr->getOperand(1); 709 } 710 711 if (!Start->isReg() && !Start->isImm()) 712 return nullptr; 713 if (!End->isReg() && !End->isImm()) 714 return nullptr; 715 716 bool CmpLess = Cmp & Comparison::L; 717 bool CmpGreater = Cmp & Comparison::G; 718 bool CmpHasEqual = Cmp & Comparison::EQ; 719 720 // Avoid certain wrap-arounds. This doesn't detect all wrap-arounds. 721 if (CmpLess && IVBump < 0) 722 // Loop going while iv is "less" with the iv value going down. Must wrap. 723 return nullptr; 724 725 if (CmpGreater && IVBump > 0) 726 // Loop going while iv is "greater" with the iv value going up. Must wrap. 727 return nullptr; 728 729 // Phis that may feed into the loop. 730 LoopFeederMap LoopFeederPhi; 731 732 // Check if the initial value may be zero and can be decremented in the first 733 // iteration. If the value is zero, the endloop instruction will not decrement 734 // the loop counter, so we shouldn't generate a hardware loop in this case. 735 if (loopCountMayWrapOrUnderFlow(Start, End, Loop->getLoopPreheader(), Loop, 736 LoopFeederPhi)) 737 return nullptr; 738 739 if (Start->isImm() && End->isImm()) { 740 // Both, start and end are immediates. 741 int64_t StartV = Start->getImm(); 742 int64_t EndV = End->getImm(); 743 int64_t Dist = EndV - StartV; 744 if (Dist == 0) 745 return nullptr; 746 747 bool Exact = (Dist % IVBump) == 0; 748 749 if (Cmp == Comparison::NE) { 750 if (!Exact) 751 return nullptr; 752 if ((Dist < 0) ^ (IVBump < 0)) 753 return nullptr; 754 } 755 756 // For comparisons that include the final value (i.e. include equality 757 // with the final value), we need to increase the distance by 1. 758 if (CmpHasEqual) 759 Dist = Dist > 0 ? Dist+1 : Dist-1; 760 761 // For the loop to iterate, CmpLess should imply Dist > 0. Similarly, 762 // CmpGreater should imply Dist < 0. These conditions could actually 763 // fail, for example, in unreachable code (which may still appear to be 764 // reachable in the CFG). 765 if ((CmpLess && Dist < 0) || (CmpGreater && Dist > 0)) 766 return nullptr; 767 768 // "Normalized" distance, i.e. with the bump set to +-1. 769 int64_t Dist1 = (IVBump > 0) ? (Dist + (IVBump - 1)) / IVBump 770 : (-Dist + (-IVBump - 1)) / (-IVBump); 771 assert (Dist1 > 0 && "Fishy thing. Both operands have the same sign."); 772 773 uint64_t Count = Dist1; 774 775 if (Count > 0xFFFFFFFFULL) 776 return nullptr; 777 778 return new CountValue(CountValue::CV_Immediate, Count); 779 } 780 781 // A general case: Start and End are some values, but the actual 782 // iteration count may not be available. If it is not, insert 783 // a computation of it into the preheader. 784 785 // If the induction variable bump is not a power of 2, quit. 786 // Othwerise we'd need a general integer division. 787 if (!isPowerOf2_64(std::abs(IVBump))) 788 return nullptr; 789 790 MachineBasicBlock *PH = Loop->getLoopPreheader(); 791 assert (PH && "Should have a preheader by now"); 792 MachineBasicBlock::iterator InsertPos = PH->getFirstTerminator(); 793 DebugLoc DL; 794 if (InsertPos != PH->end()) 795 DL = InsertPos->getDebugLoc(); 796 797 // If Start is an immediate and End is a register, the trip count 798 // will be "reg - imm". Hexagon's "subtract immediate" instruction 799 // is actually "reg + -imm". 800 801 // If the loop IV is going downwards, i.e. if the bump is negative, 802 // then the iteration count (computed as End-Start) will need to be 803 // negated. To avoid the negation, just swap Start and End. 804 if (IVBump < 0) { 805 std::swap(Start, End); 806 IVBump = -IVBump; 807 } 808 // Cmp may now have a wrong direction, e.g. LEs may now be GEs. 809 // Signedness, and "including equality" are preserved. 810 811 bool RegToImm = Start->isReg() && End->isImm(); // for (reg..imm) 812 bool RegToReg = Start->isReg() && End->isReg(); // for (reg..reg) 813 814 int64_t StartV = 0, EndV = 0; 815 if (Start->isImm()) 816 StartV = Start->getImm(); 817 if (End->isImm()) 818 EndV = End->getImm(); 819 820 int64_t AdjV = 0; 821 // To compute the iteration count, we would need this computation: 822 // Count = (End - Start + (IVBump-1)) / IVBump 823 // or, when CmpHasEqual: 824 // Count = (End - Start + (IVBump-1)+1) / IVBump 825 // The "IVBump-1" part is the adjustment (AdjV). We can avoid 826 // generating an instruction specifically to add it if we can adjust 827 // the immediate values for Start or End. 828 829 if (CmpHasEqual) { 830 // Need to add 1 to the total iteration count. 831 if (Start->isImm()) 832 StartV--; 833 else if (End->isImm()) 834 EndV++; 835 else 836 AdjV += 1; 837 } 838 839 if (Cmp != Comparison::NE) { 840 if (Start->isImm()) 841 StartV -= (IVBump-1); 842 else if (End->isImm()) 843 EndV += (IVBump-1); 844 else 845 AdjV += (IVBump-1); 846 } 847 848 unsigned R = 0, SR = 0; 849 if (Start->isReg()) { 850 R = Start->getReg(); 851 SR = Start->getSubReg(); 852 } else { 853 R = End->getReg(); 854 SR = End->getSubReg(); 855 } 856 const TargetRegisterClass *RC = MRI->getRegClass(R); 857 // Hardware loops cannot handle 64-bit registers. If it's a double 858 // register, it has to have a subregister. 859 if (!SR && RC == &Hexagon::DoubleRegsRegClass) 860 return nullptr; 861 const TargetRegisterClass *IntRC = &Hexagon::IntRegsRegClass; 862 863 // Compute DistR (register with the distance between Start and End). 864 unsigned DistR, DistSR; 865 866 // Avoid special case, where the start value is an imm(0). 867 if (Start->isImm() && StartV == 0) { 868 DistR = End->getReg(); 869 DistSR = End->getSubReg(); 870 } else { 871 const MCInstrDesc &SubD = RegToReg ? TII->get(Hexagon::A2_sub) : 872 (RegToImm ? TII->get(Hexagon::A2_subri) : 873 TII->get(Hexagon::A2_addi)); 874 if (RegToReg || RegToImm) { 875 unsigned SubR = MRI->createVirtualRegister(IntRC); 876 MachineInstrBuilder SubIB = 877 BuildMI(*PH, InsertPos, DL, SubD, SubR); 878 879 if (RegToReg) 880 SubIB.addReg(End->getReg(), 0, End->getSubReg()) 881 .addReg(Start->getReg(), 0, Start->getSubReg()); 882 else 883 SubIB.addImm(EndV) 884 .addReg(Start->getReg(), 0, Start->getSubReg()); 885 DistR = SubR; 886 } else { 887 // If the loop has been unrolled, we should use the original loop count 888 // instead of recalculating the value. This will avoid additional 889 // 'Add' instruction. 890 const MachineInstr *EndValInstr = MRI->getVRegDef(End->getReg()); 891 if (EndValInstr->getOpcode() == Hexagon::A2_addi && 892 EndValInstr->getOperand(2).getImm() == StartV) { 893 DistR = EndValInstr->getOperand(1).getReg(); 894 } else { 895 unsigned SubR = MRI->createVirtualRegister(IntRC); 896 MachineInstrBuilder SubIB = 897 BuildMI(*PH, InsertPos, DL, SubD, SubR); 898 SubIB.addReg(End->getReg(), 0, End->getSubReg()) 899 .addImm(-StartV); 900 DistR = SubR; 901 } 902 } 903 DistSR = 0; 904 } 905 906 // From DistR, compute AdjR (register with the adjusted distance). 907 unsigned AdjR, AdjSR; 908 909 if (AdjV == 0) { 910 AdjR = DistR; 911 AdjSR = DistSR; 912 } else { 913 // Generate CountR = ADD DistR, AdjVal 914 unsigned AddR = MRI->createVirtualRegister(IntRC); 915 MCInstrDesc const &AddD = TII->get(Hexagon::A2_addi); 916 BuildMI(*PH, InsertPos, DL, AddD, AddR) 917 .addReg(DistR, 0, DistSR) 918 .addImm(AdjV); 919 920 AdjR = AddR; 921 AdjSR = 0; 922 } 923 924 // From AdjR, compute CountR (register with the final count). 925 unsigned CountR, CountSR; 926 927 if (IVBump == 1) { 928 CountR = AdjR; 929 CountSR = AdjSR; 930 } else { 931 // The IV bump is a power of two. Log_2(IV bump) is the shift amount. 932 unsigned Shift = Log2_32(IVBump); 933 934 // Generate NormR = LSR DistR, Shift. 935 unsigned LsrR = MRI->createVirtualRegister(IntRC); 936 const MCInstrDesc &LsrD = TII->get(Hexagon::S2_lsr_i_r); 937 BuildMI(*PH, InsertPos, DL, LsrD, LsrR) 938 .addReg(AdjR, 0, AdjSR) 939 .addImm(Shift); 940 941 CountR = LsrR; 942 CountSR = 0; 943 } 944 945 return new CountValue(CountValue::CV_Register, CountR, CountSR); 946 } 947 948 /// \brief Return true if the operation is invalid within hardware loop. 949 bool HexagonHardwareLoops::isInvalidLoopOperation(const MachineInstr *MI, 950 bool IsInnerHWLoop) const { 951 952 // Call is not allowed because the callee may use a hardware loop except for 953 // the case when the call never returns. 954 if (MI->getDesc().isCall() && MI->getOpcode() != Hexagon::CALLv3nr) 955 return true; 956 957 // Check if the instruction defines a hardware loop register. 958 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) { 959 const MachineOperand &MO = MI->getOperand(i); 960 if (!MO.isReg() || !MO.isDef()) 961 continue; 962 unsigned R = MO.getReg(); 963 if (IsInnerHWLoop && (R == Hexagon::LC0 || R == Hexagon::SA0 || 964 R == Hexagon::LC1 || R == Hexagon::SA1)) 965 return true; 966 if (!IsInnerHWLoop && (R == Hexagon::LC1 || R == Hexagon::SA1)) 967 return true; 968 } 969 return false; 970 } 971 972 /// \brief Return true if the loop contains an instruction that inhibits 973 /// the use of the hardware loop instruction. 974 bool HexagonHardwareLoops::containsInvalidInstruction(MachineLoop *L, 975 bool IsInnerHWLoop) const { 976 const std::vector<MachineBasicBlock *> &Blocks = L->getBlocks(); 977 DEBUG(dbgs() << "\nhw_loop head, BB#" << Blocks[0]->getNumber();); 978 for (unsigned i = 0, e = Blocks.size(); i != e; ++i) { 979 MachineBasicBlock *MBB = Blocks[i]; 980 for (MachineBasicBlock::iterator 981 MII = MBB->begin(), E = MBB->end(); MII != E; ++MII) { 982 const MachineInstr *MI = &*MII; 983 if (isInvalidLoopOperation(MI, IsInnerHWLoop)) { 984 DEBUG(dbgs()<< "\nCannot convert to hw_loop due to:"; MI->dump();); 985 return true; 986 } 987 } 988 } 989 return false; 990 } 991 992 /// \brief Returns true if the instruction is dead. This was essentially 993 /// copied from DeadMachineInstructionElim::isDead, but with special cases 994 /// for inline asm, physical registers and instructions with side effects 995 /// removed. 996 bool HexagonHardwareLoops::isDead(const MachineInstr *MI, 997 SmallVectorImpl<MachineInstr *> &DeadPhis) const { 998 // Examine each operand. 999 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) { 1000 const MachineOperand &MO = MI->getOperand(i); 1001 if (!MO.isReg() || !MO.isDef()) 1002 continue; 1003 1004 unsigned Reg = MO.getReg(); 1005 if (MRI->use_nodbg_empty(Reg)) 1006 continue; 1007 1008 typedef MachineRegisterInfo::use_nodbg_iterator use_nodbg_iterator; 1009 1010 // This instruction has users, but if the only user is the phi node for the 1011 // parent block, and the only use of that phi node is this instruction, then 1012 // this instruction is dead: both it (and the phi node) can be removed. 1013 use_nodbg_iterator I = MRI->use_nodbg_begin(Reg); 1014 use_nodbg_iterator End = MRI->use_nodbg_end(); 1015 if (std::next(I) != End || !I->getParent()->isPHI()) 1016 return false; 1017 1018 MachineInstr *OnePhi = I->getParent(); 1019 for (unsigned j = 0, f = OnePhi->getNumOperands(); j != f; ++j) { 1020 const MachineOperand &OPO = OnePhi->getOperand(j); 1021 if (!OPO.isReg() || !OPO.isDef()) 1022 continue; 1023 1024 unsigned OPReg = OPO.getReg(); 1025 use_nodbg_iterator nextJ; 1026 for (use_nodbg_iterator J = MRI->use_nodbg_begin(OPReg); 1027 J != End; J = nextJ) { 1028 nextJ = std::next(J); 1029 MachineOperand &Use = *J; 1030 MachineInstr *UseMI = Use.getParent(); 1031 1032 // If the phi node has a user that is not MI, bail. 1033 if (MI != UseMI) 1034 return false; 1035 } 1036 } 1037 DeadPhis.push_back(OnePhi); 1038 } 1039 1040 // If there are no defs with uses, the instruction is dead. 1041 return true; 1042 } 1043 1044 void HexagonHardwareLoops::removeIfDead(MachineInstr *MI) { 1045 // This procedure was essentially copied from DeadMachineInstructionElim. 1046 1047 SmallVector<MachineInstr*, 1> DeadPhis; 1048 if (isDead(MI, DeadPhis)) { 1049 DEBUG(dbgs() << "HW looping will remove: " << *MI); 1050 1051 // It is possible that some DBG_VALUE instructions refer to this 1052 // instruction. Examine each def operand for such references; 1053 // if found, mark the DBG_VALUE as undef (but don't delete it). 1054 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) { 1055 const MachineOperand &MO = MI->getOperand(i); 1056 if (!MO.isReg() || !MO.isDef()) 1057 continue; 1058 unsigned Reg = MO.getReg(); 1059 MachineRegisterInfo::use_iterator nextI; 1060 for (MachineRegisterInfo::use_iterator I = MRI->use_begin(Reg), 1061 E = MRI->use_end(); I != E; I = nextI) { 1062 nextI = std::next(I); // I is invalidated by the setReg 1063 MachineOperand &Use = *I; 1064 MachineInstr *UseMI = I->getParent(); 1065 if (UseMI == MI) 1066 continue; 1067 if (Use.isDebug()) 1068 UseMI->getOperand(0).setReg(0U); 1069 } 1070 } 1071 1072 MI->eraseFromParent(); 1073 for (unsigned i = 0; i < DeadPhis.size(); ++i) 1074 DeadPhis[i]->eraseFromParent(); 1075 } 1076 } 1077 1078 /// \brief Check if the loop is a candidate for converting to a hardware 1079 /// loop. If so, then perform the transformation. 1080 /// 1081 /// This function works on innermost loops first. A loop can be converted 1082 /// if it is a counting loop; either a register value or an immediate. 1083 /// 1084 /// The code makes several assumptions about the representation of the loop 1085 /// in llvm. 1086 bool HexagonHardwareLoops::convertToHardwareLoop(MachineLoop *L, 1087 bool &RecL0used, 1088 bool &RecL1used) { 1089 // This is just for sanity. 1090 assert(L->getHeader() && "Loop without a header?"); 1091 1092 bool Changed = false; 1093 bool L0Used = false; 1094 bool L1Used = false; 1095 1096 // Process nested loops first. 1097 for (MachineLoop::iterator I = L->begin(), E = L->end(); I != E; ++I) { 1098 Changed |= convertToHardwareLoop(*I, RecL0used, RecL1used); 1099 L0Used |= RecL0used; 1100 L1Used |= RecL1used; 1101 } 1102 1103 // If a nested loop has been converted, then we can't convert this loop. 1104 if (Changed && L0Used && L1Used) 1105 return Changed; 1106 1107 unsigned LOOP_i; 1108 unsigned LOOP_r; 1109 unsigned ENDLOOP; 1110 1111 // Flag used to track loopN instruction: 1112 // 1 - Hardware loop is being generated for the inner most loop. 1113 // 0 - Hardware loop is being generated for the outer loop. 1114 unsigned IsInnerHWLoop = 1; 1115 1116 if (L0Used) { 1117 LOOP_i = Hexagon::J2_loop1i; 1118 LOOP_r = Hexagon::J2_loop1r; 1119 ENDLOOP = Hexagon::ENDLOOP1; 1120 IsInnerHWLoop = 0; 1121 } else { 1122 LOOP_i = Hexagon::J2_loop0i; 1123 LOOP_r = Hexagon::J2_loop0r; 1124 ENDLOOP = Hexagon::ENDLOOP0; 1125 } 1126 1127 #ifndef NDEBUG 1128 // Stop trying after reaching the limit (if any). 1129 int Limit = HWLoopLimit; 1130 if (Limit >= 0) { 1131 if (Counter >= HWLoopLimit) 1132 return false; 1133 Counter++; 1134 } 1135 #endif 1136 1137 // Does the loop contain any invalid instructions? 1138 if (containsInvalidInstruction(L, IsInnerHWLoop)) 1139 return false; 1140 1141 MachineBasicBlock *LastMBB = getExitingBlock(L); 1142 // Don't generate hw loop if the loop has more than one exit. 1143 if (!LastMBB) 1144 return false; 1145 1146 MachineBasicBlock::iterator LastI = LastMBB->getFirstTerminator(); 1147 if (LastI == LastMBB->end()) 1148 return false; 1149 1150 // Is the induction variable bump feeding the latch condition? 1151 if (!fixupInductionVariable(L)) 1152 return false; 1153 1154 // Ensure the loop has a preheader: the loop instruction will be 1155 // placed there. 1156 MachineBasicBlock *Preheader = L->getLoopPreheader(); 1157 if (!Preheader) { 1158 Preheader = createPreheaderForLoop(L); 1159 if (!Preheader) 1160 return false; 1161 } 1162 1163 MachineBasicBlock::iterator InsertPos = Preheader->getFirstTerminator(); 1164 1165 SmallVector<MachineInstr*, 2> OldInsts; 1166 // Are we able to determine the trip count for the loop? 1167 CountValue *TripCount = getLoopTripCount(L, OldInsts); 1168 if (!TripCount) 1169 return false; 1170 1171 // Is the trip count available in the preheader? 1172 if (TripCount->isReg()) { 1173 // There will be a use of the register inserted into the preheader, 1174 // so make sure that the register is actually defined at that point. 1175 MachineInstr *TCDef = MRI->getVRegDef(TripCount->getReg()); 1176 MachineBasicBlock *BBDef = TCDef->getParent(); 1177 if (!MDT->dominates(BBDef, Preheader)) 1178 return false; 1179 } 1180 1181 // Determine the loop start. 1182 MachineBasicBlock *TopBlock = L->getTopBlock(); 1183 MachineBasicBlock *ExitingBlock = getExitingBlock(L); 1184 MachineBasicBlock *LoopStart = 0; 1185 if (ExitingBlock != L->getLoopLatch()) { 1186 MachineBasicBlock *TB = 0, *FB = 0; 1187 SmallVector<MachineOperand, 2> Cond; 1188 1189 if (TII->AnalyzeBranch(*ExitingBlock, TB, FB, Cond, false)) 1190 return false; 1191 1192 if (L->contains(TB)) 1193 LoopStart = TB; 1194 else if (L->contains(FB)) 1195 LoopStart = FB; 1196 else 1197 return false; 1198 } 1199 else 1200 LoopStart = TopBlock; 1201 1202 // Convert the loop to a hardware loop. 1203 DEBUG(dbgs() << "Change to hardware loop at "; L->dump()); 1204 DebugLoc DL; 1205 if (InsertPos != Preheader->end()) 1206 DL = InsertPos->getDebugLoc(); 1207 1208 if (TripCount->isReg()) { 1209 // Create a copy of the loop count register. 1210 unsigned CountReg = MRI->createVirtualRegister(&Hexagon::IntRegsRegClass); 1211 BuildMI(*Preheader, InsertPos, DL, TII->get(TargetOpcode::COPY), CountReg) 1212 .addReg(TripCount->getReg(), 0, TripCount->getSubReg()); 1213 // Add the Loop instruction to the beginning of the loop. 1214 BuildMI(*Preheader, InsertPos, DL, TII->get(LOOP_r)).addMBB(LoopStart) 1215 .addReg(CountReg); 1216 } else { 1217 assert(TripCount->isImm() && "Expecting immediate value for trip count"); 1218 // Add the Loop immediate instruction to the beginning of the loop, 1219 // if the immediate fits in the instructions. Otherwise, we need to 1220 // create a new virtual register. 1221 int64_t CountImm = TripCount->getImm(); 1222 if (!TII->isValidOffset(LOOP_i, CountImm)) { 1223 unsigned CountReg = MRI->createVirtualRegister(&Hexagon::IntRegsRegClass); 1224 BuildMI(*Preheader, InsertPos, DL, TII->get(Hexagon::A2_tfrsi), CountReg) 1225 .addImm(CountImm); 1226 BuildMI(*Preheader, InsertPos, DL, TII->get(LOOP_r)) 1227 .addMBB(LoopStart).addReg(CountReg); 1228 } else 1229 BuildMI(*Preheader, InsertPos, DL, TII->get(LOOP_i)) 1230 .addMBB(LoopStart).addImm(CountImm); 1231 } 1232 1233 // Make sure the loop start always has a reference in the CFG. We need 1234 // to create a BlockAddress operand to get this mechanism to work both the 1235 // MachineBasicBlock and BasicBlock objects need the flag set. 1236 LoopStart->setHasAddressTaken(); 1237 // This line is needed to set the hasAddressTaken flag on the BasicBlock 1238 // object. 1239 BlockAddress::get(const_cast<BasicBlock *>(LoopStart->getBasicBlock())); 1240 1241 // Replace the loop branch with an endloop instruction. 1242 DebugLoc LastIDL = LastI->getDebugLoc(); 1243 BuildMI(*LastMBB, LastI, LastIDL, TII->get(ENDLOOP)).addMBB(LoopStart); 1244 1245 // The loop ends with either: 1246 // - a conditional branch followed by an unconditional branch, or 1247 // - a conditional branch to the loop start. 1248 if (LastI->getOpcode() == Hexagon::J2_jumpt || 1249 LastI->getOpcode() == Hexagon::J2_jumpf) { 1250 // Delete one and change/add an uncond. branch to out of the loop. 1251 MachineBasicBlock *BranchTarget = LastI->getOperand(1).getMBB(); 1252 LastI = LastMBB->erase(LastI); 1253 if (!L->contains(BranchTarget)) { 1254 if (LastI != LastMBB->end()) 1255 LastI = LastMBB->erase(LastI); 1256 SmallVector<MachineOperand, 0> Cond; 1257 TII->InsertBranch(*LastMBB, BranchTarget, nullptr, Cond, LastIDL); 1258 } 1259 } else { 1260 // Conditional branch to loop start; just delete it. 1261 LastMBB->erase(LastI); 1262 } 1263 delete TripCount; 1264 1265 // The induction operation and the comparison may now be 1266 // unneeded. If these are unneeded, then remove them. 1267 for (unsigned i = 0; i < OldInsts.size(); ++i) 1268 removeIfDead(OldInsts[i]); 1269 1270 ++NumHWLoops; 1271 1272 // Set RecL1used and RecL0used only after hardware loop has been 1273 // successfully generated. Doing it earlier can cause wrong loop instruction 1274 // to be used. 1275 if (L0Used) // Loop0 was already used. So, the correct loop must be loop1. 1276 RecL1used = true; 1277 else 1278 RecL0used = true; 1279 1280 return true; 1281 } 1282 1283 bool HexagonHardwareLoops::orderBumpCompare(MachineInstr *BumpI, 1284 MachineInstr *CmpI) { 1285 assert (BumpI != CmpI && "Bump and compare in the same instruction?"); 1286 1287 MachineBasicBlock *BB = BumpI->getParent(); 1288 if (CmpI->getParent() != BB) 1289 return false; 1290 1291 typedef MachineBasicBlock::instr_iterator instr_iterator; 1292 // Check if things are in order to begin with. 1293 for (instr_iterator I(BumpI), E = BB->instr_end(); I != E; ++I) 1294 if (&*I == CmpI) 1295 return true; 1296 1297 // Out of order. 1298 unsigned PredR = CmpI->getOperand(0).getReg(); 1299 bool FoundBump = false; 1300 instr_iterator CmpIt = CmpI->getIterator(), NextIt = std::next(CmpIt); 1301 for (instr_iterator I = NextIt, E = BB->instr_end(); I != E; ++I) { 1302 MachineInstr *In = &*I; 1303 for (unsigned i = 0, n = In->getNumOperands(); i < n; ++i) { 1304 MachineOperand &MO = In->getOperand(i); 1305 if (MO.isReg() && MO.isUse()) { 1306 if (MO.getReg() == PredR) // Found an intervening use of PredR. 1307 return false; 1308 } 1309 } 1310 1311 if (In == BumpI) { 1312 BB->splice(++BumpI->getIterator(), BB, CmpI->getIterator()); 1313 FoundBump = true; 1314 break; 1315 } 1316 } 1317 assert (FoundBump && "Cannot determine instruction order"); 1318 return FoundBump; 1319 } 1320 1321 /// This function is required to break recursion. Visiting phis in a loop may 1322 /// result in recursion during compilation. We break the recursion by making 1323 /// sure that we visit a MachineOperand and its definition in a 1324 /// MachineInstruction only once. If we attempt to visit more than once, then 1325 /// there is recursion, and will return false. 1326 bool HexagonHardwareLoops::isLoopFeeder(MachineLoop *L, MachineBasicBlock *A, 1327 MachineInstr *MI, 1328 const MachineOperand *MO, 1329 LoopFeederMap &LoopFeederPhi) const { 1330 if (LoopFeederPhi.find(MO->getReg()) == LoopFeederPhi.end()) { 1331 const std::vector<MachineBasicBlock *> &Blocks = L->getBlocks(); 1332 DEBUG(dbgs() << "\nhw_loop head, BB#" << Blocks[0]->getNumber();); 1333 // Ignore all BBs that form Loop. 1334 for (unsigned i = 0, e = Blocks.size(); i != e; ++i) { 1335 MachineBasicBlock *MBB = Blocks[i]; 1336 if (A == MBB) 1337 return false; 1338 } 1339 MachineInstr *Def = MRI->getVRegDef(MO->getReg()); 1340 LoopFeederPhi.insert(std::make_pair(MO->getReg(), Def)); 1341 return true; 1342 } else 1343 // Already visited node. 1344 return false; 1345 } 1346 1347 /// Return true if a Phi may generate a value that can underflow. 1348 /// This function calls loopCountMayWrapOrUnderFlow for each Phi operand. 1349 bool HexagonHardwareLoops::phiMayWrapOrUnderflow( 1350 MachineInstr *Phi, const MachineOperand *EndVal, MachineBasicBlock *MBB, 1351 MachineLoop *L, LoopFeederMap &LoopFeederPhi) const { 1352 assert(Phi->isPHI() && "Expecting a Phi."); 1353 // Walk through each Phi, and its used operands. Make sure that 1354 // if there is recursion in Phi, we won't generate hardware loops. 1355 for (int i = 1, n = Phi->getNumOperands(); i < n; i += 2) 1356 if (isLoopFeeder(L, MBB, Phi, &(Phi->getOperand(i)), LoopFeederPhi)) 1357 if (loopCountMayWrapOrUnderFlow(&(Phi->getOperand(i)), EndVal, 1358 Phi->getParent(), L, LoopFeederPhi)) 1359 return true; 1360 return false; 1361 } 1362 1363 /// Return true if the induction variable can underflow in the first iteration. 1364 /// An example, is an initial unsigned value that is 0 and is decrement in the 1365 /// first itertion of a do-while loop. In this case, we cannot generate a 1366 /// hardware loop because the endloop instruction does not decrement the loop 1367 /// counter if it is <= 1. We only need to perform this analysis if the 1368 /// initial value is a register. 1369 /// 1370 /// This function assumes the initial value may underfow unless proven 1371 /// otherwise. If the type is signed, then we don't care because signed 1372 /// underflow is undefined. We attempt to prove the initial value is not 1373 /// zero by perfoming a crude analysis of the loop counter. This function 1374 /// checks if the initial value is used in any comparison prior to the loop 1375 /// and, if so, assumes the comparison is a range check. This is inexact, 1376 /// but will catch the simple cases. 1377 bool HexagonHardwareLoops::loopCountMayWrapOrUnderFlow( 1378 const MachineOperand *InitVal, const MachineOperand *EndVal, 1379 MachineBasicBlock *MBB, MachineLoop *L, 1380 LoopFeederMap &LoopFeederPhi) const { 1381 // Only check register values since they are unknown. 1382 if (!InitVal->isReg()) 1383 return false; 1384 1385 if (!EndVal->isImm()) 1386 return false; 1387 1388 // A register value that is assigned an immediate is a known value, and it 1389 // won't underflow in the first iteration. 1390 int64_t Imm; 1391 if (checkForImmediate(*InitVal, Imm)) 1392 return (EndVal->getImm() == Imm); 1393 1394 unsigned Reg = InitVal->getReg(); 1395 1396 // We don't know the value of a physical register. 1397 if (!TargetRegisterInfo::isVirtualRegister(Reg)) 1398 return true; 1399 1400 MachineInstr *Def = MRI->getVRegDef(Reg); 1401 if (!Def) 1402 return true; 1403 1404 // If the initial value is a Phi or copy and the operands may not underflow, 1405 // then the definition cannot be underflow either. 1406 if (Def->isPHI() && !phiMayWrapOrUnderflow(Def, EndVal, Def->getParent(), 1407 L, LoopFeederPhi)) 1408 return false; 1409 if (Def->isCopy() && !loopCountMayWrapOrUnderFlow(&(Def->getOperand(1)), 1410 EndVal, Def->getParent(), 1411 L, LoopFeederPhi)) 1412 return false; 1413 1414 // Iterate over the uses of the initial value. If the initial value is used 1415 // in a compare, then we assume this is a range check that ensures the loop 1416 // doesn't underflow. This is not an exact test and should be improved. 1417 for (MachineRegisterInfo::use_instr_nodbg_iterator I = MRI->use_instr_nodbg_begin(Reg), 1418 E = MRI->use_instr_nodbg_end(); I != E; ++I) { 1419 MachineInstr *MI = &*I; 1420 unsigned CmpReg1 = 0, CmpReg2 = 0; 1421 int CmpMask = 0, CmpValue = 0; 1422 1423 if (!TII->analyzeCompare(MI, CmpReg1, CmpReg2, CmpMask, CmpValue)) 1424 continue; 1425 1426 MachineBasicBlock *TBB = 0, *FBB = 0; 1427 SmallVector<MachineOperand, 2> Cond; 1428 if (TII->AnalyzeBranch(*MI->getParent(), TBB, FBB, Cond, false)) 1429 continue; 1430 1431 Comparison::Kind Cmp = getComparisonKind(MI->getOpcode(), 0, 0, 0); 1432 if (Cmp == 0) 1433 continue; 1434 if (TII->predOpcodeHasNot(Cond) ^ (TBB != MBB)) 1435 Cmp = Comparison::getNegatedComparison(Cmp); 1436 if (CmpReg2 != 0 && CmpReg2 == Reg) 1437 Cmp = Comparison::getSwappedComparison(Cmp); 1438 1439 // Signed underflow is undefined. 1440 if (Comparison::isSigned(Cmp)) 1441 return false; 1442 1443 // Check if there is a comparison of the initial value. If the initial value 1444 // is greater than or not equal to another value, then assume this is a 1445 // range check. 1446 if ((Cmp & Comparison::G) || Cmp == Comparison::NE) 1447 return false; 1448 } 1449 1450 // OK - this is a hack that needs to be improved. We really need to analyze 1451 // the instructions performed on the initial value. This works on the simplest 1452 // cases only. 1453 if (!Def->isCopy() && !Def->isPHI()) 1454 return false; 1455 1456 return true; 1457 } 1458 1459 bool HexagonHardwareLoops::checkForImmediate(const MachineOperand &MO, 1460 int64_t &Val) const { 1461 if (MO.isImm()) { 1462 Val = MO.getImm(); 1463 return true; 1464 } 1465 if (!MO.isReg()) 1466 return false; 1467 1468 // MO is a register. Check whether it is defined as an immediate value, 1469 // and if so, get the value of it in TV. That value will then need to be 1470 // processed to handle potential subregisters in MO. 1471 int64_t TV; 1472 1473 unsigned R = MO.getReg(); 1474 if (!TargetRegisterInfo::isVirtualRegister(R)) 1475 return false; 1476 MachineInstr *DI = MRI->getVRegDef(R); 1477 unsigned DOpc = DI->getOpcode(); 1478 switch (DOpc) { 1479 case TargetOpcode::COPY: 1480 case Hexagon::A2_tfrsi: 1481 case Hexagon::A2_tfrpi: 1482 case Hexagon::CONST32_Int_Real: 1483 case Hexagon::CONST64_Int_Real: { 1484 // Call recursively to avoid an extra check whether operand(1) is 1485 // indeed an immediate (it could be a global address, for example), 1486 // plus we can handle COPY at the same time. 1487 if (!checkForImmediate(DI->getOperand(1), TV)) 1488 return false; 1489 break; 1490 } 1491 case Hexagon::A2_combineii: 1492 case Hexagon::A4_combineir: 1493 case Hexagon::A4_combineii: 1494 case Hexagon::A4_combineri: 1495 case Hexagon::A2_combinew: { 1496 const MachineOperand &S1 = DI->getOperand(1); 1497 const MachineOperand &S2 = DI->getOperand(2); 1498 int64_t V1, V2; 1499 if (!checkForImmediate(S1, V1) || !checkForImmediate(S2, V2)) 1500 return false; 1501 TV = V2 | (V1 << 32); 1502 break; 1503 } 1504 case TargetOpcode::REG_SEQUENCE: { 1505 const MachineOperand &S1 = DI->getOperand(1); 1506 const MachineOperand &S3 = DI->getOperand(3); 1507 int64_t V1, V3; 1508 if (!checkForImmediate(S1, V1) || !checkForImmediate(S3, V3)) 1509 return false; 1510 unsigned Sub2 = DI->getOperand(2).getImm(); 1511 unsigned Sub4 = DI->getOperand(4).getImm(); 1512 if (Sub2 == Hexagon::subreg_loreg && Sub4 == Hexagon::subreg_hireg) 1513 TV = V1 | (V3 << 32); 1514 else if (Sub2 == Hexagon::subreg_hireg && Sub4 == Hexagon::subreg_loreg) 1515 TV = V3 | (V1 << 32); 1516 else 1517 llvm_unreachable("Unexpected form of REG_SEQUENCE"); 1518 break; 1519 } 1520 1521 default: 1522 return false; 1523 } 1524 1525 // By now, we should have successfuly obtained the immediate value defining 1526 // the register referenced in MO. Handle a potential use of a subregister. 1527 switch (MO.getSubReg()) { 1528 case Hexagon::subreg_loreg: 1529 Val = TV & 0xFFFFFFFFULL; 1530 break; 1531 case Hexagon::subreg_hireg: 1532 Val = (TV >> 32) & 0xFFFFFFFFULL; 1533 break; 1534 default: 1535 Val = TV; 1536 break; 1537 } 1538 return true; 1539 } 1540 1541 void HexagonHardwareLoops::setImmediate(MachineOperand &MO, int64_t Val) { 1542 if (MO.isImm()) { 1543 MO.setImm(Val); 1544 return; 1545 } 1546 1547 assert(MO.isReg()); 1548 unsigned R = MO.getReg(); 1549 MachineInstr *DI = MRI->getVRegDef(R); 1550 1551 const TargetRegisterClass *RC = MRI->getRegClass(R); 1552 unsigned NewR = MRI->createVirtualRegister(RC); 1553 MachineBasicBlock &B = *DI->getParent(); 1554 DebugLoc DL = DI->getDebugLoc(); 1555 BuildMI(B, DI, DL, TII->get(DI->getOpcode()), NewR).addImm(Val); 1556 MO.setReg(NewR); 1557 } 1558 1559 static bool isImmValidForOpcode(unsigned CmpOpc, int64_t Imm) { 1560 // These two instructions are not extendable. 1561 if (CmpOpc == Hexagon::A4_cmpbeqi) 1562 return isUInt<8>(Imm); 1563 if (CmpOpc == Hexagon::A4_cmpbgti) 1564 return isInt<8>(Imm); 1565 // The rest of the comparison-with-immediate instructions are extendable. 1566 return true; 1567 } 1568 1569 bool HexagonHardwareLoops::fixupInductionVariable(MachineLoop *L) { 1570 MachineBasicBlock *Header = L->getHeader(); 1571 MachineBasicBlock *Latch = L->getLoopLatch(); 1572 MachineBasicBlock *ExitingBlock = getExitingBlock(L); 1573 1574 if (!(Header && Latch && ExitingBlock)) 1575 return false; 1576 1577 // These data structures follow the same concept as the corresponding 1578 // ones in findInductionRegister (where some comments are). 1579 typedef std::pair<unsigned,int64_t> RegisterBump; 1580 typedef std::pair<unsigned,RegisterBump> RegisterInduction; 1581 typedef std::set<RegisterInduction> RegisterInductionSet; 1582 1583 // Register candidates for induction variables, with their associated bumps. 1584 RegisterInductionSet IndRegs; 1585 1586 // Look for induction patterns: 1587 // vreg1 = PHI ..., [ latch, vreg2 ] 1588 // vreg2 = ADD vreg1, imm 1589 typedef MachineBasicBlock::instr_iterator instr_iterator; 1590 for (instr_iterator I = Header->instr_begin(), E = Header->instr_end(); 1591 I != E && I->isPHI(); ++I) { 1592 MachineInstr *Phi = &*I; 1593 1594 // Have a PHI instruction. 1595 for (unsigned i = 1, n = Phi->getNumOperands(); i < n; i += 2) { 1596 if (Phi->getOperand(i+1).getMBB() != Latch) 1597 continue; 1598 1599 unsigned PhiReg = Phi->getOperand(i).getReg(); 1600 MachineInstr *DI = MRI->getVRegDef(PhiReg); 1601 unsigned UpdOpc = DI->getOpcode(); 1602 bool isAdd = (UpdOpc == Hexagon::A2_addi || UpdOpc == Hexagon::A2_addp); 1603 1604 if (isAdd) { 1605 // If the register operand to the add/sub is the PHI we are looking 1606 // at, this meets the induction pattern. 1607 unsigned IndReg = DI->getOperand(1).getReg(); 1608 MachineOperand &Opnd2 = DI->getOperand(2); 1609 int64_t V; 1610 if (MRI->getVRegDef(IndReg) == Phi && checkForImmediate(Opnd2, V)) { 1611 unsigned UpdReg = DI->getOperand(0).getReg(); 1612 IndRegs.insert(std::make_pair(UpdReg, std::make_pair(IndReg, V))); 1613 } 1614 } 1615 } // for (i) 1616 } // for (instr) 1617 1618 if (IndRegs.empty()) 1619 return false; 1620 1621 MachineBasicBlock *TB = nullptr, *FB = nullptr; 1622 SmallVector<MachineOperand,2> Cond; 1623 // AnalyzeBranch returns true if it fails to analyze branch. 1624 bool NotAnalyzed = TII->AnalyzeBranch(*ExitingBlock, TB, FB, Cond, false); 1625 if (NotAnalyzed || Cond.empty()) 1626 return false; 1627 1628 if (ExitingBlock != Latch && (TB == Latch || FB == Latch)) { 1629 MachineBasicBlock *LTB = 0, *LFB = 0; 1630 SmallVector<MachineOperand,2> LCond; 1631 bool NotAnalyzed = TII->AnalyzeBranch(*Latch, LTB, LFB, LCond, false); 1632 if (NotAnalyzed) 1633 return false; 1634 1635 // Since latch is not the exiting block, the latch branch should be an 1636 // unconditional branch to the loop header. 1637 if (TB == Latch) 1638 TB = (LTB == Header) ? LTB : LFB; 1639 else 1640 FB = (LTB == Header) ? LTB : LFB; 1641 } 1642 if (TB != Header) { 1643 if (FB != Header) { 1644 // The latch/exit block does not go back to the header. 1645 return false; 1646 } 1647 // FB is the header (i.e., uncond. jump to branch header) 1648 // In this case, the LoopBody -> TB should not be a back edge otherwise 1649 // it could result in an infinite loop after conversion to hw_loop. 1650 // This case can happen when the Latch has two jumps like this: 1651 // Jmp_c OuterLoopHeader <-- TB 1652 // Jmp InnerLoopHeader <-- FB 1653 if (MDT->dominates(TB, FB)) 1654 return false; 1655 } 1656 1657 // Expecting a predicate register as a condition. It won't be a hardware 1658 // predicate register at this point yet, just a vreg. 1659 // HexagonInstrInfo::AnalyzeBranch for negated branches inserts imm(0) 1660 // into Cond, followed by the predicate register. For non-negated branches 1661 // it's just the register. 1662 unsigned CSz = Cond.size(); 1663 if (CSz != 1 && CSz != 2) 1664 return false; 1665 1666 if (!Cond[CSz-1].isReg()) 1667 return false; 1668 1669 unsigned P = Cond[CSz-1].getReg(); 1670 MachineInstr *PredDef = MRI->getVRegDef(P); 1671 1672 if (!PredDef->isCompare()) 1673 return false; 1674 1675 SmallSet<unsigned,2> CmpRegs; 1676 MachineOperand *CmpImmOp = nullptr; 1677 1678 // Go over all operands to the compare and look for immediate and register 1679 // operands. Assume that if the compare has a single register use and a 1680 // single immediate operand, then the register is being compared with the 1681 // immediate value. 1682 for (unsigned i = 0, n = PredDef->getNumOperands(); i < n; ++i) { 1683 MachineOperand &MO = PredDef->getOperand(i); 1684 if (MO.isReg()) { 1685 // Skip all implicit references. In one case there was: 1686 // %vreg140<def> = FCMPUGT32_rr %vreg138, %vreg139, %USR<imp-use> 1687 if (MO.isImplicit()) 1688 continue; 1689 if (MO.isUse()) { 1690 if (!isImmediate(MO)) { 1691 CmpRegs.insert(MO.getReg()); 1692 continue; 1693 } 1694 // Consider the register to be the "immediate" operand. 1695 if (CmpImmOp) 1696 return false; 1697 CmpImmOp = &MO; 1698 } 1699 } else if (MO.isImm()) { 1700 if (CmpImmOp) // A second immediate argument? Confusing. Bail out. 1701 return false; 1702 CmpImmOp = &MO; 1703 } 1704 } 1705 1706 if (CmpRegs.empty()) 1707 return false; 1708 1709 // Check if the compared register follows the order we want. Fix if needed. 1710 for (RegisterInductionSet::iterator I = IndRegs.begin(), E = IndRegs.end(); 1711 I != E; ++I) { 1712 // This is a success. If the register used in the comparison is one that 1713 // we have identified as a bumped (updated) induction register, there is 1714 // nothing to do. 1715 if (CmpRegs.count(I->first)) 1716 return true; 1717 1718 // Otherwise, if the register being compared comes out of a PHI node, 1719 // and has been recognized as following the induction pattern, and is 1720 // compared against an immediate, we can fix it. 1721 const RegisterBump &RB = I->second; 1722 if (CmpRegs.count(RB.first)) { 1723 if (!CmpImmOp) { 1724 // If both operands to the compare instruction are registers, see if 1725 // it can be changed to use induction register as one of the operands. 1726 MachineInstr *IndI = nullptr; 1727 MachineInstr *nonIndI = nullptr; 1728 MachineOperand *IndMO = nullptr; 1729 MachineOperand *nonIndMO = nullptr; 1730 1731 for (unsigned i = 1, n = PredDef->getNumOperands(); i < n; ++i) { 1732 MachineOperand &MO = PredDef->getOperand(i); 1733 if (MO.isReg() && MO.getReg() == RB.first) { 1734 DEBUG(dbgs() << "\n DefMI(" << i << ") = " 1735 << *(MRI->getVRegDef(I->first))); 1736 if (IndI) 1737 return false; 1738 1739 IndI = MRI->getVRegDef(I->first); 1740 IndMO = &MO; 1741 } else if (MO.isReg()) { 1742 DEBUG(dbgs() << "\n DefMI(" << i << ") = " 1743 << *(MRI->getVRegDef(MO.getReg()))); 1744 if (nonIndI) 1745 return false; 1746 1747 nonIndI = MRI->getVRegDef(MO.getReg()); 1748 nonIndMO = &MO; 1749 } 1750 } 1751 if (IndI && nonIndI && 1752 nonIndI->getOpcode() == Hexagon::A2_addi && 1753 nonIndI->getOperand(2).isImm() && 1754 nonIndI->getOperand(2).getImm() == - RB.second) { 1755 bool Order = orderBumpCompare(IndI, PredDef); 1756 if (Order) { 1757 IndMO->setReg(I->first); 1758 nonIndMO->setReg(nonIndI->getOperand(1).getReg()); 1759 return true; 1760 } 1761 } 1762 return false; 1763 } 1764 1765 // It is not valid to do this transformation on an unsigned comparison 1766 // because it may underflow. 1767 Comparison::Kind Cmp = getComparisonKind(PredDef->getOpcode(), 0, 0, 0); 1768 if (!Cmp || Comparison::isUnsigned(Cmp)) 1769 return false; 1770 1771 // If the register is being compared against an immediate, try changing 1772 // the compare instruction to use induction register and adjust the 1773 // immediate operand. 1774 int64_t CmpImm = getImmediate(*CmpImmOp); 1775 int64_t V = RB.second; 1776 // Handle Overflow (64-bit). 1777 if (((V > 0) && (CmpImm > INT64_MAX - V)) || 1778 ((V < 0) && (CmpImm < INT64_MIN - V))) 1779 return false; 1780 CmpImm += V; 1781 // Most comparisons of register against an immediate value allow 1782 // the immediate to be constant-extended. There are some exceptions 1783 // though. Make sure the new combination will work. 1784 if (CmpImmOp->isImm()) 1785 if (!isImmValidForOpcode(PredDef->getOpcode(), CmpImm)) 1786 return false; 1787 1788 // Make sure that the compare happens after the bump. Otherwise, 1789 // after the fixup, the compare would use a yet-undefined register. 1790 MachineInstr *BumpI = MRI->getVRegDef(I->first); 1791 bool Order = orderBumpCompare(BumpI, PredDef); 1792 if (!Order) 1793 return false; 1794 1795 // Finally, fix the compare instruction. 1796 setImmediate(*CmpImmOp, CmpImm); 1797 for (unsigned i = 0, n = PredDef->getNumOperands(); i < n; ++i) { 1798 MachineOperand &MO = PredDef->getOperand(i); 1799 if (MO.isReg() && MO.getReg() == RB.first) { 1800 MO.setReg(I->first); 1801 return true; 1802 } 1803 } 1804 } 1805 } 1806 1807 return false; 1808 } 1809 1810 /// \brief Create a preheader for a given loop. 1811 MachineBasicBlock *HexagonHardwareLoops::createPreheaderForLoop( 1812 MachineLoop *L) { 1813 if (MachineBasicBlock *TmpPH = L->getLoopPreheader()) 1814 return TmpPH; 1815 1816 if (!HWCreatePreheader) 1817 return nullptr; 1818 1819 MachineBasicBlock *Header = L->getHeader(); 1820 MachineBasicBlock *Latch = L->getLoopLatch(); 1821 MachineBasicBlock *ExitingBlock = getExitingBlock(L); 1822 MachineFunction *MF = Header->getParent(); 1823 DebugLoc DL; 1824 1825 #ifndef NDEBUG 1826 if ((PHFn != "") && (PHFn != MF->getName())) 1827 return nullptr; 1828 #endif 1829 1830 if (!Latch || !ExitingBlock || Header->hasAddressTaken()) 1831 return nullptr; 1832 1833 typedef MachineBasicBlock::instr_iterator instr_iterator; 1834 1835 // Verify that all existing predecessors have analyzable branches 1836 // (or no branches at all). 1837 typedef std::vector<MachineBasicBlock*> MBBVector; 1838 MBBVector Preds(Header->pred_begin(), Header->pred_end()); 1839 SmallVector<MachineOperand,2> Tmp1; 1840 MachineBasicBlock *TB = nullptr, *FB = nullptr; 1841 1842 if (TII->AnalyzeBranch(*ExitingBlock, TB, FB, Tmp1, false)) 1843 return nullptr; 1844 1845 for (MBBVector::iterator I = Preds.begin(), E = Preds.end(); I != E; ++I) { 1846 MachineBasicBlock *PB = *I; 1847 bool NotAnalyzed = TII->AnalyzeBranch(*PB, TB, FB, Tmp1, false); 1848 if (NotAnalyzed) 1849 return nullptr; 1850 } 1851 1852 MachineBasicBlock *NewPH = MF->CreateMachineBasicBlock(); 1853 MF->insert(Header->getIterator(), NewPH); 1854 1855 if (Header->pred_size() > 2) { 1856 // Ensure that the header has only two predecessors: the preheader and 1857 // the loop latch. Any additional predecessors of the header should 1858 // join at the newly created preheader. Inspect all PHI nodes from the 1859 // header and create appropriate corresponding PHI nodes in the preheader. 1860 1861 for (instr_iterator I = Header->instr_begin(), E = Header->instr_end(); 1862 I != E && I->isPHI(); ++I) { 1863 MachineInstr *PN = &*I; 1864 1865 const MCInstrDesc &PD = TII->get(TargetOpcode::PHI); 1866 MachineInstr *NewPN = MF->CreateMachineInstr(PD, DL); 1867 NewPH->insert(NewPH->end(), NewPN); 1868 1869 unsigned PR = PN->getOperand(0).getReg(); 1870 const TargetRegisterClass *RC = MRI->getRegClass(PR); 1871 unsigned NewPR = MRI->createVirtualRegister(RC); 1872 NewPN->addOperand(MachineOperand::CreateReg(NewPR, true)); 1873 1874 // Copy all non-latch operands of a header's PHI node to the newly 1875 // created PHI node in the preheader. 1876 for (unsigned i = 1, n = PN->getNumOperands(); i < n; i += 2) { 1877 unsigned PredR = PN->getOperand(i).getReg(); 1878 unsigned PredRSub = PN->getOperand(i).getSubReg(); 1879 MachineBasicBlock *PredB = PN->getOperand(i+1).getMBB(); 1880 if (PredB == Latch) 1881 continue; 1882 1883 MachineOperand MO = MachineOperand::CreateReg(PredR, false); 1884 MO.setSubReg(PredRSub); 1885 NewPN->addOperand(MO); 1886 NewPN->addOperand(MachineOperand::CreateMBB(PredB)); 1887 } 1888 1889 // Remove copied operands from the old PHI node and add the value 1890 // coming from the preheader's PHI. 1891 for (int i = PN->getNumOperands()-2; i > 0; i -= 2) { 1892 MachineBasicBlock *PredB = PN->getOperand(i+1).getMBB(); 1893 if (PredB != Latch) { 1894 PN->RemoveOperand(i+1); 1895 PN->RemoveOperand(i); 1896 } 1897 } 1898 PN->addOperand(MachineOperand::CreateReg(NewPR, false)); 1899 PN->addOperand(MachineOperand::CreateMBB(NewPH)); 1900 } 1901 1902 } else { 1903 assert(Header->pred_size() == 2); 1904 1905 // The header has only two predecessors, but the non-latch predecessor 1906 // is not a preheader (e.g. it has other successors, etc.) 1907 // In such a case we don't need any extra PHI nodes in the new preheader, 1908 // all we need is to adjust existing PHIs in the header to now refer to 1909 // the new preheader. 1910 for (instr_iterator I = Header->instr_begin(), E = Header->instr_end(); 1911 I != E && I->isPHI(); ++I) { 1912 MachineInstr *PN = &*I; 1913 for (unsigned i = 1, n = PN->getNumOperands(); i < n; i += 2) { 1914 MachineOperand &MO = PN->getOperand(i+1); 1915 if (MO.getMBB() != Latch) 1916 MO.setMBB(NewPH); 1917 } 1918 } 1919 } 1920 1921 // "Reroute" the CFG edges to link in the new preheader. 1922 // If any of the predecessors falls through to the header, insert a branch 1923 // to the new preheader in that place. 1924 SmallVector<MachineOperand,1> Tmp2; 1925 SmallVector<MachineOperand,1> EmptyCond; 1926 1927 TB = FB = nullptr; 1928 1929 for (MBBVector::iterator I = Preds.begin(), E = Preds.end(); I != E; ++I) { 1930 MachineBasicBlock *PB = *I; 1931 if (PB != Latch) { 1932 Tmp2.clear(); 1933 bool NotAnalyzed = TII->AnalyzeBranch(*PB, TB, FB, Tmp2, false); 1934 (void)NotAnalyzed; // suppress compiler warning 1935 assert (!NotAnalyzed && "Should be analyzable!"); 1936 if (TB != Header && (Tmp2.empty() || FB != Header)) 1937 TII->InsertBranch(*PB, NewPH, nullptr, EmptyCond, DL); 1938 PB->ReplaceUsesOfBlockWith(Header, NewPH); 1939 } 1940 } 1941 1942 // It can happen that the latch block will fall through into the header. 1943 // Insert an unconditional branch to the header. 1944 TB = FB = nullptr; 1945 bool LatchNotAnalyzed = TII->AnalyzeBranch(*Latch, TB, FB, Tmp2, false); 1946 (void)LatchNotAnalyzed; // suppress compiler warning 1947 assert (!LatchNotAnalyzed && "Should be analyzable!"); 1948 if (!TB && !FB) 1949 TII->InsertBranch(*Latch, Header, nullptr, EmptyCond, DL); 1950 1951 // Finally, the branch from the preheader to the header. 1952 TII->InsertBranch(*NewPH, Header, nullptr, EmptyCond, DL); 1953 NewPH->addSuccessor(Header); 1954 1955 MachineLoop *ParentLoop = L->getParentLoop(); 1956 if (ParentLoop) 1957 ParentLoop->addBasicBlockToLoop(NewPH, MLI->getBase()); 1958 1959 // Update the dominator information with the new preheader. 1960 if (MDT) { 1961 MachineDomTreeNode *HDom = MDT->getNode(Header); 1962 MDT->addNewBlock(NewPH, HDom->getIDom()->getBlock()); 1963 MDT->changeImmediateDominator(Header, NewPH); 1964 } 1965 1966 return NewPH; 1967 } 1968