NVPTXTargetMachine.h revision 302408 
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