lib/Target/NVPTX/NVPTXRegisterInfo.h

   1 //===- NVPTXRegisterInfo.h - NVPTX Register Information Impl ----*- C++ -*-===//
   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 contains the NVPTX implementation of the TargetRegisterInfo class.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXREGISTERINFO_H
  15 #define LLVM_LIB_TARGET_NVPTX_NVPTXREGISTERINFO_H
  16
  17 #include "ManagedStringPool.h"
  18 #include "llvm/Target/TargetRegisterInfo.h"
  19 #include <sstream>
  20
  21 #define GET_REGINFO_HEADER
  22 #include "NVPTXGenRegisterInfo.inc"
  23
  24 namespace llvm {
  25
  26 // Forward Declarations.
  27 class TargetInstrInfo;
  28 class NVPTXSubtarget;
  29
  30 class NVPTXRegisterInfo : public NVPTXGenRegisterInfo {
  31 private:
  32   bool Is64Bit;
  33   // Hold Strings that can be free'd all together with NVPTXRegisterInfo
  34   ManagedStringPool ManagedStrPool;
  35
  36 public:
  37   NVPTXRegisterInfo(const NVPTXSubtarget &st);
  38
  39   //------------------------------------------------------
  40   // Pure virtual functions from TargetRegisterInfo
  41   //------------------------------------------------------
  42
  43   // NVPTX callee saved registers
  44   const MCPhysReg *
  45   getCalleeSavedRegs(const MachineFunction *MF = nullptr) const override;
  46
  47   BitVector getReservedRegs(const MachineFunction &MF) const override;
  48
  49   void eliminateFrameIndex(MachineBasicBlock::iterator MI, int SPAdj,
  50                            unsigned FIOperandNum,
  51                            RegScavenger *RS = nullptr) const override;
  52
  53   unsigned getFrameRegister(const MachineFunction &MF) const override;
  54
  55   ManagedStringPool *getStrPool() const {
  56     return const_cast<ManagedStringPool *>(&ManagedStrPool);
  57   }
  58
  59   const char *getName(unsigned RegNo) const {
  60     std::stringstream O;
  61     O << "reg" << RegNo;
  62     return getStrPool()->getManagedString(O.str().c_str())->c_str();
  63   }
  64
  65 };
  66
  67 std::string getNVPTXRegClassName(const TargetRegisterClass *RC);
  68 std::string getNVPTXRegClassStr(const TargetRegisterClass *RC);
  69
  70 } // end namespace llvm
  71
  72 #endif