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 int NumInsts = 0; 147 for (const MachineInstr &MI : *MBB) 148 if (!MI.isDebugInstr()) 149 NumInsts++; 150 151 // When reprocessing a block, the only thing we need to do is check whether 152 // there is now a more recent incoming reaching definition from a predecessor. 153 for (MachineBasicBlock *pred : MBB->predecessors()) { 154 assert(unsigned(pred->getNumber()) < MBBOutRegsInfos.size() && 155 "Should have pre-allocated MBBInfos for all MBBs"); 156 const LiveRegsDefInfo &Incoming = MBBOutRegsInfos[pred->getNumber()]; 157 // Incoming may be empty for dead predecessors. 158 if (Incoming.empty()) 159 continue; 160 161 for (unsigned Unit = 0; Unit != NumRegUnits; ++Unit) { 162 int Def = Incoming[Unit]; 163 if (Def == ReachingDefDefaultVal) 164 continue; 165 166 auto Start = MBBReachingDefs[MBBNumber][Unit].begin(); 167 if (Start != MBBReachingDefs[MBBNumber][Unit].end() && *Start < 0) { 168 if (*Start >= Def) 169 continue; 170 171 // Update existing reaching def from predecessor to a more recent one. 172 *Start = Def; 173 } else { 174 // Insert new reaching def from predecessor. 175 MBBReachingDefs[MBBNumber][Unit].insert(Start, Def); 176 } 177 178 // Update reaching def at end of of BB. Keep in mind that these are 179 // adjusted relative to the end of the basic block. 180 if (MBBOutRegsInfos[MBBNumber][Unit] < Def - NumInsts) 181 MBBOutRegsInfos[MBBNumber][Unit] = Def - NumInsts; 182 } 183 } 184 } 185 186 void ReachingDefAnalysis::processBasicBlock( 187 const LoopTraversal::TraversedMBBInfo &TraversedMBB) { 188 MachineBasicBlock *MBB = TraversedMBB.MBB; 189 LLVM_DEBUG(dbgs() << printMBBReference(*MBB) 190 << (!TraversedMBB.IsDone ? ": incomplete\n" 191 : ": all preds known\n")); 192 193 if (!TraversedMBB.PrimaryPass) { 194 // Reprocess MBB that is part of a loop. 195 reprocessBasicBlock(MBB); 196 return; 197 } 198 199 enterBasicBlock(MBB); 200 for (MachineInstr &MI : *MBB) { 201 if (!MI.isDebugInstr()) 202 processDefs(&MI); 203 } 204 leaveBasicBlock(MBB); 205 } 206 207 bool ReachingDefAnalysis::runOnMachineFunction(MachineFunction &mf) { 208 MF = &mf; 209 TRI = MF->getSubtarget().getRegisterInfo(); 210 LLVM_DEBUG(dbgs() << "********** REACHING DEFINITION ANALYSIS **********\n"); 211 init(); 212 traverse(); 213 return false; 214 } 215 216 void ReachingDefAnalysis::releaseMemory() { 217 // Clear the internal vectors. 218 MBBOutRegsInfos.clear(); 219 MBBReachingDefs.clear(); 220 InstIds.clear(); 221 LiveRegs.clear(); 222 } 223 224 void ReachingDefAnalysis::reset() { 225 releaseMemory(); 226 init(); 227 traverse(); 228 } 229 230 void ReachingDefAnalysis::init() { 231 NumRegUnits = TRI->getNumRegUnits(); 232 MBBReachingDefs.resize(MF->getNumBlockIDs()); 233 // Initialize the MBBOutRegsInfos 234 MBBOutRegsInfos.resize(MF->getNumBlockIDs()); 235 LoopTraversal Traversal; 236 TraversedMBBOrder = Traversal.traverse(*MF); 237 } 238 239 void ReachingDefAnalysis::traverse() { 240 // Traverse the basic blocks. 241 for (LoopTraversal::TraversedMBBInfo TraversedMBB : TraversedMBBOrder) 242 processBasicBlock(TraversedMBB); 243 #ifndef NDEBUG 244 // Make sure reaching defs are sorted and unique. 245 for (MBBDefsInfo &MBBDefs : MBBReachingDefs) { 246 for (MBBRegUnitDefs &RegUnitDefs : MBBDefs) { 247 int LastDef = ReachingDefDefaultVal; 248 for (int Def : RegUnitDefs) { 249 assert(Def > LastDef && "Defs must be sorted and unique"); 250 LastDef = Def; 251 } 252 } 253 } 254 #endif 255 } 256 257 int ReachingDefAnalysis::getReachingDef(MachineInstr *MI, int PhysReg) const { 258 assert(InstIds.count(MI) && "Unexpected machine instuction."); 259 int InstId = InstIds.lookup(MI); 260 int DefRes = ReachingDefDefaultVal; 261 unsigned MBBNumber = MI->getParent()->getNumber(); 262 assert(MBBNumber < MBBReachingDefs.size() && 263 "Unexpected basic block number."); 264 int LatestDef = ReachingDefDefaultVal; 265 for (MCRegUnitIterator Unit(PhysReg, TRI); Unit.isValid(); ++Unit) { 266 for (int Def : MBBReachingDefs[MBBNumber][*Unit]) { 267 if (Def >= InstId) 268 break; 269 DefRes = Def; 270 } 271 LatestDef = std::max(LatestDef, DefRes); 272 } 273 return LatestDef; 274 } 275 276 MachineInstr* ReachingDefAnalysis::getReachingLocalMIDef(MachineInstr *MI, 277 int PhysReg) const { 278 return getInstFromId(MI->getParent(), getReachingDef(MI, PhysReg)); 279 } 280 281 bool ReachingDefAnalysis::hasSameReachingDef(MachineInstr *A, MachineInstr *B, 282 int PhysReg) const { 283 MachineBasicBlock *ParentA = A->getParent(); 284 MachineBasicBlock *ParentB = B->getParent(); 285 if (ParentA != ParentB) 286 return false; 287 288 return getReachingDef(A, PhysReg) == getReachingDef(B, PhysReg); 289 } 290 291 MachineInstr *ReachingDefAnalysis::getInstFromId(MachineBasicBlock *MBB, 292 int InstId) const { 293 assert(static_cast<size_t>(MBB->getNumber()) < MBBReachingDefs.size() && 294 "Unexpected basic block number."); 295 assert(InstId < static_cast<int>(MBB->size()) && 296 "Unexpected instruction id."); 297 298 if (InstId < 0) 299 return nullptr; 300 301 for (auto &MI : *MBB) { 302 auto F = InstIds.find(&MI); 303 if (F != InstIds.end() && F->second == InstId) 304 return &MI; 305 } 306 307 return nullptr; 308 } 309 310 int 311 ReachingDefAnalysis::getClearance(MachineInstr *MI, MCPhysReg PhysReg) const { 312 assert(InstIds.count(MI) && "Unexpected machine instuction."); 313 return InstIds.lookup(MI) - getReachingDef(MI, PhysReg); 314 } 315 316 bool 317 ReachingDefAnalysis::hasLocalDefBefore(MachineInstr *MI, int PhysReg) const { 318 return getReachingDef(MI, PhysReg) >= 0; 319 } 320 321 void ReachingDefAnalysis::getReachingLocalUses(MachineInstr *Def, int PhysReg, 322 InstSet &Uses) const { 323 MachineBasicBlock *MBB = Def->getParent(); 324 MachineBasicBlock::iterator MI = MachineBasicBlock::iterator(Def); 325 while (++MI != MBB->end()) { 326 if (MI->isDebugInstr()) 327 continue; 328 329 // If/when we find a new reaching def, we know that there's no more uses 330 // of 'Def'. 331 if (getReachingLocalMIDef(&*MI, PhysReg) != Def) 332 return; 333 334 for (auto &MO : MI->operands()) { 335 if (!isValidRegUseOf(MO, PhysReg)) 336 continue; 337 338 Uses.insert(&*MI); 339 if (MO.isKill()) 340 return; 341 } 342 } 343 } 344 345 bool 346 ReachingDefAnalysis::getLiveInUses(MachineBasicBlock *MBB, int PhysReg, 347 InstSet &Uses) const { 348 for (auto &MI : *MBB) { 349 if (MI.isDebugInstr()) 350 continue; 351 for (auto &MO : MI.operands()) { 352 if (!isValidRegUseOf(MO, PhysReg)) 353 continue; 354 if (getReachingDef(&MI, PhysReg) >= 0) 355 return false; 356 Uses.insert(&MI); 357 } 358 } 359 return isReachingDefLiveOut(&MBB->back(), PhysReg); 360 } 361 362 void 363 ReachingDefAnalysis::getGlobalUses(MachineInstr *MI, int PhysReg, 364 InstSet &Uses) const { 365 MachineBasicBlock *MBB = MI->getParent(); 366 367 // Collect the uses that each def touches within the block. 368 getReachingLocalUses(MI, PhysReg, Uses); 369 370 // Handle live-out values. 371 if (auto *LiveOut = getLocalLiveOutMIDef(MI->getParent(), PhysReg)) { 372 if (LiveOut != MI) 373 return; 374 375 SmallVector<MachineBasicBlock*, 4> ToVisit; 376 ToVisit.insert(ToVisit.begin(), MBB->successors().begin(), 377 MBB->successors().end()); 378 SmallPtrSet<MachineBasicBlock*, 4>Visited; 379 while (!ToVisit.empty()) { 380 MachineBasicBlock *MBB = ToVisit.back(); 381 ToVisit.pop_back(); 382 if (Visited.count(MBB) || !MBB->isLiveIn(PhysReg)) 383 continue; 384 if (getLiveInUses(MBB, PhysReg, Uses)) 385 ToVisit.insert(ToVisit.end(), MBB->successors().begin(), 386 MBB->successors().end()); 387 Visited.insert(MBB); 388 } 389 } 390 } 391 392 void 393 ReachingDefAnalysis::getLiveOuts(MachineBasicBlock *MBB, int PhysReg, 394 InstSet &Defs, BlockSet &VisitedBBs) const { 395 if (VisitedBBs.count(MBB)) 396 return; 397 398 VisitedBBs.insert(MBB); 399 LivePhysRegs LiveRegs(*TRI); 400 LiveRegs.addLiveOuts(*MBB); 401 if (!LiveRegs.contains(PhysReg)) 402 return; 403 404 if (auto *Def = getLocalLiveOutMIDef(MBB, PhysReg)) 405 Defs.insert(Def); 406 else 407 for (auto *Pred : MBB->predecessors()) 408 getLiveOuts(Pred, PhysReg, Defs, VisitedBBs); 409 } 410 411 MachineInstr *ReachingDefAnalysis::getUniqueReachingMIDef(MachineInstr *MI, 412 int PhysReg) const { 413 // If there's a local def before MI, return it. 414 MachineInstr *LocalDef = getReachingLocalMIDef(MI, PhysReg); 415 if (LocalDef && InstIds.lookup(LocalDef) < InstIds.lookup(MI)) 416 return LocalDef; 417 418 SmallPtrSet<MachineBasicBlock*, 4> VisitedBBs; 419 SmallPtrSet<MachineInstr*, 2> Incoming; 420 for (auto *Pred : MI->getParent()->predecessors()) 421 getLiveOuts(Pred, PhysReg, Incoming, VisitedBBs); 422 423 // If we have a local def and an incoming instruction, then there's not a 424 // unique instruction def. 425 if (!Incoming.empty() && LocalDef) 426 return nullptr; 427 else if (Incoming.size() == 1) 428 return *Incoming.begin(); 429 else 430 return LocalDef; 431 } 432 433 MachineInstr *ReachingDefAnalysis::getMIOperand(MachineInstr *MI, 434 unsigned Idx) const { 435 assert(MI->getOperand(Idx).isReg() && "Expected register operand"); 436 return getUniqueReachingMIDef(MI, MI->getOperand(Idx).getReg()); 437 } 438 439 MachineInstr *ReachingDefAnalysis::getMIOperand(MachineInstr *MI, 440 MachineOperand &MO) const { 441 assert(MO.isReg() && "Expected register operand"); 442 return getUniqueReachingMIDef(MI, MO.getReg()); 443 } 444 445 bool ReachingDefAnalysis::isRegUsedAfter(MachineInstr *MI, int PhysReg) const { 446 MachineBasicBlock *MBB = MI->getParent(); 447 LivePhysRegs LiveRegs(*TRI); 448 LiveRegs.addLiveOuts(*MBB); 449 450 // Yes if the register is live out of the basic block. 451 if (LiveRegs.contains(PhysReg)) 452 return true; 453 454 // Walk backwards through the block to see if the register is live at some 455 // point. 456 for (auto Last = MBB->rbegin(), End = MBB->rend(); Last != End; ++Last) { 457 LiveRegs.stepBackward(*Last); 458 if (LiveRegs.contains(PhysReg)) 459 return InstIds.lookup(&*Last) > InstIds.lookup(MI); 460 } 461 return false; 462 } 463 464 bool ReachingDefAnalysis::isRegDefinedAfter(MachineInstr *MI, 465 int PhysReg) const { 466 MachineBasicBlock *MBB = MI->getParent(); 467 if (getReachingDef(MI, PhysReg) != getReachingDef(&MBB->back(), PhysReg)) 468 return true; 469 470 if (auto *Def = getLocalLiveOutMIDef(MBB, PhysReg)) 471 return Def == getReachingLocalMIDef(MI, PhysReg); 472 473 return false; 474 } 475 476 bool 477 ReachingDefAnalysis::isReachingDefLiveOut(MachineInstr *MI, int PhysReg) const { 478 MachineBasicBlock *MBB = MI->getParent(); 479 LivePhysRegs LiveRegs(*TRI); 480 LiveRegs.addLiveOuts(*MBB); 481 if (!LiveRegs.contains(PhysReg)) 482 return false; 483 484 MachineInstr *Last = &MBB->back(); 485 int Def = getReachingDef(MI, PhysReg); 486 if (getReachingDef(Last, PhysReg) != Def) 487 return false; 488 489 // Finally check that the last instruction doesn't redefine the register. 490 for (auto &MO : Last->operands()) 491 if (isValidRegDefOf(MO, PhysReg)) 492 return false; 493 494 return true; 495 } 496 497 MachineInstr* ReachingDefAnalysis::getLocalLiveOutMIDef(MachineBasicBlock *MBB, 498 int PhysReg) const { 499 LivePhysRegs LiveRegs(*TRI); 500 LiveRegs.addLiveOuts(*MBB); 501 if (!LiveRegs.contains(PhysReg)) 502 return nullptr; 503 504 MachineInstr *Last = &MBB->back(); 505 int Def = getReachingDef(Last, PhysReg); 506 for (auto &MO : Last->operands()) 507 if (isValidRegDefOf(MO, PhysReg)) 508 return Last; 509 510 return Def < 0 ? nullptr : getInstFromId(MBB, Def); 511 } 512 513 static bool mayHaveSideEffects(MachineInstr &MI) { 514 return MI.mayLoadOrStore() || MI.mayRaiseFPException() || 515 MI.hasUnmodeledSideEffects() || MI.isTerminator() || 516 MI.isCall() || MI.isBarrier() || MI.isBranch() || MI.isReturn(); 517 } 518 519 // Can we safely move 'From' to just before 'To'? To satisfy this, 'From' must 520 // not define a register that is used by any instructions, after and including, 521 // 'To'. These instructions also must not redefine any of Froms operands. 522 template<typename Iterator> 523 bool ReachingDefAnalysis::isSafeToMove(MachineInstr *From, 524 MachineInstr *To) const { 525 if (From->getParent() != To->getParent()) 526 return false; 527 528 SmallSet<int, 2> Defs; 529 // First check that From would compute the same value if moved. 530 for (auto &MO : From->operands()) { 531 if (!isValidReg(MO)) 532 continue; 533 if (MO.isDef()) 534 Defs.insert(MO.getReg()); 535 else if (!hasSameReachingDef(From, To, MO.getReg())) 536 return false; 537 } 538 539 // Now walk checking that the rest of the instructions will compute the same 540 // value and that we're not overwriting anything. Don't move the instruction 541 // past any memory, control-flow or other ambigious instructions. 542 for (auto I = ++Iterator(From), E = Iterator(To); I != E; ++I) { 543 if (mayHaveSideEffects(*I)) 544 return false; 545 for (auto &MO : I->operands()) 546 if (MO.isReg() && MO.getReg() && Defs.count(MO.getReg())) 547 return false; 548 } 549 return true; 550 } 551 552 bool ReachingDefAnalysis::isSafeToMoveForwards(MachineInstr *From, 553 MachineInstr *To) const { 554 return isSafeToMove<MachineBasicBlock::reverse_iterator>(From, To); 555 } 556 557 bool ReachingDefAnalysis::isSafeToMoveBackwards(MachineInstr *From, 558 MachineInstr *To) const { 559 return isSafeToMove<MachineBasicBlock::iterator>(From, To); 560 } 561 562 bool ReachingDefAnalysis::isSafeToRemove(MachineInstr *MI, 563 InstSet &ToRemove) const { 564 SmallPtrSet<MachineInstr*, 1> Ignore; 565 SmallPtrSet<MachineInstr*, 2> Visited; 566 return isSafeToRemove(MI, Visited, ToRemove, Ignore); 567 } 568 569 bool 570 ReachingDefAnalysis::isSafeToRemove(MachineInstr *MI, InstSet &ToRemove, 571 InstSet &Ignore) const { 572 SmallPtrSet<MachineInstr*, 2> Visited; 573 return isSafeToRemove(MI, Visited, ToRemove, Ignore); 574 } 575 576 bool 577 ReachingDefAnalysis::isSafeToRemove(MachineInstr *MI, InstSet &Visited, 578 InstSet &ToRemove, InstSet &Ignore) const { 579 if (Visited.count(MI) || Ignore.count(MI)) 580 return true; 581 else if (mayHaveSideEffects(*MI)) { 582 // Unless told to ignore the instruction, don't remove anything which has 583 // side effects. 584 return false; 585 } 586 587 Visited.insert(MI); 588 for (auto &MO : MI->operands()) { 589 if (!isValidRegDef(MO)) 590 continue; 591 592 SmallPtrSet<MachineInstr*, 4> Uses; 593 getGlobalUses(MI, MO.getReg(), Uses); 594 595 for (auto I : Uses) { 596 if (Ignore.count(I) || ToRemove.count(I)) 597 continue; 598 if (!isSafeToRemove(I, Visited, ToRemove, Ignore)) 599 return false; 600 } 601 } 602 ToRemove.insert(MI); 603 return true; 604 } 605 606 void ReachingDefAnalysis::collectKilledOperands(MachineInstr *MI, 607 InstSet &Dead) const { 608 Dead.insert(MI); 609 auto IsDead = [this, &Dead](MachineInstr *Def, int PhysReg) { 610 unsigned LiveDefs = 0; 611 for (auto &MO : Def->operands()) { 612 if (!isValidRegDef(MO)) 613 continue; 614 if (!MO.isDead()) 615 ++LiveDefs; 616 } 617 618 if (LiveDefs > 1) 619 return false; 620 621 SmallPtrSet<MachineInstr*, 4> Uses; 622 getGlobalUses(Def, PhysReg, Uses); 623 for (auto *Use : Uses) 624 if (!Dead.count(Use)) 625 return false; 626 return true; 627 }; 628 629 for (auto &MO : MI->operands()) { 630 if (!isValidRegUse(MO)) 631 continue; 632 if (MachineInstr *Def = getMIOperand(MI, MO)) 633 if (IsDead(Def, MO.getReg())) 634 collectKilledOperands(Def, Dead); 635 } 636 } 637 638 bool ReachingDefAnalysis::isSafeToDefRegAt(MachineInstr *MI, 639 int PhysReg) const { 640 SmallPtrSet<MachineInstr*, 1> Ignore; 641 return isSafeToDefRegAt(MI, PhysReg, Ignore); 642 } 643 644 bool ReachingDefAnalysis::isSafeToDefRegAt(MachineInstr *MI, int PhysReg, 645 InstSet &Ignore) const { 646 // Check for any uses of the register after MI. 647 if (isRegUsedAfter(MI, PhysReg)) { 648 if (auto *Def = getReachingLocalMIDef(MI, PhysReg)) { 649 SmallPtrSet<MachineInstr*, 2> Uses; 650 getReachingLocalUses(Def, PhysReg, Uses); 651 for (auto *Use : Uses) 652 if (!Ignore.count(Use)) 653 return false; 654 } else 655 return false; 656 } 657 658 MachineBasicBlock *MBB = MI->getParent(); 659 // Check for any defs after MI. 660 if (isRegDefinedAfter(MI, PhysReg)) { 661 auto I = MachineBasicBlock::iterator(MI); 662 for (auto E = MBB->end(); I != E; ++I) { 663 if (Ignore.count(&*I)) 664 continue; 665 for (auto &MO : I->operands()) 666 if (isValidRegDefOf(MO, PhysReg)) 667 return false; 668 } 669 } 670 return true; 671 } 672