1 //===-- NVPTXTargetMachine.h - Define TargetMachine for NVPTX ---*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file declares the NVPTX specific subclass of TargetMachine.
12 //===----------------------------------------------------------------------===//
14 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXTARGETMACHINE_H
15 #define LLVM_LIB_TARGET_NVPTX_NVPTXTARGETMACHINE_H
17 #include "NVPTXSubtarget.h"
18 #include "ManagedStringPool.h"
19 #include "llvm/Target/TargetFrameLowering.h"
20 #include "llvm/Target/TargetMachine.h"
21 #include "llvm/Target/TargetSelectionDAGInfo.h"
25 /// NVPTXTargetMachine
27 class NVPTXTargetMachine : public LLVMTargetMachine {
28 std::unique_ptr<TargetLoweringObjectFile> TLOF;
29 NVPTXSubtarget Subtarget;
31 // Hold Strings that can be free'd all together with NVPTXTargetMachine
32 ManagedStringPool ManagedStrPool;
35 NVPTXTargetMachine(const Target &T, StringRef TT, StringRef CPU, StringRef FS,
36 const TargetOptions &Options, Reloc::Model RM,
37 CodeModel::Model CM, CodeGenOpt::Level OP, bool is64bit);
39 ~NVPTXTargetMachine() override;
41 const NVPTXSubtarget *getSubtargetImpl() const override { return &Subtarget; }
43 ManagedStringPool *getManagedStrPool() const {
44 return const_cast<ManagedStringPool *>(&ManagedStrPool);
47 TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
49 // Emission of machine code through MCJIT is not supported.
50 bool addPassesToEmitMC(PassManagerBase &, MCContext *&, raw_ostream &,
51 bool = true) override {
54 TargetLoweringObjectFile *getObjFileLowering() const override {
58 /// \brief Register NVPTX analysis passes with a pass manager.
59 void addAnalysisPasses(PassManagerBase &PM) override;
61 }; // NVPTXTargetMachine.
63 class NVPTXTargetMachine32 : public NVPTXTargetMachine {
64 virtual void anchor();
66 NVPTXTargetMachine32(const Target &T, StringRef TT, StringRef CPU,
67 StringRef FS, const TargetOptions &Options,
68 Reloc::Model RM, CodeModel::Model CM,
69 CodeGenOpt::Level OL);
72 class NVPTXTargetMachine64 : public NVPTXTargetMachine {
73 virtual void anchor();
75 NVPTXTargetMachine64(const Target &T, StringRef TT, StringRef CPU,
76 StringRef FS, const TargetOptions &Options,
77 Reloc::Model RM, CodeModel::Model CM,
78 CodeGenOpt::Level OL);
81 } // end namespace llvm