LCOV - code coverage report
Current view: top level - lib/CodeGen - ExpandPostRAPseudos.cpp (source / functions) Hit Total Coverage
Test: llvm-toolchain.info Lines: 62 66 93.9 %
Date: 2018-02-22 16:16:46 Functions: 8 9 88.9 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : //===-- ExpandPostRAPseudos.cpp - Pseudo instruction expansion pass -------===//
       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 file defines a pass that expands COPY and SUBREG_TO_REG pseudo
      11             : // instructions after register allocation.
      12             : //
      13             : //===----------------------------------------------------------------------===//
      14             : 
      15             : #include "llvm/CodeGen/MachineFunctionPass.h"
      16             : #include "llvm/CodeGen/MachineInstr.h"
      17             : #include "llvm/CodeGen/MachineInstrBuilder.h"
      18             : #include "llvm/CodeGen/MachineRegisterInfo.h"
      19             : #include "llvm/CodeGen/Passes.h"
      20             : #include "llvm/CodeGen/TargetInstrInfo.h"
      21             : #include "llvm/CodeGen/TargetRegisterInfo.h"
      22             : #include "llvm/CodeGen/TargetSubtargetInfo.h"
      23             : #include "llvm/Support/Debug.h"
      24             : #include "llvm/Support/raw_ostream.h"
      25             : 
      26             : using namespace llvm;
      27             : 
      28             : #define DEBUG_TYPE "postrapseudos"
      29             : 
      30             : namespace {
      31       18702 : struct ExpandPostRA : public MachineFunctionPass {
      32             : private:
      33             :   const TargetRegisterInfo *TRI;
      34             :   const TargetInstrInfo *TII;
      35             : 
      36             : public:
      37             :   static char ID; // Pass identification, replacement for typeid
      38       18808 :   ExpandPostRA() : MachineFunctionPass(ID) {}
      39             : 
      40       18683 :   void getAnalysisUsage(AnalysisUsage &AU) const override {
      41       18683 :     AU.setPreservesCFG();
      42       18683 :     AU.addPreservedID(MachineLoopInfoID);
      43       18683 :     AU.addPreservedID(MachineDominatorsID);
      44       18683 :     MachineFunctionPass::getAnalysisUsage(AU);
      45       18683 :   }
      46             : 
      47             :   /// runOnMachineFunction - pass entry point
      48             :   bool runOnMachineFunction(MachineFunction&) override;
      49             : 
      50             : private:
      51             :   bool LowerSubregToReg(MachineInstr *MI);
      52             :   bool LowerCopy(MachineInstr *MI);
      53             : 
      54             :   void TransferImplicitOperands(MachineInstr *MI);
      55             : };
      56             : } // end anonymous namespace
      57             : 
      58             : char ExpandPostRA::ID = 0;
      59             : char &llvm::ExpandPostRAPseudosID = ExpandPostRA::ID;
      60             : 
      61      126396 : INITIALIZE_PASS(ExpandPostRA, DEBUG_TYPE,
      62             :                 "Post-RA pseudo instruction expansion pass", false, false)
      63             : 
      64             : /// TransferImplicitOperands - MI is a pseudo-instruction, and the lowered
      65             : /// replacement instructions immediately precede it.  Copy any implicit
      66             : /// operands from MI to the replacement instruction.
      67       32162 : void ExpandPostRA::TransferImplicitOperands(MachineInstr *MI) {
      68             :   MachineBasicBlock::iterator CopyMI = MI;
      69             :   --CopyMI;
      70             : 
      71       97784 :   for (const MachineOperand &MO : MI->implicit_operands())
      72       32811 :     if (MO.isReg())
      73       32811 :       CopyMI->addOperand(MO);
      74       32162 : }
      75             : 
      76         245 : bool ExpandPostRA::LowerSubregToReg(MachineInstr *MI) {
      77         245 :   MachineBasicBlock *MBB = MI->getParent();
      78             :   assert((MI->getOperand(0).isReg() && MI->getOperand(0).isDef()) &&
      79             :          MI->getOperand(1).isImm() &&
      80             :          (MI->getOperand(2).isReg() && MI->getOperand(2).isUse()) &&
      81             :           MI->getOperand(3).isImm() && "Invalid subreg_to_reg");
      82             : 
      83         245 :   unsigned DstReg  = MI->getOperand(0).getReg();
      84         245 :   unsigned InsReg  = MI->getOperand(2).getReg();
      85             :   assert(!MI->getOperand(2).getSubReg() && "SubIdx on physreg?");
      86         245 :   unsigned SubIdx  = MI->getOperand(3).getImm();
      87             : 
      88             :   assert(SubIdx != 0 && "Invalid index for insert_subreg");
      89         245 :   unsigned DstSubReg = TRI->getSubReg(DstReg, SubIdx);
      90             : 
      91             :   assert(TargetRegisterInfo::isPhysicalRegister(DstReg) &&
      92             :          "Insert destination must be in a physical register");
      93             :   assert(TargetRegisterInfo::isPhysicalRegister(InsReg) &&
      94             :          "Inserted value must be in a physical register");
      95             : 
      96             :   DEBUG(dbgs() << "subreg: CONVERTING: " << *MI);
      97             : 
      98         245 :   if (MI->allDefsAreDead()) {
      99           0 :     MI->setDesc(TII->get(TargetOpcode::KILL));
     100             :     DEBUG(dbgs() << "subreg: replaced by: " << *MI);
     101             :     return true;
     102             :   }
     103             : 
     104         245 :   if (DstSubReg == InsReg) {
     105             :     // No need to insert an identity copy instruction.
     106             :     // Watch out for case like this:
     107             :     // %rax = SUBREG_TO_REG 0, killed %eax, 3
     108             :     // We must leave %rax live.
     109          18 :     if (DstReg != InsReg) {
     110          18 :       MI->setDesc(TII->get(TargetOpcode::KILL));
     111          18 :       MI->RemoveOperand(3);     // SubIdx
     112          18 :       MI->RemoveOperand(1);     // Imm
     113             :       DEBUG(dbgs() << "subreg: replace by: " << *MI);
     114             :       return true;
     115             :     }
     116             :     DEBUG(dbgs() << "subreg: eliminated!");
     117             :   } else {
     118         681 :     TII->copyPhysReg(*MBB, MI, MI->getDebugLoc(), DstSubReg, InsReg,
     119         227 :                      MI->getOperand(2).isKill());
     120             : 
     121             :     // Implicitly define DstReg for subsequent uses.
     122             :     MachineBasicBlock::iterator CopyMI = MI;
     123             :     --CopyMI;
     124         227 :     CopyMI->addRegisterDefined(DstReg);
     125             :     DEBUG(dbgs() << "subreg: " << *CopyMI);
     126             :   }
     127             : 
     128             :   DEBUG(dbgs() << '\n');
     129             :   MBB->erase(MI);
     130             :   return true;
     131             : }
     132             : 
     133      170936 : bool ExpandPostRA::LowerCopy(MachineInstr *MI) {
     134             : 
     135      170936 :   if (MI->allDefsAreDead()) {
     136             :     DEBUG(dbgs() << "dead copy: " << *MI);
     137         317 :     MI->setDesc(TII->get(TargetOpcode::KILL));
     138             :     DEBUG(dbgs() << "replaced by: " << *MI);
     139             :     return true;
     140             :   }
     141             : 
     142      170619 :   MachineOperand &DstMO = MI->getOperand(0);
     143             :   MachineOperand &SrcMO = MI->getOperand(1);
     144             : 
     145      170619 :   bool IdentityCopy = (SrcMO.getReg() == DstMO.getReg());
     146      341233 :   if (IdentityCopy || SrcMO.isUndef()) {
     147             :     DEBUG(dbgs() << (IdentityCopy ? "identity copy: " : "undef copy:    ") << *MI);
     148             :     // No need to insert an identity copy instruction, but replace with a KILL
     149             :     // if liveness is changed.
     150          11 :     if (SrcMO.isUndef() || MI->getNumOperands() > 2) {
     151             :       // We must make sure the super-register gets killed. Replace the
     152             :       // instruction with KILL.
     153           6 :       MI->setDesc(TII->get(TargetOpcode::KILL));
     154             :       DEBUG(dbgs() << "replaced by:   " << *MI);
     155             :       return true;
     156             :     }
     157             :     // Vanilla identity copy.
     158           0 :     MI->eraseFromParent();
     159             :     return true;
     160             :   }
     161             : 
     162             :   DEBUG(dbgs() << "real copy:   " << *MI);
     163      511839 :   TII->copyPhysReg(*MI->getParent(), MI, MI->getDebugLoc(),
     164      170613 :                    DstMO.getReg(), SrcMO.getReg(), SrcMO.isKill());
     165             : 
     166      170613 :   if (MI->getNumOperands() > 2)
     167       32162 :     TransferImplicitOperands(MI);
     168             :   DEBUG({
     169             :     MachineBasicBlock::iterator dMI = MI;
     170             :     dbgs() << "replaced by: " << *(--dMI);
     171             :   });
     172      170613 :   MI->eraseFromParent();
     173             :   return true;
     174             : }
     175             : 
     176             : /// runOnMachineFunction - Reduce subregister inserts and extracts to register
     177             : /// copies.
     178             : ///
     179      167275 : bool ExpandPostRA::runOnMachineFunction(MachineFunction &MF) {
     180             :   DEBUG(dbgs() << "Machine Function\n"
     181             :                << "********** EXPANDING POST-RA PSEUDO INSTRS **********\n"
     182             :                << "********** Function: " << MF.getName() << '\n');
     183      167275 :   TRI = MF.getSubtarget().getRegisterInfo();
     184      167275 :   TII = MF.getSubtarget().getInstrInfo();
     185             : 
     186             :   bool MadeChange = false;
     187             : 
     188             :   for (MachineFunction::iterator mbbi = MF.begin(), mbbe = MF.end();
     189      485511 :        mbbi != mbbe; ++mbbi) {
     190      318236 :     for (MachineBasicBlock::iterator mi = mbbi->begin(), me = mbbi->end();
     191     3470670 :          mi != me;) {
     192             :       MachineInstr &MI = *mi;
     193             :       // Advance iterator here because MI may be erased.
     194             :       ++mi;
     195             : 
     196             :       // Only expand pseudos.
     197     3152434 :       if (!MI.isPseudo())
     198     2275378 :         continue;
     199             : 
     200             :       // Give targets a chance to expand even standard pseudos.
     201      877056 :       if (TII->expandPostRAPseudo(MI)) {
     202             :         MadeChange = true;
     203      110478 :         continue;
     204             :       }
     205             : 
     206             :       // Expand standard pseudos.
     207     1553902 :       switch (MI.getOpcode()) {
     208         245 :       case TargetOpcode::SUBREG_TO_REG:
     209         245 :         MadeChange |= LowerSubregToReg(&MI);
     210         245 :         break;
     211      170936 :       case TargetOpcode::COPY:
     212      170936 :         MadeChange |= LowerCopy(&MI);
     213      170936 :         break;
     214       20746 :       case TargetOpcode::DBG_VALUE:
     215       20746 :         continue;
     216           0 :       case TargetOpcode::INSERT_SUBREG:
     217             :       case TargetOpcode::EXTRACT_SUBREG:
     218           0 :         llvm_unreachable("Sub-register pseudos should have been eliminated.");
     219             :       }
     220             :     }
     221             :   }
     222             : 
     223      167275 :   return MadeChange;
     224             : }

Generated by: LCOV version 1.13