Home | History | Annotate | Line # | Download | only in PowerPC
      1 //===-- PPCFrameLowering.h - Define frame lowering for PowerPC --*- 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 //
     10 //===----------------------------------------------------------------------===//
     11 
     12 #ifndef LLVM_LIB_TARGET_POWERPC_PPCFRAMELOWERING_H
     13 #define LLVM_LIB_TARGET_POWERPC_PPCFRAMELOWERING_H
     14 
     15 #include "llvm/ADT/STLExtras.h"
     16 #include "llvm/CodeGen/TargetFrameLowering.h"
     17 #include "llvm/Target/TargetMachine.h"
     18 
     19 namespace llvm {
     20 class PPCSubtarget;
     21 
     22 class PPCFrameLowering: public TargetFrameLowering {
     23   const PPCSubtarget &Subtarget;
     24   const unsigned ReturnSaveOffset;
     25   const unsigned TOCSaveOffset;
     26   const unsigned FramePointerSaveOffset;
     27   const unsigned LinkageSize;
     28   const unsigned BasePointerSaveOffset;
     29   const unsigned CRSaveOffset;
     30 
     31   // Map each group of one or two GPRs to corresponding VSR for spilling.
     32   // TODO: Use local table in methods to avoid this mutable member.
     33   mutable DenseMap<unsigned, std::pair<Register, Register>> VSRContainingGPRs;
     34 
     35   /**
     36    * Find register[s] that can be used in function prologue and epilogue
     37    *
     38    * Find register[s] that can be use as scratch register[s] in function
     39    * prologue and epilogue to save various registers (Link Register, Base
     40    * Pointer, etc.). Prefer R0/R12, if available. Otherwise choose whatever
     41    * register[s] are available.
     42    *
     43    * This method will return true if it is able to find enough unique scratch
     44    * registers (1 or 2 depending on the requirement). If it is unable to find
     45    * enough available registers in the block, it will return false and set
     46    * any passed output parameter that corresponds to a required unique register
     47    * to PPC::NoRegister.
     48    *
     49    * \param[in] MBB The machine basic block to find an available register for
     50    * \param[in] UseAtEnd Specify whether the scratch register will be used at
     51    *                     the end of the basic block (i.e., will the scratch
     52    *                     register kill a register defined in the basic block)
     53    * \param[in] TwoUniqueRegsRequired Specify whether this basic block will
     54    *                                  require two unique scratch registers.
     55    * \param[out] SR1 The scratch register to use
     56    * \param[out] SR2 The second scratch register. If this pointer is not null
     57    *                 the function will attempt to set it to an available
     58    *                 register regardless of whether there is a hard requirement
     59    *                 for two unique scratch registers.
     60    * \return true if the required number of registers was found.
     61    *         false if the required number of scratch register weren't available.
     62    *         If either output parameter refers to a required scratch register
     63    *         that isn't available, it will be set to an invalid value.
     64    */
     65   bool findScratchRegister(MachineBasicBlock *MBB,
     66                            bool UseAtEnd,
     67                            bool TwoUniqueRegsRequired = false,
     68                            Register *SR1 = nullptr,
     69                            Register *SR2 = nullptr) const;
     70   bool twoUniqueScratchRegsRequired(MachineBasicBlock *MBB) const;
     71 
     72   /**
     73    * Create branch instruction for PPC::TCRETURN* (tail call return)
     74    *
     75    * \param[in] MBB that is terminated by PPC::TCRETURN*
     76    */
     77   void createTailCallBranchInstr(MachineBasicBlock &MBB) const;
     78 
     79   /**
     80     * Check if the conditions are correct to allow for the stack update
     81     * to be moved past the CSR save/restore code.
     82     */
     83   bool stackUpdateCanBeMoved(MachineFunction &MF) const;
     84 
     85 public:
     86   PPCFrameLowering(const PPCSubtarget &STI);
     87 
     88   /**
     89    * Determine the frame layout and update the machine function.
     90    */
     91   unsigned determineFrameLayoutAndUpdate(MachineFunction &MF,
     92                                          bool UseEstimate = false) const;
     93 
     94   /**
     95    * Determine the frame layout but do not update the machine function.
     96    * The MachineFunction object can be const in this case as it is not
     97    * modified.
     98    */
     99   unsigned determineFrameLayout(const MachineFunction &MF,
    100                                 bool UseEstimate = false,
    101                                 unsigned *NewMaxCallFrameSize = nullptr) const;
    102 
    103   /// emitProlog/emitEpilog - These methods insert prolog and epilog code into
    104   /// the function.
    105   void emitPrologue(MachineFunction &MF, MachineBasicBlock &MBB) const override;
    106   void emitEpilogue(MachineFunction &MF, MachineBasicBlock &MBB) const override;
    107   void inlineStackProbe(MachineFunction &MF,
    108                         MachineBasicBlock &PrologMBB) const override;
    109 
    110   bool hasFP(const MachineFunction &MF) const override;
    111   bool needsFP(const MachineFunction &MF) const;
    112   void replaceFPWithRealFP(MachineFunction &MF) const;
    113 
    114   void determineCalleeSaves(MachineFunction &MF, BitVector &SavedRegs,
    115                             RegScavenger *RS = nullptr) const override;
    116   void processFunctionBeforeFrameFinalized(MachineFunction &MF,
    117                                      RegScavenger *RS = nullptr) const override;
    118   void addScavengingSpillSlot(MachineFunction &MF, RegScavenger *RS) const;
    119 
    120   bool spillCalleeSavedRegisters(MachineBasicBlock &MBB,
    121                                  MachineBasicBlock::iterator MI,
    122                                  ArrayRef<CalleeSavedInfo> CSI,
    123                                  const TargetRegisterInfo *TRI) const override;
    124   /// This function will assign callee saved gprs to volatile vector registers
    125   /// for prologue spills when applicable. It returns false if there are any
    126   /// registers which were not spilled to volatile vector registers.
    127   bool
    128   assignCalleeSavedSpillSlots(MachineFunction &MF,
    129                               const TargetRegisterInfo *TRI,
    130                               std::vector<CalleeSavedInfo> &CSI) const override;
    131 
    132   MachineBasicBlock::iterator
    133   eliminateCallFramePseudoInstr(MachineFunction &MF, MachineBasicBlock &MBB,
    134                                 MachineBasicBlock::iterator I) const override;
    135 
    136   bool
    137   restoreCalleeSavedRegisters(MachineBasicBlock &MBB,
    138                               MachineBasicBlock::iterator MI,
    139                               MutableArrayRef<CalleeSavedInfo> CSI,
    140                               const TargetRegisterInfo *TRI) const override;
    141 
    142   /// targetHandlesStackFrameRounding - Returns true if the target is
    143   /// responsible for rounding up the stack frame (probably at emitPrologue
    144   /// time).
    145   bool targetHandlesStackFrameRounding() const override { return true; }
    146 
    147   /// getReturnSaveOffset - Return the previous frame offset to save the
    148   /// return address.
    149   unsigned getReturnSaveOffset() const { return ReturnSaveOffset; }
    150 
    151   /// getTOCSaveOffset - Return the previous frame offset to save the
    152   /// TOC register -- 64-bit SVR4 ABI only.
    153   unsigned getTOCSaveOffset() const;
    154 
    155   /// getFramePointerSaveOffset - Return the previous frame offset to save the
    156   /// frame pointer.
    157   unsigned getFramePointerSaveOffset() const;
    158 
    159   /// getBasePointerSaveOffset - Return the previous frame offset to save the
    160   /// base pointer.
    161   unsigned getBasePointerSaveOffset() const;
    162 
    163   /// getLinkageSize - Return the size of the PowerPC ABI linkage area.
    164   ///
    165   unsigned getLinkageSize() const { return LinkageSize; }
    166 
    167   const SpillSlot *
    168   getCalleeSavedSpillSlots(unsigned &NumEntries) const override;
    169 
    170   bool enableShrinkWrapping(const MachineFunction &MF) const override;
    171 
    172   /// Methods used by shrink wrapping to determine if MBB can be used for the
    173   /// function prologue/epilogue.
    174   bool canUseAsPrologue(const MachineBasicBlock &MBB) const override;
    175   bool canUseAsEpilogue(const MachineBasicBlock &MBB) const override;
    176 };
    177 } // End llvm namespace
    178 
    179 #endif
    180