lib/Target/NVPTX/NVPTXTargetMachine.h

   1 //===-- NVPTXTargetMachine.h - Define TargetMachine for NVPTX ---*- 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 declares the NVPTX specific subclass of TargetMachine.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXTARGETMACHINE_H
  15 #define LLVM_LIB_TARGET_NVPTX_NVPTXTARGETMACHINE_H
  16
  17 #include "ManagedStringPool.h"
  18 #include "NVPTXSubtarget.h"
  19 #include "llvm/Target/TargetFrameLowering.h"
  20 #include "llvm/Target/TargetMachine.h"
  21 #include "llvm/Target/TargetSelectionDAGInfo.h"
  22
  23 namespace llvm {
  24
  25 /// NVPTXTargetMachine
  26 ///
  27 class NVPTXTargetMachine : public LLVMTargetMachine {
  28   std::unique_ptr<TargetLoweringObjectFile> TLOF;
  29   const DataLayout DL; // Calculates type size & alignment
  30   NVPTXSubtarget Subtarget;
  31
  32   // Hold Strings that can be free'd all together with NVPTXTargetMachine
  33   ManagedStringPool ManagedStrPool;
  34
  35 public:
  36   NVPTXTargetMachine(const Target &T, StringRef TT, StringRef CPU, StringRef FS,
  37                      const TargetOptions &Options, Reloc::Model RM,
  38                      CodeModel::Model CM, CodeGenOpt::Level OP, bool is64bit);
  39
  40   ~NVPTXTargetMachine() override;
  41   const DataLayout *getDataLayout() const override { return &DL; }
  42   const NVPTXSubtarget *getSubtargetImpl() const override { return &Subtarget; }
  43
  44   ManagedStringPool *getManagedStrPool() const {
  45     return const_cast<ManagedStringPool *>(&ManagedStrPool);
  46   }
  47
  48   TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
  49
  50   // Emission of machine code through MCJIT is not supported.
  51   bool addPassesToEmitMC(PassManagerBase &, MCContext *&, raw_ostream &,
  52                          bool = true) override {
  53     return true;
  54   }
  55   TargetLoweringObjectFile *getObjFileLowering() const override {
  56     return TLOF.get();
  57   }
  58
  59   /// \brief Register NVPTX analysis passes with a pass manager.
  60   void addAnalysisPasses(PassManagerBase &PM) override;
  61
  62 }; // NVPTXTargetMachine.
  63
  64 class NVPTXTargetMachine32 : public NVPTXTargetMachine {
  65   virtual void anchor();
  66 public:
  67   NVPTXTargetMachine32(const Target &T, StringRef TT, StringRef CPU,
  68                        StringRef FS, const TargetOptions &Options,
  69                        Reloc::Model RM, CodeModel::Model CM,
  70                        CodeGenOpt::Level OL);
  71 };
  72
  73 class NVPTXTargetMachine64 : public NVPTXTargetMachine {
  74   virtual void anchor();
  75 public:
  76   NVPTXTargetMachine64(const Target &T, StringRef TT, StringRef CPU,
  77                        StringRef FS, const TargetOptions &Options,
  78                        Reloc::Model RM, CodeModel::Model CM,
  79                        CodeGenOpt::Level OL);
  80 };
  81
  82 } // end namespace llvm
  83
  84 #endif