1 //===- NVPTXRegisterInfo.h - NVPTX Register Information Impl ----*- 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 // This file contains the NVPTX implementation of the TargetRegisterInfo class.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXREGISTERINFO_H
14 #define LLVM_LIB_TARGET_NVPTX_NVPTXREGISTERINFO_H
15 
16 #include "ManagedStringPool.h"
17 #include "llvm/CodeGen/TargetRegisterInfo.h"
18 #include <sstream>
19 
20 #define GET_REGINFO_HEADER
21 #include "NVPTXGenRegisterInfo.inc"
22 
23 namespace llvm {
24 class NVPTXRegisterInfo : public NVPTXGenRegisterInfo {
25 private:
26   // Hold Strings that can be free'd all together with NVPTXRegisterInfo
27   ManagedStringPool ManagedStrPool;
28 
29 public:
30   NVPTXRegisterInfo();
31 
32   //------------------------------------------------------
33   // Pure virtual functions from TargetRegisterInfo
34   //------------------------------------------------------
35 
36   // NVPTX callee saved registers
37   const MCPhysReg *getCalleeSavedRegs(const MachineFunction *MF) const override;
38 
39   BitVector getReservedRegs(const MachineFunction &MF) const override;
40 
41   void eliminateFrameIndex(MachineBasicBlock::iterator MI, int SPAdj,
42                            unsigned FIOperandNum,
43                            RegScavenger *RS = nullptr) const override;
44 
45   Register getFrameRegister(const MachineFunction &MF) const override;
46   Register getFrameLocalRegister(const MachineFunction &MF) const;
47 
48   ManagedStringPool *getStrPool() const {
49     return const_cast<ManagedStringPool *>(&ManagedStrPool);
50   }
51 
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