llvm/lib/Target/NVPTX/NVPTXRegisterInfo.h

   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 "llvm/CodeGen/TargetRegisterInfo.h"
  17 #include "llvm/Support/StringSaver.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   BumpPtrAllocator StrAlloc;
  28   UniqueStringSaver StrPool;
  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   bool eliminateFrameIndex(MachineBasicBlock::iterator MI, int SPAdj,
  43                            unsigned FIOperandNum,
  44                            RegScavenger *RS = nullptr) const override;
  45
  46   Register getFrameRegister(const MachineFunction &MF) const override;
  47   Register getFrameLocalRegister(const MachineFunction &MF) const;
  48
  49   UniqueStringSaver &getStrPool() const {
  50     return const_cast<UniqueStringSaver &>(StrPool);
  51   }
  52
  53   const char *getName(unsigned RegNo) const {
  54     std::stringstream O;
  55     O << "reg" << RegNo;
  56     return getStrPool().save(O.str()).data();
  57   }
  58
  59 };
  60
  61 std::string getNVPTXRegClassName(const TargetRegisterClass *RC);
  62 std::string getNVPTXRegClassStr(const TargetRegisterClass *RC);
  63
  64 } // end namespace llvm
  65
  66 #endif