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 getStrPool()55 ManagedStringPool *getStrPool() const { 56 return const_cast<ManagedStringPool *>(&ManagedStrPool); 57 } 58 getName(unsigned RegNo)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 73