Home | History | Annotate | Line # | Download | only in CodeGen
      1 //===- lib/Codegen/MachineRegisterInfo.cpp --------------------------------===//
      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 // Implementation of the MachineRegisterInfo class.
     10 //
     11 //===----------------------------------------------------------------------===//
     12 
     13 #include "llvm/CodeGen/MachineRegisterInfo.h"
     14 #include "llvm/ADT/iterator_range.h"
     15 #include "llvm/CodeGen/LowLevelType.h"
     16 #include "llvm/CodeGen/MachineBasicBlock.h"
     17 #include "llvm/CodeGen/MachineFunction.h"
     18 #include "llvm/CodeGen/MachineInstr.h"
     19 #include "llvm/CodeGen/MachineInstrBuilder.h"
     20 #include "llvm/CodeGen/MachineOperand.h"
     21 #include "llvm/CodeGen/TargetInstrInfo.h"
     22 #include "llvm/CodeGen/TargetRegisterInfo.h"
     23 #include "llvm/CodeGen/TargetSubtargetInfo.h"
     24 #include "llvm/Config/llvm-config.h"
     25 #include "llvm/IR/Attributes.h"
     26 #include "llvm/IR/DebugLoc.h"
     27 #include "llvm/IR/Function.h"
     28 #include "llvm/MC/MCRegisterInfo.h"
     29 #include "llvm/Support/Casting.h"
     30 #include "llvm/Support/CommandLine.h"
     31 #include "llvm/Support/Compiler.h"
     32 #include "llvm/Support/ErrorHandling.h"
     33 #include "llvm/Support/raw_ostream.h"
     34 #include <cassert>
     35 
     36 using namespace llvm;
     37 
     38 static cl::opt<bool> EnableSubRegLiveness("enable-subreg-liveness", cl::Hidden,
     39   cl::init(true), cl::desc("Enable subregister liveness tracking."));
     40 
     41 // Pin the vtable to this file.
     42 void MachineRegisterInfo::Delegate::anchor() {}
     43 
     44 MachineRegisterInfo::MachineRegisterInfo(MachineFunction *MF)
     45     : MF(MF), TracksSubRegLiveness(MF->getSubtarget().enableSubRegLiveness() &&
     46                                    EnableSubRegLiveness),
     47       IsUpdatedCSRsInitialized(false) {
     48   unsigned NumRegs = getTargetRegisterInfo()->getNumRegs();
     49   VRegInfo.reserve(256);
     50   RegAllocHints.reserve(256);
     51   UsedPhysRegMask.resize(NumRegs);
     52   PhysRegUseDefLists.reset(new MachineOperand*[NumRegs]());
     53 }
     54 
     55 /// setRegClass - Set the register class of the specified virtual register.
     56 ///
     57 void
     58 MachineRegisterInfo::setRegClass(Register Reg, const TargetRegisterClass *RC) {
     59   assert(RC && RC->isAllocatable() && "Invalid RC for virtual register");
     60   VRegInfo[Reg].first = RC;
     61 }
     62 
     63 void MachineRegisterInfo::setRegBank(Register Reg,
     64                                      const RegisterBank &RegBank) {
     65   VRegInfo[Reg].first = &RegBank;
     66 }
     67 
     68 static const TargetRegisterClass *
     69 constrainRegClass(MachineRegisterInfo &MRI, Register Reg,
     70                   const TargetRegisterClass *OldRC,
     71                   const TargetRegisterClass *RC, unsigned MinNumRegs) {
     72   if (OldRC == RC)
     73     return RC;
     74   const TargetRegisterClass *NewRC =
     75       MRI.getTargetRegisterInfo()->getCommonSubClass(OldRC, RC);
     76   if (!NewRC || NewRC == OldRC)
     77     return NewRC;
     78   if (NewRC->getNumRegs() < MinNumRegs)
     79     return nullptr;
     80   MRI.setRegClass(Reg, NewRC);
     81   return NewRC;
     82 }
     83 
     84 const TargetRegisterClass *
     85 MachineRegisterInfo::constrainRegClass(Register Reg,
     86                                        const TargetRegisterClass *RC,
     87                                        unsigned MinNumRegs) {
     88   return ::constrainRegClass(*this, Reg, getRegClass(Reg), RC, MinNumRegs);
     89 }
     90 
     91 bool
     92 MachineRegisterInfo::constrainRegAttrs(Register Reg,
     93                                        Register ConstrainingReg,
     94                                        unsigned MinNumRegs) {
     95   const LLT RegTy = getType(Reg);
     96   const LLT ConstrainingRegTy = getType(ConstrainingReg);
     97   if (RegTy.isValid() && ConstrainingRegTy.isValid() &&
     98       RegTy != ConstrainingRegTy)
     99     return false;
    100   const auto ConstrainingRegCB = getRegClassOrRegBank(ConstrainingReg);
    101   if (!ConstrainingRegCB.isNull()) {
    102     const auto RegCB = getRegClassOrRegBank(Reg);
    103     if (RegCB.isNull())
    104       setRegClassOrRegBank(Reg, ConstrainingRegCB);
    105     else if (RegCB.is<const TargetRegisterClass *>() !=
    106              ConstrainingRegCB.is<const TargetRegisterClass *>())
    107       return false;
    108     else if (RegCB.is<const TargetRegisterClass *>()) {
    109       if (!::constrainRegClass(
    110               *this, Reg, RegCB.get<const TargetRegisterClass *>(),
    111               ConstrainingRegCB.get<const TargetRegisterClass *>(), MinNumRegs))
    112         return false;
    113     } else if (RegCB != ConstrainingRegCB)
    114       return false;
    115   }
    116   if (ConstrainingRegTy.isValid())
    117     setType(Reg, ConstrainingRegTy);
    118   return true;
    119 }
    120 
    121 bool
    122 MachineRegisterInfo::recomputeRegClass(Register Reg) {
    123   const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
    124   const TargetRegisterClass *OldRC = getRegClass(Reg);
    125   const TargetRegisterClass *NewRC =
    126       getTargetRegisterInfo()->getLargestLegalSuperClass(OldRC, *MF);
    127 
    128   // Stop early if there is no room to grow.
    129   if (NewRC == OldRC)
    130     return false;
    131 
    132   // Accumulate constraints from all uses.
    133   for (MachineOperand &MO : reg_nodbg_operands(Reg)) {
    134     // Apply the effect of the given operand to NewRC.
    135     MachineInstr *MI = MO.getParent();
    136     unsigned OpNo = &MO - &MI->getOperand(0);
    137     NewRC = MI->getRegClassConstraintEffect(OpNo, NewRC, TII,
    138                                             getTargetRegisterInfo());
    139     if (!NewRC || NewRC == OldRC)
    140       return false;
    141   }
    142   setRegClass(Reg, NewRC);
    143   return true;
    144 }
    145 
    146 Register MachineRegisterInfo::createIncompleteVirtualRegister(StringRef Name) {
    147   Register Reg = Register::index2VirtReg(getNumVirtRegs());
    148   VRegInfo.grow(Reg);
    149   RegAllocHints.grow(Reg);
    150   insertVRegByName(Name, Reg);
    151   return Reg;
    152 }
    153 
    154 /// createVirtualRegister - Create and return a new virtual register in the
    155 /// function with the specified register class.
    156 ///
    157 Register
    158 MachineRegisterInfo::createVirtualRegister(const TargetRegisterClass *RegClass,
    159                                            StringRef Name) {
    160   assert(RegClass && "Cannot create register without RegClass!");
    161   assert(RegClass->isAllocatable() &&
    162          "Virtual register RegClass must be allocatable.");
    163 
    164   // New virtual register number.
    165   Register Reg = createIncompleteVirtualRegister(Name);
    166   VRegInfo[Reg].first = RegClass;
    167   if (TheDelegate)
    168     TheDelegate->MRI_NoteNewVirtualRegister(Reg);
    169   return Reg;
    170 }
    171 
    172 Register MachineRegisterInfo::cloneVirtualRegister(Register VReg,
    173                                                    StringRef Name) {
    174   Register Reg = createIncompleteVirtualRegister(Name);
    175   VRegInfo[Reg].first = VRegInfo[VReg].first;
    176   setType(Reg, getType(VReg));
    177   if (TheDelegate)
    178     TheDelegate->MRI_NoteNewVirtualRegister(Reg);
    179   return Reg;
    180 }
    181 
    182 void MachineRegisterInfo::setType(Register VReg, LLT Ty) {
    183   VRegToType.grow(VReg);
    184   VRegToType[VReg] = Ty;
    185 }
    186 
    187 Register
    188 MachineRegisterInfo::createGenericVirtualRegister(LLT Ty, StringRef Name) {
    189   // New virtual register number.
    190   Register Reg = createIncompleteVirtualRegister(Name);
    191   // FIXME: Should we use a dummy register class?
    192   VRegInfo[Reg].first = static_cast<RegisterBank *>(nullptr);
    193   setType(Reg, Ty);
    194   if (TheDelegate)
    195     TheDelegate->MRI_NoteNewVirtualRegister(Reg);
    196   return Reg;
    197 }
    198 
    199 void MachineRegisterInfo::clearVirtRegTypes() { VRegToType.clear(); }
    200 
    201 /// clearVirtRegs - Remove all virtual registers (after physreg assignment).
    202 void MachineRegisterInfo::clearVirtRegs() {
    203 #ifndef NDEBUG
    204   for (unsigned i = 0, e = getNumVirtRegs(); i != e; ++i) {
    205     Register Reg = Register::index2VirtReg(i);
    206     if (!VRegInfo[Reg].second)
    207       continue;
    208     verifyUseList(Reg);
    209     llvm_unreachable("Remaining virtual register operands");
    210   }
    211 #endif
    212   VRegInfo.clear();
    213   for (auto &I : LiveIns)
    214     I.second = 0;
    215 }
    216 
    217 void MachineRegisterInfo::verifyUseList(Register Reg) const {
    218 #ifndef NDEBUG
    219   bool Valid = true;
    220   for (MachineOperand &M : reg_operands(Reg)) {
    221     MachineOperand *MO = &M;
    222     MachineInstr *MI = MO->getParent();
    223     if (!MI) {
    224       errs() << printReg(Reg, getTargetRegisterInfo())
    225              << " use list MachineOperand " << MO
    226              << " has no parent instruction.\n";
    227       Valid = false;
    228       continue;
    229     }
    230     MachineOperand *MO0 = &MI->getOperand(0);
    231     unsigned NumOps = MI->getNumOperands();
    232     if (!(MO >= MO0 && MO < MO0+NumOps)) {
    233       errs() << printReg(Reg, getTargetRegisterInfo())
    234              << " use list MachineOperand " << MO
    235              << " doesn't belong to parent MI: " << *MI;
    236       Valid = false;
    237     }
    238     if (!MO->isReg()) {
    239       errs() << printReg(Reg, getTargetRegisterInfo())
    240              << " MachineOperand " << MO << ": " << *MO
    241              << " is not a register\n";
    242       Valid = false;
    243     }
    244     if (MO->getReg() != Reg) {
    245       errs() << printReg(Reg, getTargetRegisterInfo())
    246              << " use-list MachineOperand " << MO << ": "
    247              << *MO << " is the wrong register\n";
    248       Valid = false;
    249     }
    250   }
    251   assert(Valid && "Invalid use list");
    252 #endif
    253 }
    254 
    255 void MachineRegisterInfo::verifyUseLists() const {
    256 #ifndef NDEBUG
    257   for (unsigned i = 0, e = getNumVirtRegs(); i != e; ++i)
    258     verifyUseList(Register::index2VirtReg(i));
    259   for (unsigned i = 1, e = getTargetRegisterInfo()->getNumRegs(); i != e; ++i)
    260     verifyUseList(i);
    261 #endif
    262 }
    263 
    264 /// Add MO to the linked list of operands for its register.
    265 void MachineRegisterInfo::addRegOperandToUseList(MachineOperand *MO) {
    266   assert(!MO->isOnRegUseList() && "Already on list");
    267   MachineOperand *&HeadRef = getRegUseDefListHead(MO->getReg());
    268   MachineOperand *const Head = HeadRef;
    269 
    270   // Head points to the first list element.
    271   // Next is NULL on the last list element.
    272   // Prev pointers are circular, so Head->Prev == Last.
    273 
    274   // Head is NULL for an empty list.
    275   if (!Head) {
    276     MO->Contents.Reg.Prev = MO;
    277     MO->Contents.Reg.Next = nullptr;
    278     HeadRef = MO;
    279     return;
    280   }
    281   assert(MO->getReg() == Head->getReg() && "Different regs on the same list!");
    282 
    283   // Insert MO between Last and Head in the circular Prev chain.
    284   MachineOperand *Last = Head->Contents.Reg.Prev;
    285   assert(Last && "Inconsistent use list");
    286   assert(MO->getReg() == Last->getReg() && "Different regs on the same list!");
    287   Head->Contents.Reg.Prev = MO;
    288   MO->Contents.Reg.Prev = Last;
    289 
    290   // Def operands always precede uses. This allows def_iterator to stop early.
    291   // Insert def operands at the front, and use operands at the back.
    292   if (MO->isDef()) {
    293     // Insert def at the front.
    294     MO->Contents.Reg.Next = Head;
    295     HeadRef = MO;
    296   } else {
    297     // Insert use at the end.
    298     MO->Contents.Reg.Next = nullptr;
    299     Last->Contents.Reg.Next = MO;
    300   }
    301 }
    302 
    303 /// Remove MO from its use-def list.
    304 void MachineRegisterInfo::removeRegOperandFromUseList(MachineOperand *MO) {
    305   assert(MO->isOnRegUseList() && "Operand not on use list");
    306   MachineOperand *&HeadRef = getRegUseDefListHead(MO->getReg());
    307   MachineOperand *const Head = HeadRef;
    308   assert(Head && "List already empty");
    309 
    310   // Unlink this from the doubly linked list of operands.
    311   MachineOperand *Next = MO->Contents.Reg.Next;
    312   MachineOperand *Prev = MO->Contents.Reg.Prev;
    313 
    314   // Prev links are circular, next link is NULL instead of looping back to Head.
    315   if (MO == Head)
    316     HeadRef = Next;
    317   else
    318     Prev->Contents.Reg.Next = Next;
    319 
    320   (Next ? Next : Head)->Contents.Reg.Prev = Prev;
    321 
    322   MO->Contents.Reg.Prev = nullptr;
    323   MO->Contents.Reg.Next = nullptr;
    324 }
    325 
    326 /// Move NumOps operands from Src to Dst, updating use-def lists as needed.
    327 ///
    328 /// The Dst range is assumed to be uninitialized memory. (Or it may contain
    329 /// operands that won't be destroyed, which is OK because the MO destructor is
    330 /// trivial anyway).
    331 ///
    332 /// The Src and Dst ranges may overlap.
    333 void MachineRegisterInfo::moveOperands(MachineOperand *Dst,
    334                                        MachineOperand *Src,
    335                                        unsigned NumOps) {
    336   assert(Src != Dst && NumOps && "Noop moveOperands");
    337 
    338   // Copy backwards if Dst is within the Src range.
    339   int Stride = 1;
    340   if (Dst >= Src && Dst < Src + NumOps) {
    341     Stride = -1;
    342     Dst += NumOps - 1;
    343     Src += NumOps - 1;
    344   }
    345 
    346   // Copy one operand at a time.
    347   do {
    348     new (Dst) MachineOperand(*Src);
    349 
    350     // Dst takes Src's place in the use-def chain.
    351     if (Src->isReg()) {
    352       MachineOperand *&Head = getRegUseDefListHead(Src->getReg());
    353       MachineOperand *Prev = Src->Contents.Reg.Prev;
    354       MachineOperand *Next = Src->Contents.Reg.Next;
    355       assert(Head && "List empty, but operand is chained");
    356       assert(Prev && "Operand was not on use-def list");
    357 
    358       // Prev links are circular, next link is NULL instead of looping back to
    359       // Head.
    360       if (Src == Head)
    361         Head = Dst;
    362       else
    363         Prev->Contents.Reg.Next = Dst;
    364 
    365       // Update Prev pointer. This also works when Src was pointing to itself
    366       // in a 1-element list. In that case Head == Dst.
    367       (Next ? Next : Head)->Contents.Reg.Prev = Dst;
    368     }
    369 
    370     Dst += Stride;
    371     Src += Stride;
    372   } while (--NumOps);
    373 }
    374 
    375 /// replaceRegWith - Replace all instances of FromReg with ToReg in the
    376 /// machine function.  This is like llvm-level X->replaceAllUsesWith(Y),
    377 /// except that it also changes any definitions of the register as well.
    378 /// If ToReg is a physical register we apply the sub register to obtain the
    379 /// final/proper physical register.
    380 void MachineRegisterInfo::replaceRegWith(Register FromReg, Register ToReg) {
    381   assert(FromReg != ToReg && "Cannot replace a reg with itself");
    382 
    383   const TargetRegisterInfo *TRI = getTargetRegisterInfo();
    384 
    385   // TODO: This could be more efficient by bulk changing the operands.
    386   for (reg_iterator I = reg_begin(FromReg), E = reg_end(); I != E; ) {
    387     MachineOperand &O = *I;
    388     ++I;
    389     if (Register::isPhysicalRegister(ToReg)) {
    390       O.substPhysReg(ToReg, *TRI);
    391     } else {
    392       O.setReg(ToReg);
    393     }
    394   }
    395 }
    396 
    397 /// getVRegDef - Return the machine instr that defines the specified virtual
    398 /// register or null if none is found.  This assumes that the code is in SSA
    399 /// form, so there should only be one definition.
    400 MachineInstr *MachineRegisterInfo::getVRegDef(Register Reg) const {
    401   // Since we are in SSA form, we can use the first definition.
    402   def_instr_iterator I = def_instr_begin(Reg);
    403   assert((I.atEnd() || std::next(I) == def_instr_end()) &&
    404          "getVRegDef assumes a single definition or no definition");
    405   return !I.atEnd() ? &*I : nullptr;
    406 }
    407 
    408 /// getUniqueVRegDef - Return the unique machine instr that defines the
    409 /// specified virtual register or null if none is found.  If there are
    410 /// multiple definitions or no definition, return null.
    411 MachineInstr *MachineRegisterInfo::getUniqueVRegDef(Register Reg) const {
    412   if (def_empty(Reg)) return nullptr;
    413   def_instr_iterator I = def_instr_begin(Reg);
    414   if (std::next(I) != def_instr_end())
    415     return nullptr;
    416   return &*I;
    417 }
    418 
    419 bool MachineRegisterInfo::hasOneNonDBGUse(Register RegNo) const {
    420   return hasSingleElement(use_nodbg_operands(RegNo));
    421 }
    422 
    423 bool MachineRegisterInfo::hasOneNonDBGUser(Register RegNo) const {
    424   return hasSingleElement(use_nodbg_instructions(RegNo));
    425 }
    426 
    427 /// clearKillFlags - Iterate over all the uses of the given register and
    428 /// clear the kill flag from the MachineOperand. This function is used by
    429 /// optimization passes which extend register lifetimes and need only
    430 /// preserve conservative kill flag information.
    431 void MachineRegisterInfo::clearKillFlags(Register Reg) const {
    432   for (MachineOperand &MO : use_operands(Reg))
    433     MO.setIsKill(false);
    434 }
    435 
    436 bool MachineRegisterInfo::isLiveIn(Register Reg) const {
    437   for (const std::pair<MCRegister, Register> &LI : liveins())
    438     if ((Register)LI.first == Reg || LI.second == Reg)
    439       return true;
    440   return false;
    441 }
    442 
    443 /// getLiveInPhysReg - If VReg is a live-in virtual register, return the
    444 /// corresponding live-in physical register.
    445 MCRegister MachineRegisterInfo::getLiveInPhysReg(Register VReg) const {
    446   for (const std::pair<MCRegister, Register> &LI : liveins())
    447     if (LI.second == VReg)
    448       return LI.first;
    449   return MCRegister();
    450 }
    451 
    452 /// getLiveInVirtReg - If PReg is a live-in physical register, return the
    453 /// corresponding live-in physical register.
    454 Register MachineRegisterInfo::getLiveInVirtReg(MCRegister PReg) const {
    455   for (const std::pair<MCRegister, Register> &LI : liveins())
    456     if (LI.first == PReg)
    457       return LI.second;
    458   return Register();
    459 }
    460 
    461 /// EmitLiveInCopies - Emit copies to initialize livein virtual registers
    462 /// into the given entry block.
    463 void
    464 MachineRegisterInfo::EmitLiveInCopies(MachineBasicBlock *EntryMBB,
    465                                       const TargetRegisterInfo &TRI,
    466                                       const TargetInstrInfo &TII) {
    467   // Emit the copies into the top of the block.
    468   for (unsigned i = 0, e = LiveIns.size(); i != e; ++i)
    469     if (LiveIns[i].second) {
    470       if (use_nodbg_empty(LiveIns[i].second)) {
    471         // The livein has no non-dbg uses. Drop it.
    472         //
    473         // It would be preferable to have isel avoid creating live-in
    474         // records for unused arguments in the first place, but it's
    475         // complicated by the debug info code for arguments.
    476         LiveIns.erase(LiveIns.begin() + i);
    477         --i; --e;
    478       } else {
    479         // Emit a copy.
    480         BuildMI(*EntryMBB, EntryMBB->begin(), DebugLoc(),
    481                 TII.get(TargetOpcode::COPY), LiveIns[i].second)
    482           .addReg(LiveIns[i].first);
    483 
    484         // Add the register to the entry block live-in set.
    485         EntryMBB->addLiveIn(LiveIns[i].first);
    486       }
    487     } else {
    488       // Add the register to the entry block live-in set.
    489       EntryMBB->addLiveIn(LiveIns[i].first);
    490     }
    491 }
    492 
    493 LaneBitmask MachineRegisterInfo::getMaxLaneMaskForVReg(Register Reg) const {
    494   // Lane masks are only defined for vregs.
    495   assert(Register::isVirtualRegister(Reg));
    496   const TargetRegisterClass &TRC = *getRegClass(Reg);
    497   return TRC.getLaneMask();
    498 }
    499 
    500 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
    501 LLVM_DUMP_METHOD void MachineRegisterInfo::dumpUses(Register Reg) const {
    502   for (MachineInstr &I : use_instructions(Reg))
    503     I.dump();
    504 }
    505 #endif
    506 
    507 void MachineRegisterInfo::freezeReservedRegs(const MachineFunction &MF) {
    508   ReservedRegs = getTargetRegisterInfo()->getReservedRegs(MF);
    509   assert(ReservedRegs.size() == getTargetRegisterInfo()->getNumRegs() &&
    510          "Invalid ReservedRegs vector from target");
    511 }
    512 
    513 bool MachineRegisterInfo::isConstantPhysReg(MCRegister PhysReg) const {
    514   assert(Register::isPhysicalRegister(PhysReg));
    515 
    516   const TargetRegisterInfo *TRI = getTargetRegisterInfo();
    517   if (TRI->isConstantPhysReg(PhysReg))
    518     return true;
    519 
    520   // Check if any overlapping register is modified, or allocatable so it may be
    521   // used later.
    522   for (MCRegAliasIterator AI(PhysReg, TRI, true);
    523        AI.isValid(); ++AI)
    524     if (!def_empty(*AI) || isAllocatable(*AI))
    525       return false;
    526   return true;
    527 }
    528 
    529 /// markUsesInDebugValueAsUndef - Mark every DBG_VALUE referencing the
    530 /// specified register as undefined which causes the DBG_VALUE to be
    531 /// deleted during LiveDebugVariables analysis.
    532 void MachineRegisterInfo::markUsesInDebugValueAsUndef(Register Reg) const {
    533   // Mark any DBG_VALUE* that uses Reg as undef (but don't delete it.)
    534   // We use make_early_inc_range because setReg invalidates the iterator.
    535   for (MachineInstr &UseMI : llvm::make_early_inc_range(use_instructions(Reg))) {
    536     if (UseMI.isDebugValue() && UseMI.hasDebugOperandForReg(Reg))
    537       UseMI.setDebugValueUndef();
    538   }
    539 }
    540 
    541 static const Function *getCalledFunction(const MachineInstr &MI) {
    542   for (const MachineOperand &MO : MI.operands()) {
    543     if (!MO.isGlobal())
    544       continue;
    545     const Function *Func = dyn_cast<Function>(MO.getGlobal());
    546     if (Func != nullptr)
    547       return Func;
    548   }
    549   return nullptr;
    550 }
    551 
    552 static bool isNoReturnDef(const MachineOperand &MO) {
    553   // Anything which is not a noreturn function is a real def.
    554   const MachineInstr &MI = *MO.getParent();
    555   if (!MI.isCall())
    556     return false;
    557   const MachineBasicBlock &MBB = *MI.getParent();
    558   if (!MBB.succ_empty())
    559     return false;
    560   const MachineFunction &MF = *MBB.getParent();
    561   // We need to keep correct unwind information even if the function will
    562   // not return, since the runtime may need it.
    563   if (MF.getFunction().hasFnAttribute(Attribute::UWTable))
    564     return false;
    565   const Function *Called = getCalledFunction(MI);
    566   return !(Called == nullptr || !Called->hasFnAttribute(Attribute::NoReturn) ||
    567            !Called->hasFnAttribute(Attribute::NoUnwind));
    568 }
    569 
    570 bool MachineRegisterInfo::isPhysRegModified(MCRegister PhysReg,
    571                                             bool SkipNoReturnDef) const {
    572   if (UsedPhysRegMask.test(PhysReg))
    573     return true;
    574   const TargetRegisterInfo *TRI = getTargetRegisterInfo();
    575   for (MCRegAliasIterator AI(PhysReg, TRI, true); AI.isValid(); ++AI) {
    576     for (const MachineOperand &MO : make_range(def_begin(*AI), def_end())) {
    577       if (!SkipNoReturnDef && isNoReturnDef(MO))
    578         continue;
    579       return true;
    580     }
    581   }
    582   return false;
    583 }
    584 
    585 bool MachineRegisterInfo::isPhysRegUsed(MCRegister PhysReg) const {
    586   if (UsedPhysRegMask.test(PhysReg))
    587     return true;
    588   const TargetRegisterInfo *TRI = getTargetRegisterInfo();
    589   for (MCRegAliasIterator AliasReg(PhysReg, TRI, true); AliasReg.isValid();
    590        ++AliasReg) {
    591     if (!reg_nodbg_empty(*AliasReg))
    592       return true;
    593   }
    594   return false;
    595 }
    596 
    597 void MachineRegisterInfo::disableCalleeSavedRegister(MCRegister Reg) {
    598 
    599   const TargetRegisterInfo *TRI = getTargetRegisterInfo();
    600   assert(Reg && (Reg < TRI->getNumRegs()) &&
    601          "Trying to disable an invalid register");
    602 
    603   if (!IsUpdatedCSRsInitialized) {
    604     const MCPhysReg *CSR = TRI->getCalleeSavedRegs(MF);
    605     for (const MCPhysReg *I = CSR; *I; ++I)
    606       UpdatedCSRs.push_back(*I);
    607 
    608     // Zero value represents the end of the register list
    609     // (no more registers should be pushed).
    610     UpdatedCSRs.push_back(0);
    611 
    612     IsUpdatedCSRsInitialized = true;
    613   }
    614 
    615   // Remove the register (and its aliases from the list).
    616   for (MCRegAliasIterator AI(Reg, TRI, true); AI.isValid(); ++AI)
    617     llvm::erase_value(UpdatedCSRs, *AI);
    618 }
    619 
    620 const MCPhysReg *MachineRegisterInfo::getCalleeSavedRegs() const {
    621   if (IsUpdatedCSRsInitialized)
    622     return UpdatedCSRs.data();
    623 
    624   return getTargetRegisterInfo()->getCalleeSavedRegs(MF);
    625 }
    626 
    627 void MachineRegisterInfo::setCalleeSavedRegs(ArrayRef<MCPhysReg> CSRs) {
    628   if (IsUpdatedCSRsInitialized)
    629     UpdatedCSRs.clear();
    630 
    631   append_range(UpdatedCSRs, CSRs);
    632 
    633   // Zero value represents the end of the register list
    634   // (no more registers should be pushed).
    635   UpdatedCSRs.push_back(0);
    636   IsUpdatedCSRsInitialized = true;
    637 }
    638 
    639 bool MachineRegisterInfo::isReservedRegUnit(unsigned Unit) const {
    640   const TargetRegisterInfo *TRI = getTargetRegisterInfo();
    641   for (MCRegUnitRootIterator Root(Unit, TRI); Root.isValid(); ++Root) {
    642     bool IsRootReserved = true;
    643     for (MCSuperRegIterator Super(*Root, TRI, /*IncludeSelf=*/true);
    644          Super.isValid(); ++Super) {
    645       MCRegister Reg = *Super;
    646       if (!isReserved(Reg)) {
    647         IsRootReserved = false;
    648         break;
    649       }
    650     }
    651     if (IsRootReserved)
    652       return true;
    653   }
    654   return false;
    655 }
    656