NVPTXTargetMachine.h revision dce4a407a24b04eebc6a376f8e62b41aaa7b071f 
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 NVPTX_TARGETMACHINE_H
15#define NVPTX_TARGETMACHINE_H
16
17#include "ManagedStringPool.h"
18#include "NVPTXFrameLowering.h"
19#include "NVPTXISelLowering.h"
20#include "NVPTXInstrInfo.h"
21#include "NVPTXRegisterInfo.h"
22#include "NVPTXSubtarget.h"
23#include "llvm/IR/DataLayout.h"
24#include "llvm/Target/TargetFrameLowering.h"
25#include "llvm/Target/TargetMachine.h"
26#include "llvm/Target/TargetSelectionDAGInfo.h"
27
28namespace llvm {
29
30/// NVPTXTargetMachine
31///
32class NVPTXTargetMachine : public LLVMTargetMachine {
33  NVPTXSubtarget Subtarget;
34  const DataLayout DL; // Calculates type size & alignment
35  NVPTXInstrInfo InstrInfo;
36  NVPTXTargetLowering TLInfo;
37  TargetSelectionDAGInfo TSInfo;
38
39  // NVPTX does not have any call stack frame, but need a NVPTX specific
40  // FrameLowering class because TargetFrameLowering is abstract.
41  NVPTXFrameLowering FrameLowering;
42
43  // Hold Strings that can be free'd all together with NVPTXTargetMachine
44  ManagedStringPool ManagedStrPool;
45
46  //bool addCommonCodeGenPasses(PassManagerBase &, CodeGenOpt::Level,
47  //                            bool DisableVerify, MCContext *&OutCtx);
48
49public:
50  NVPTXTargetMachine(const Target &T, StringRef TT, StringRef CPU, StringRef FS,
51                     const TargetOptions &Options, Reloc::Model RM,
52                     CodeModel::Model CM, CodeGenOpt::Level OP, bool is64bit);
53
54  const TargetFrameLowering *getFrameLowering() const override {
55    return &FrameLowering;
56  }
57  const NVPTXInstrInfo *getInstrInfo() const override { return &InstrInfo; }
58  const DataLayout *getDataLayout() const override { return &DL; }
59  const NVPTXSubtarget *getSubtargetImpl() const override { return &Subtarget; }
60
61  const NVPTXRegisterInfo *getRegisterInfo() const override {
62    return &(InstrInfo.getRegisterInfo());
63  }
64
65  NVPTXTargetLowering *getTargetLowering() const override {
66    return const_cast<NVPTXTargetLowering *>(&TLInfo);
67  }
68
69  const TargetSelectionDAGInfo *getSelectionDAGInfo() const override {
70    return &TSInfo;
71  }
72
73  //virtual bool addInstSelector(PassManagerBase &PM,
74  //                             CodeGenOpt::Level OptLevel);
75
76  //virtual bool addPreRegAlloc(PassManagerBase &, CodeGenOpt::Level);
77
78  ManagedStringPool *getManagedStrPool() const {
79    return const_cast<ManagedStringPool *>(&ManagedStrPool);
80  }
81
82  TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
83
84  // Emission of machine code through JITCodeEmitter is not supported.
85  bool addPassesToEmitMachineCode(PassManagerBase &, JITCodeEmitter &,
86                                  bool = true) override {
87    return true;
88  }
89
90  // Emission of machine code through MCJIT is not supported.
91  bool addPassesToEmitMC(PassManagerBase &, MCContext *&, raw_ostream &,
92                         bool = true) override {
93    return true;
94  }
95
96}; // NVPTXTargetMachine.
97
98class NVPTXTargetMachine32 : public NVPTXTargetMachine {
99  virtual void anchor();
100public:
101  NVPTXTargetMachine32(const Target &T, StringRef TT, StringRef CPU,
102                       StringRef FS, const TargetOptions &Options,
103                       Reloc::Model RM, CodeModel::Model CM,
104                       CodeGenOpt::Level OL);
105};
106
107class NVPTXTargetMachine64 : public NVPTXTargetMachine {
108  virtual void anchor();
109public:
110  NVPTXTargetMachine64(const Target &T, StringRef TT, StringRef CPU,
111                       StringRef FS, const TargetOptions &Options,
112                       Reloc::Model RM, CodeModel::Model CM,
113                       CodeGenOpt::Level OL);
114};
115
116} // end namespace llvm
117
118#endif
119