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 23namespace llvm { 24 25/// NVPTXTargetMachine 26/// 27class 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 36public: 37 NVPTXTargetMachine(const Target &T, const Triple &TT, StringRef CPU, 38 StringRef FS, const TargetOptions &Options, 39 Reloc::Model RM, CodeModel::Model CM, CodeGenOpt::Level OP, 40 bool is64bit); 41 42 ~NVPTXTargetMachine() override; 43 const NVPTXSubtarget *getSubtargetImpl(const Function &) const override { 44 return &Subtarget; 45 } 46 const NVPTXSubtarget *getSubtargetImpl() const { return &Subtarget; } 47 bool is64Bit() const { return is64bit; } 48 NVPTX::DrvInterface getDrvInterface() const { return drvInterface; } 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 } 60 TargetLoweringObjectFile *getObjFileLowering() const override { 61 return TLOF.get(); 62 } 63 64 TargetIRAnalysis getTargetIRAnalysis() override; 65 66}; // NVPTXTargetMachine. 67 68class NVPTXTargetMachine32 : public NVPTXTargetMachine { 69 virtual void anchor(); 70public: 71 NVPTXTargetMachine32(const Target &T, const Triple &TT, StringRef CPU, 72 StringRef FS, const TargetOptions &Options, 73 Reloc::Model RM, CodeModel::Model CM, 74 CodeGenOpt::Level OL); 75}; 76 77class NVPTXTargetMachine64 : public NVPTXTargetMachine { 78 virtual void anchor(); 79public: 80 NVPTXTargetMachine64(const Target &T, const Triple &TT, StringRef CPU, 81 StringRef FS, const TargetOptions &Options, 82 Reloc::Model RM, CodeModel::Model CM, 83 CodeGenOpt::Level OL); 84}; 85 86} // end namespace llvm 87 88#endif 89