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 class NVPTXRegisterInfo : public NVPTXGenRegisterInfo { 26 private: 27 // Hold Strings that can be free'd all together with NVPTXRegisterInfo 28 ManagedStringPool ManagedStrPool; 29 30 public: 31 NVPTXRegisterInfo(); 32 33 //------------------------------------------------------ 34 // Pure virtual functions from TargetRegisterInfo 35 //------------------------------------------------------ 36 37 // NVPTX callee saved registers 38 const MCPhysReg *getCalleeSavedRegs(const MachineFunction *MF) const override; 39 40 BitVector getReservedRegs(const MachineFunction &MF) const override; 41 42 void eliminateFrameIndex(MachineBasicBlock::iterator MI, int SPAdj, 43 unsigned FIOperandNum, 44 RegScavenger *RS = nullptr) const override; 45 46 unsigned getFrameRegister(const MachineFunction &MF) const override; 47 getStrPool()48 ManagedStringPool *getStrPool() const { 49 return const_cast<ManagedStringPool *>(&ManagedStrPool); 50 } 51 getName(unsigned RegNo)52 const char *getName(unsigned RegNo) const { 53 std::stringstream O; 54 O << "reg" << RegNo; 55 return getStrPool()->getManagedString(O.str().c_str())->c_str(); 56 } 57 58 }; 59 60 std::string getNVPTXRegClassName(const TargetRegisterClass *RC); 61 std::string getNVPTXRegClassStr(const TargetRegisterClass *RC); 62 63 } // end namespace llvm 64 65 #endif 66