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/CodeGen/SelectionDAGTargetInfo.h" 20 #include "llvm/Target/TargetFrameLowering.h" 21 #include "llvm/Target/TargetMachine.h" 22 23 namespace llvm { 24 25 /// NVPTXTargetMachine 26 /// 27 class NVPTXTargetMachine : public LLVMTargetMachine { 28 bool is64bit; 29 std::unique_ptr<TargetLoweringObjectFile> TLOF; 30 NVPTX::DrvInterface drvInterface; 31 NVPTXSubtarget Subtarget; 32 33 // Hold Strings that can be free'd all together with NVPTXTargetMachine 34 ManagedStringPool ManagedStrPool; 35 36 public: 37 NVPTXTargetMachine(const Target &T, const Triple &TT, StringRef CPU, 38 StringRef FS, const TargetOptions &Options, 39 Optional<Reloc::Model> RM, CodeModel::Model CM, 40 CodeGenOpt::Level OP, bool is64bit); 41 42 ~NVPTXTargetMachine() override; getSubtargetImpl(const Function &)43 const NVPTXSubtarget *getSubtargetImpl(const Function &) const override { 44 return &Subtarget; 45 } getSubtargetImpl()46 const NVPTXSubtarget *getSubtargetImpl() const { return &Subtarget; } is64Bit()47 bool is64Bit() const { return is64bit; } getDrvInterface()48 NVPTX::DrvInterface getDrvInterface() const { return drvInterface; } getManagedStrPool()49 ManagedStringPool *getManagedStrPool() const { 50 return const_cast<ManagedStringPool *>(&ManagedStrPool); 51 } 52 53 TargetPassConfig *createPassConfig(PassManagerBase &PM) override; 54 55 // Emission of machine code through MCJIT is not supported. 56 bool addPassesToEmitMC(PassManagerBase &, MCContext *&, raw_pwrite_stream &, 57 bool = true) override { 58 return true; 59 } getObjFileLowering()60 TargetLoweringObjectFile *getObjFileLowering() const override { 61 return TLOF.get(); 62 } 63 64 void addEarlyAsPossiblePasses(PassManagerBase &PM) override; 65 TargetIRAnalysis getTargetIRAnalysis() override; 66 67 }; // NVPTXTargetMachine. 68 69 class NVPTXTargetMachine32 : public NVPTXTargetMachine { 70 virtual void anchor(); 71 public: 72 NVPTXTargetMachine32(const Target &T, const Triple &TT, StringRef CPU, 73 StringRef FS, const TargetOptions &Options, 74 Optional<Reloc::Model> RM, CodeModel::Model CM, 75 CodeGenOpt::Level OL); 76 }; 77 78 class NVPTXTargetMachine64 : public NVPTXTargetMachine { 79 virtual void anchor(); 80 public: 81 NVPTXTargetMachine64(const Target &T, const Triple &TT, StringRef CPU, 82 StringRef FS, const TargetOptions &Options, 83 Optional<Reloc::Model> RM, CodeModel::Model CM, 84 CodeGenOpt::Level OL); 85 }; 86 87 } // end namespace llvm 88 89 #endif 90