1 //===---- ReachingDefAnalysis.cpp - Reaching Def Analysis ---*- C++ -*-----===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/ADT/SmallSet.h" 10 #include "llvm/CodeGen/LivePhysRegs.h" 11 #include "llvm/CodeGen/ReachingDefAnalysis.h" 12 #include "llvm/CodeGen/TargetRegisterInfo.h" 13 #include "llvm/CodeGen/TargetSubtargetInfo.h" 14 #include "llvm/Support/Debug.h" 15 16 using namespace llvm; 17 18 #define DEBUG_TYPE "reaching-deps-analysis" 19 20 char ReachingDefAnalysis::ID = 0; 21 INITIALIZE_PASS(ReachingDefAnalysis, DEBUG_TYPE, "ReachingDefAnalysis", false, 22 true) 23 24 static bool isValidReg(const MachineOperand &MO) { 25 return MO.isReg() && MO.getReg(); 26 } 27 28 static bool isValidRegUse(const MachineOperand &MO) { 29 return isValidReg(MO) && MO.isUse(); 30 } 31 32 static bool isValidRegUseOf(const MachineOperand &MO, int PhysReg) { 33 return isValidRegUse(MO) && MO.getReg() == PhysReg; 34 } 35 36 static bool isValidRegDef(const MachineOperand &MO) { 37 return isValidReg(MO) && MO.isDef(); 38 } 39 40 static bool isValidRegDefOf(const MachineOperand &MO, int PhysReg) { 41 return isValidRegDef(MO) && MO.getReg() == PhysReg; 42 } 43 44 void ReachingDefAnalysis::enterBasicBlock(MachineBasicBlock *MBB) { 45 unsigned MBBNumber = MBB->getNumber(); 46 assert(MBBNumber < MBBReachingDefs.size() && 47 "Unexpected basic block number."); 48 MBBReachingDefs[MBBNumber].resize(NumRegUnits); 49 50 // Reset instruction counter in each basic block. 51 CurInstr = 0; 52 53 // Set up LiveRegs to represent registers entering MBB. 54 // Default values are 'nothing happened a long time ago'. 55 if (LiveRegs.empty()) 56 LiveRegs.assign(NumRegUnits, ReachingDefDefaultVal); 57 58 // This is the entry block. 59 if (MBB->pred_empty()) { 60 for (const auto &LI : MBB->liveins()) { 61 for (MCRegUnitIterator Unit(LI.PhysReg, TRI); Unit.isValid(); ++Unit) { 62 // Treat function live-ins as if they were defined just before the first 63 // instruction. Usually, function arguments are set up immediately 64 // before the call. 65 if (LiveRegs[*Unit] != -1) { 66 LiveRegs[*Unit] = -1; 67 MBBReachingDefs[MBBNumber][*Unit].push_back(-1); 68 } 69 } 70 } 71 LLVM_DEBUG(dbgs() << printMBBReference(*MBB) << ": entry\n"); 72 return; 73 } 74 75 // Try to coalesce live-out registers from predecessors. 76 for (MachineBasicBlock *pred : MBB->predecessors()) { 77 assert(unsigned(pred->getNumber()) < MBBOutRegsInfos.size() && 78 "Should have pre-allocated MBBInfos for all MBBs"); 79 const LiveRegsDefInfo &Incoming = MBBOutRegsInfos[pred->getNumber()]; 80 // Incoming is null if this is a backedge from a BB 81 // we haven't processed yet 82 if (Incoming.empty()) 83 continue; 84 85 // Find the most recent reaching definition from a predecessor. 86 for (unsigned Unit = 0; Unit != NumRegUnits; ++Unit) 87 LiveRegs[Unit] = std::max(LiveRegs[Unit], Incoming[Unit]); 88 } 89 90 // Insert the most recent reaching definition we found. 91 for (unsigned Unit = 0; Unit != NumRegUnits; ++Unit) 92 if (LiveRegs[Unit] != ReachingDefDefaultVal) 93 MBBReachingDefs[MBBNumber][Unit].push_back(LiveRegs[Unit]); 94 } 95 96 void ReachingDefAnalysis::leaveBasicBlock(MachineBasicBlock *MBB) { 97 assert(!LiveRegs.empty() && "Must enter basic block first."); 98 unsigned MBBNumber = MBB->getNumber(); 99 assert(MBBNumber < MBBOutRegsInfos.size() && 100 "Unexpected basic block number."); 101 // Save register clearances at end of MBB - used by enterBasicBlock(). 102 MBBOutRegsInfos[MBBNumber] = LiveRegs; 103 104 // While processing the basic block, we kept `Def` relative to the start 105 // of the basic block for convenience. However, future use of this information 106 // only cares about the clearance from the end of the block, so adjust 107 // everything to be relative to the end of the basic block. 108 for (int &OutLiveReg : MBBOutRegsInfos[MBBNumber]) 109 if (OutLiveReg != ReachingDefDefaultVal) 110 OutLiveReg -= CurInstr; 111 LiveRegs.clear(); 112 } 113 114 void ReachingDefAnalysis::processDefs(MachineInstr *MI) { 115 assert(!MI->isDebugInstr() && "Won't process debug instructions"); 116 117 unsigned MBBNumber = MI->getParent()->getNumber(); 118 assert(MBBNumber < MBBReachingDefs.size() && 119 "Unexpected basic block number."); 120 121 for (auto &MO : MI->operands()) { 122 if (!isValidRegDef(MO)) 123 continue; 124 for (MCRegUnitIterator Unit(MO.getReg(), TRI); Unit.isValid(); ++Unit) { 125 // This instruction explicitly defines the current reg unit. 126 LLVM_DEBUG(dbgs() << printReg(*Unit, TRI) << ":\t" << CurInstr 127 << '\t' << *MI); 128 129 // How many instructions since this reg unit was last written? 130 if (LiveRegs[*Unit] != CurInstr) { 131 LiveRegs[*Unit] = CurInstr; 132 MBBReachingDefs[MBBNumber][*Unit].push_back(CurInstr); 133 } 134 } 135 } 136 InstIds[MI] = CurInstr; 137 ++CurInstr; 138 } 139 140 void ReachingDefAnalysis::reprocessBasicBlock(MachineBasicBlock *MBB) { 141 unsigned MBBNumber = MBB->getNumber(); 142 assert(MBBNumber < MBBReachingDefs.size() && 143 "Unexpected basic block number."); 144 145 // Count number of non-debug instructions for end of block adjustment. 146 auto NonDbgInsts = 147 instructionsWithoutDebug(MBB->instr_begin(), MBB->instr_end()); 148 int NumInsts = std::distance(NonDbgInsts.begin(), NonDbgInsts.end()); 149 150 // When reprocessing a block, the only thing we need to do is check whether 151 // there is now a more recent incoming reaching definition from a predecessor. 152 for (MachineBasicBlock *pred : MBB->predecessors()) { 153 assert(unsigned(pred->getNumber()) < MBBOutRegsInfos.size() && 154 "Should have pre-allocated MBBInfos for all MBBs"); 155 const LiveRegsDefInfo &Incoming = MBBOutRegsInfos[pred->getNumber()]; 156 // Incoming may be empty for dead predecessors. 157 if (Incoming.empty()) 158 continue; 159 160 for (unsigned Unit = 0; Unit != NumRegUnits; ++Unit) { 161 int Def = Incoming[Unit]; 162 if (Def == ReachingDefDefaultVal) 163 continue; 164 165 auto Start = MBBReachingDefs[MBBNumber][Unit].begin(); 166 if (Start != MBBReachingDefs[MBBNumber][Unit].end() && *Start < 0) { 167 if (*Start >= Def) 168 continue; 169 170 // Update existing reaching def from predecessor to a more recent one. 171 *Start = Def; 172 } else { 173 // Insert new reaching def from predecessor. 174 MBBReachingDefs[MBBNumber][Unit].insert(Start, Def); 175 } 176 177 // Update reaching def at end of of BB. Keep in mind that these are 178 // adjusted relative to the end of the basic block. 179 if (MBBOutRegsInfos[MBBNumber][Unit] < Def - NumInsts) 180 MBBOutRegsInfos[MBBNumber][Unit] = Def - NumInsts; 181 } 182 } 183 } 184 185 void ReachingDefAnalysis::processBasicBlock( 186 const LoopTraversal::TraversedMBBInfo &TraversedMBB) { 187 MachineBasicBlock *MBB = TraversedMBB.MBB; 188 LLVM_DEBUG(dbgs() << printMBBReference(*MBB) 189 << (!TraversedMBB.IsDone ? ": incomplete\n" 190 : ": all preds known\n")); 191 192 if (!TraversedMBB.PrimaryPass) { 193 // Reprocess MBB that is part of a loop. 194 reprocessBasicBlock(MBB); 195 return; 196 } 197 198 enterBasicBlock(MBB); 199 for (MachineInstr &MI : 200 instructionsWithoutDebug(MBB->instr_begin(), MBB->instr_end())) 201 processDefs(&MI); 202 leaveBasicBlock(MBB); 203 } 204 205 bool ReachingDefAnalysis::runOnMachineFunction(MachineFunction &mf) { 206 MF = &mf; 207 TRI = MF->getSubtarget().getRegisterInfo(); 208 LLVM_DEBUG(dbgs() << "********** REACHING DEFINITION ANALYSIS **********\n"); 209 init(); 210 traverse(); 211 return false; 212 } 213 214 void ReachingDefAnalysis::releaseMemory() { 215 // Clear the internal vectors. 216 MBBOutRegsInfos.clear(); 217 MBBReachingDefs.clear(); 218 InstIds.clear(); 219 LiveRegs.clear(); 220 } 221 222 void ReachingDefAnalysis::reset() { 223 releaseMemory(); 224 init(); 225 traverse(); 226 } 227 228 void ReachingDefAnalysis::init() { 229 NumRegUnits = TRI->getNumRegUnits(); 230 MBBReachingDefs.resize(MF->getNumBlockIDs()); 231 // Initialize the MBBOutRegsInfos 232 MBBOutRegsInfos.resize(MF->getNumBlockIDs()); 233 LoopTraversal Traversal; 234 TraversedMBBOrder = Traversal.traverse(*MF); 235 } 236 237 void ReachingDefAnalysis::traverse() { 238 // Traverse the basic blocks. 239 for (LoopTraversal::TraversedMBBInfo TraversedMBB : TraversedMBBOrder) 240 processBasicBlock(TraversedMBB); 241 #ifndef NDEBUG 242 // Make sure reaching defs are sorted and unique. 243 for (MBBDefsInfo &MBBDefs : MBBReachingDefs) { 244 for (MBBRegUnitDefs &RegUnitDefs : MBBDefs) { 245 int LastDef = ReachingDefDefaultVal; 246 for (int Def : RegUnitDefs) { 247 assert(Def > LastDef && "Defs must be sorted and unique"); 248 LastDef = Def; 249 } 250 } 251 } 252 #endif 253 } 254 255 int ReachingDefAnalysis::getReachingDef(MachineInstr *MI, int PhysReg) const { 256 assert(InstIds.count(MI) && "Unexpected machine instuction."); 257 int InstId = InstIds.lookup(MI); 258 int DefRes = ReachingDefDefaultVal; 259 unsigned MBBNumber = MI->getParent()->getNumber(); 260 assert(MBBNumber < MBBReachingDefs.size() && 261 "Unexpected basic block number."); 262 int LatestDef = ReachingDefDefaultVal; 263 for (MCRegUnitIterator Unit(PhysReg, TRI); Unit.isValid(); ++Unit) { 264 for (int Def : MBBReachingDefs[MBBNumber][*Unit]) { 265 if (Def >= InstId) 266 break; 267 DefRes = Def; 268 } 269 LatestDef = std::max(LatestDef, DefRes); 270 } 271 return LatestDef; 272 } 273 274 MachineInstr* ReachingDefAnalysis::getReachingLocalMIDef(MachineInstr *MI, 275 int PhysReg) const { 276 return getInstFromId(MI->getParent(), getReachingDef(MI, PhysReg)); 277 } 278 279 bool ReachingDefAnalysis::hasSameReachingDef(MachineInstr *A, MachineInstr *B, 280 int PhysReg) const { 281 MachineBasicBlock *ParentA = A->getParent(); 282 MachineBasicBlock *ParentB = B->getParent(); 283 if (ParentA != ParentB) 284 return false; 285 286 return getReachingDef(A, PhysReg) == getReachingDef(B, PhysReg); 287 } 288 289 MachineInstr *ReachingDefAnalysis::getInstFromId(MachineBasicBlock *MBB, 290 int InstId) const { 291 assert(static_cast<size_t>(MBB->getNumber()) < MBBReachingDefs.size() && 292 "Unexpected basic block number."); 293 assert(InstId < static_cast<int>(MBB->size()) && 294 "Unexpected instruction id."); 295 296 if (InstId < 0) 297 return nullptr; 298 299 for (auto &MI : *MBB) { 300 auto F = InstIds.find(&MI); 301 if (F != InstIds.end() && F->second == InstId) 302 return &MI; 303 } 304 305 return nullptr; 306 } 307 308 int 309 ReachingDefAnalysis::getClearance(MachineInstr *MI, MCPhysReg PhysReg) const { 310 assert(InstIds.count(MI) && "Unexpected machine instuction."); 311 return InstIds.lookup(MI) - getReachingDef(MI, PhysReg); 312 } 313 314 bool 315 ReachingDefAnalysis::hasLocalDefBefore(MachineInstr *MI, int PhysReg) const { 316 return getReachingDef(MI, PhysReg) >= 0; 317 } 318 319 void ReachingDefAnalysis::getReachingLocalUses(MachineInstr *Def, int PhysReg, 320 InstSet &Uses) const { 321 MachineBasicBlock *MBB = Def->getParent(); 322 MachineBasicBlock::iterator MI = MachineBasicBlock::iterator(Def); 323 while (++MI != MBB->end()) { 324 if (MI->isDebugInstr()) 325 continue; 326 327 // If/when we find a new reaching def, we know that there's no more uses 328 // of 'Def'. 329 if (getReachingLocalMIDef(&*MI, PhysReg) != Def) 330 return; 331 332 for (auto &MO : MI->operands()) { 333 if (!isValidRegUseOf(MO, PhysReg)) 334 continue; 335 336 Uses.insert(&*MI); 337 if (MO.isKill()) 338 return; 339 } 340 } 341 } 342 343 bool 344 ReachingDefAnalysis::getLiveInUses(MachineBasicBlock *MBB, int PhysReg, 345 InstSet &Uses) const { 346 for (MachineInstr &MI : 347 instructionsWithoutDebug(MBB->instr_begin(), MBB->instr_end())) { 348 for (auto &MO : MI.operands()) { 349 if (!isValidRegUseOf(MO, PhysReg)) 350 continue; 351 if (getReachingDef(&MI, PhysReg) >= 0) 352 return false; 353 Uses.insert(&MI); 354 } 355 } 356 MachineInstr *Last = &*MBB->getLastNonDebugInstr(); 357 return isReachingDefLiveOut(Last, PhysReg); 358 } 359 360 void 361 ReachingDefAnalysis::getGlobalUses(MachineInstr *MI, int PhysReg, 362 InstSet &Uses) const { 363 MachineBasicBlock *MBB = MI->getParent(); 364 365 // Collect the uses that each def touches within the block. 366 getReachingLocalUses(MI, PhysReg, Uses); 367 368 // Handle live-out values. 369 if (auto *LiveOut = getLocalLiveOutMIDef(MI->getParent(), PhysReg)) { 370 if (LiveOut != MI) 371 return; 372 373 SmallVector<MachineBasicBlock*, 4> ToVisit; 374 ToVisit.insert(ToVisit.begin(), MBB->successors().begin(), 375 MBB->successors().end()); 376 SmallPtrSet<MachineBasicBlock*, 4>Visited; 377 while (!ToVisit.empty()) { 378 MachineBasicBlock *MBB = ToVisit.back(); 379 ToVisit.pop_back(); 380 if (Visited.count(MBB) || !MBB->isLiveIn(PhysReg)) 381 continue; 382 if (getLiveInUses(MBB, PhysReg, Uses)) 383 ToVisit.insert(ToVisit.end(), MBB->successors().begin(), 384 MBB->successors().end()); 385 Visited.insert(MBB); 386 } 387 } 388 } 389 390 void ReachingDefAnalysis::getLiveOuts(MachineBasicBlock *MBB, int PhysReg, 391 InstSet &Defs) const { 392 SmallPtrSet<MachineBasicBlock*, 2> VisitedBBs; 393 getLiveOuts(MBB, PhysReg, Defs, VisitedBBs); 394 } 395 396 void 397 ReachingDefAnalysis::getLiveOuts(MachineBasicBlock *MBB, int PhysReg, 398 InstSet &Defs, BlockSet &VisitedBBs) const { 399 if (VisitedBBs.count(MBB)) 400 return; 401 402 VisitedBBs.insert(MBB); 403 LivePhysRegs LiveRegs(*TRI); 404 LiveRegs.addLiveOuts(*MBB); 405 if (!LiveRegs.contains(PhysReg)) 406 return; 407 408 if (auto *Def = getLocalLiveOutMIDef(MBB, PhysReg)) 409 Defs.insert(Def); 410 else 411 for (auto *Pred : MBB->predecessors()) 412 getLiveOuts(Pred, PhysReg, Defs, VisitedBBs); 413 } 414 415 MachineInstr *ReachingDefAnalysis::getUniqueReachingMIDef(MachineInstr *MI, 416 int PhysReg) const { 417 // If there's a local def before MI, return it. 418 MachineInstr *LocalDef = getReachingLocalMIDef(MI, PhysReg); 419 if (LocalDef && InstIds.lookup(LocalDef) < InstIds.lookup(MI)) 420 return LocalDef; 421 422 SmallPtrSet<MachineBasicBlock*, 4> VisitedBBs; 423 SmallPtrSet<MachineInstr*, 2> Incoming; 424 for (auto *Pred : MI->getParent()->predecessors()) 425 getLiveOuts(Pred, PhysReg, Incoming, VisitedBBs); 426 427 // If we have a local def and an incoming instruction, then there's not a 428 // unique instruction def. 429 if (!Incoming.empty() && LocalDef) 430 return nullptr; 431 else if (Incoming.size() == 1) 432 return *Incoming.begin(); 433 else 434 return LocalDef; 435 } 436 437 MachineInstr *ReachingDefAnalysis::getMIOperand(MachineInstr *MI, 438 unsigned Idx) const { 439 assert(MI->getOperand(Idx).isReg() && "Expected register operand"); 440 return getUniqueReachingMIDef(MI, MI->getOperand(Idx).getReg()); 441 } 442 443 MachineInstr *ReachingDefAnalysis::getMIOperand(MachineInstr *MI, 444 MachineOperand &MO) const { 445 assert(MO.isReg() && "Expected register operand"); 446 return getUniqueReachingMIDef(MI, MO.getReg()); 447 } 448 449 bool ReachingDefAnalysis::isRegUsedAfter(MachineInstr *MI, int PhysReg) const { 450 MachineBasicBlock *MBB = MI->getParent(); 451 LivePhysRegs LiveRegs(*TRI); 452 LiveRegs.addLiveOuts(*MBB); 453 454 // Yes if the register is live out of the basic block. 455 if (LiveRegs.contains(PhysReg)) 456 return true; 457 458 // Walk backwards through the block to see if the register is live at some 459 // point. 460 for (MachineInstr &Last : 461 instructionsWithoutDebug(MBB->instr_rbegin(), MBB->instr_rend())) { 462 LiveRegs.stepBackward(Last); 463 if (LiveRegs.contains(PhysReg)) 464 return InstIds.lookup(&Last) > InstIds.lookup(MI); 465 } 466 return false; 467 } 468 469 bool ReachingDefAnalysis::isRegDefinedAfter(MachineInstr *MI, 470 int PhysReg) const { 471 MachineBasicBlock *MBB = MI->getParent(); 472 MachineInstr *Last = &*MBB->getLastNonDebugInstr(); 473 if (getReachingDef(MI, PhysReg) != getReachingDef(Last, PhysReg)) 474 return true; 475 476 if (auto *Def = getLocalLiveOutMIDef(MBB, PhysReg)) 477 return Def == getReachingLocalMIDef(MI, PhysReg); 478 479 return false; 480 } 481 482 bool 483 ReachingDefAnalysis::isReachingDefLiveOut(MachineInstr *MI, int PhysReg) const { 484 MachineBasicBlock *MBB = MI->getParent(); 485 LivePhysRegs LiveRegs(*TRI); 486 LiveRegs.addLiveOuts(*MBB); 487 if (!LiveRegs.contains(PhysReg)) 488 return false; 489 490 MachineInstr *Last = &*MBB->getLastNonDebugInstr(); 491 int Def = getReachingDef(MI, PhysReg); 492 if (getReachingDef(Last, PhysReg) != Def) 493 return false; 494 495 // Finally check that the last instruction doesn't redefine the register. 496 for (auto &MO : Last->operands()) 497 if (isValidRegDefOf(MO, PhysReg)) 498 return false; 499 500 return true; 501 } 502 503 MachineInstr* ReachingDefAnalysis::getLocalLiveOutMIDef(MachineBasicBlock *MBB, 504 int PhysReg) const { 505 LivePhysRegs LiveRegs(*TRI); 506 LiveRegs.addLiveOuts(*MBB); 507 if (!LiveRegs.contains(PhysReg)) 508 return nullptr; 509 510 MachineInstr *Last = &*MBB->getLastNonDebugInstr(); 511 int Def = getReachingDef(Last, PhysReg); 512 for (auto &MO : Last->operands()) 513 if (isValidRegDefOf(MO, PhysReg)) 514 return Last; 515 516 return Def < 0 ? nullptr : getInstFromId(MBB, Def); 517 } 518 519 static bool mayHaveSideEffects(MachineInstr &MI) { 520 return MI.mayLoadOrStore() || MI.mayRaiseFPException() || 521 MI.hasUnmodeledSideEffects() || MI.isTerminator() || 522 MI.isCall() || MI.isBarrier() || MI.isBranch() || MI.isReturn(); 523 } 524 525 // Can we safely move 'From' to just before 'To'? To satisfy this, 'From' must 526 // not define a register that is used by any instructions, after and including, 527 // 'To'. These instructions also must not redefine any of Froms operands. 528 template<typename Iterator> 529 bool ReachingDefAnalysis::isSafeToMove(MachineInstr *From, 530 MachineInstr *To) const { 531 if (From->getParent() != To->getParent()) 532 return false; 533 534 SmallSet<int, 2> Defs; 535 // First check that From would compute the same value if moved. 536 for (auto &MO : From->operands()) { 537 if (!isValidReg(MO)) 538 continue; 539 if (MO.isDef()) 540 Defs.insert(MO.getReg()); 541 else if (!hasSameReachingDef(From, To, MO.getReg())) 542 return false; 543 } 544 545 // Now walk checking that the rest of the instructions will compute the same 546 // value and that we're not overwriting anything. Don't move the instruction 547 // past any memory, control-flow or other ambiguous instructions. 548 for (auto I = ++Iterator(From), E = Iterator(To); I != E; ++I) { 549 if (mayHaveSideEffects(*I)) 550 return false; 551 for (auto &MO : I->operands()) 552 if (MO.isReg() && MO.getReg() && Defs.count(MO.getReg())) 553 return false; 554 } 555 return true; 556 } 557 558 bool ReachingDefAnalysis::isSafeToMoveForwards(MachineInstr *From, 559 MachineInstr *To) const { 560 return isSafeToMove<MachineBasicBlock::reverse_iterator>(From, To); 561 } 562 563 bool ReachingDefAnalysis::isSafeToMoveBackwards(MachineInstr *From, 564 MachineInstr *To) const { 565 return isSafeToMove<MachineBasicBlock::iterator>(From, To); 566 } 567 568 bool ReachingDefAnalysis::isSafeToRemove(MachineInstr *MI, 569 InstSet &ToRemove) const { 570 SmallPtrSet<MachineInstr*, 1> Ignore; 571 SmallPtrSet<MachineInstr*, 2> Visited; 572 return isSafeToRemove(MI, Visited, ToRemove, Ignore); 573 } 574 575 bool 576 ReachingDefAnalysis::isSafeToRemove(MachineInstr *MI, InstSet &ToRemove, 577 InstSet &Ignore) const { 578 SmallPtrSet<MachineInstr*, 2> Visited; 579 return isSafeToRemove(MI, Visited, ToRemove, Ignore); 580 } 581 582 bool 583 ReachingDefAnalysis::isSafeToRemove(MachineInstr *MI, InstSet &Visited, 584 InstSet &ToRemove, InstSet &Ignore) const { 585 if (Visited.count(MI) || Ignore.count(MI)) 586 return true; 587 else if (mayHaveSideEffects(*MI)) { 588 // Unless told to ignore the instruction, don't remove anything which has 589 // side effects. 590 return false; 591 } 592 593 Visited.insert(MI); 594 for (auto &MO : MI->operands()) { 595 if (!isValidRegDef(MO)) 596 continue; 597 598 SmallPtrSet<MachineInstr*, 4> Uses; 599 getGlobalUses(MI, MO.getReg(), Uses); 600 601 for (auto I : Uses) { 602 if (Ignore.count(I) || ToRemove.count(I)) 603 continue; 604 if (!isSafeToRemove(I, Visited, ToRemove, Ignore)) 605 return false; 606 } 607 } 608 ToRemove.insert(MI); 609 return true; 610 } 611 612 void ReachingDefAnalysis::collectKilledOperands(MachineInstr *MI, 613 InstSet &Dead) const { 614 Dead.insert(MI); 615 auto IsDead = [this, &Dead](MachineInstr *Def, int PhysReg) { 616 unsigned LiveDefs = 0; 617 for (auto &MO : Def->operands()) { 618 if (!isValidRegDef(MO)) 619 continue; 620 if (!MO.isDead()) 621 ++LiveDefs; 622 } 623 624 if (LiveDefs > 1) 625 return false; 626 627 SmallPtrSet<MachineInstr*, 4> Uses; 628 getGlobalUses(Def, PhysReg, Uses); 629 for (auto *Use : Uses) 630 if (!Dead.count(Use)) 631 return false; 632 return true; 633 }; 634 635 for (auto &MO : MI->operands()) { 636 if (!isValidRegUse(MO)) 637 continue; 638 if (MachineInstr *Def = getMIOperand(MI, MO)) 639 if (IsDead(Def, MO.getReg())) 640 collectKilledOperands(Def, Dead); 641 } 642 } 643 644 bool ReachingDefAnalysis::isSafeToDefRegAt(MachineInstr *MI, 645 int PhysReg) const { 646 SmallPtrSet<MachineInstr*, 1> Ignore; 647 return isSafeToDefRegAt(MI, PhysReg, Ignore); 648 } 649 650 bool ReachingDefAnalysis::isSafeToDefRegAt(MachineInstr *MI, int PhysReg, 651 InstSet &Ignore) const { 652 // Check for any uses of the register after MI. 653 if (isRegUsedAfter(MI, PhysReg)) { 654 if (auto *Def = getReachingLocalMIDef(MI, PhysReg)) { 655 SmallPtrSet<MachineInstr*, 2> Uses; 656 getReachingLocalUses(Def, PhysReg, Uses); 657 for (auto *Use : Uses) 658 if (!Ignore.count(Use)) 659 return false; 660 } else 661 return false; 662 } 663 664 MachineBasicBlock *MBB = MI->getParent(); 665 // Check for any defs after MI. 666 if (isRegDefinedAfter(MI, PhysReg)) { 667 auto I = MachineBasicBlock::iterator(MI); 668 for (auto E = MBB->end(); I != E; ++I) { 669 if (Ignore.count(&*I)) 670 continue; 671 for (auto &MO : I->operands()) 672 if (isValidRegDefOf(MO, PhysReg)) 673 return false; 674 } 675 } 676 return true; 677 } 678