1//===-- NVPTXTargetMachine.h - Define TargetMachine for NVPTX ---*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file declares the NVPTX specific subclass of TargetMachine.
10//
11//===----------------------------------------------------------------------===//
12
13#ifndef LLVM_LIB_TARGET_NVPTX_NVPTXTARGETMACHINE_H
14#define LLVM_LIB_TARGET_NVPTX_NVPTXTARGETMACHINE_H
15
16#include "ManagedStringPool.h"
17#include "NVPTXSubtarget.h"
18#include "llvm/CodeGen/SelectionDAGTargetInfo.h"
19#include "llvm/CodeGen/TargetFrameLowering.h"
20#include "llvm/Target/TargetMachine.h"
21
22namespace llvm {
23
24/// NVPTXTargetMachine
25///
26class NVPTXTargetMachine : public LLVMTargetMachine {
27  bool is64bit;
28  // Use 32-bit pointers for accessing const/local/short AS.
29  bool UseShortPointers;
30  std::unique_ptr<TargetLoweringObjectFile> TLOF;
31  NVPTX::DrvInterface drvInterface;
32  NVPTXSubtarget Subtarget;
33
34  // Hold Strings that can be free'd all together with NVPTXTargetMachine
35  ManagedStringPool ManagedStrPool;
36
37public:
38  NVPTXTargetMachine(const Target &T, const Triple &TT, StringRef CPU,
39                     StringRef FS, const TargetOptions &Options,
40                     Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
41                     CodeGenOpt::Level OP, bool is64bit);
42
43  ~NVPTXTargetMachine() override;
44  const NVPTXSubtarget *getSubtargetImpl(const Function &) const override {
45    return &Subtarget;
46  }
47  const NVPTXSubtarget *getSubtargetImpl() const { return &Subtarget; }
48  bool is64Bit() const { return is64bit; }
49  bool useShortPointers() const { return UseShortPointers; }
50  NVPTX::DrvInterface getDrvInterface() const { return drvInterface; }
51  ManagedStringPool *getManagedStrPool() const {
52    return const_cast<ManagedStringPool *>(&ManagedStrPool);
53  }
54
55  TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
56
57  // Emission of machine code through MCJIT is not supported.
58  bool addPassesToEmitMC(PassManagerBase &, MCContext *&, raw_pwrite_stream &,
59                         bool = true) override {
60    return true;
61  }
62  TargetLoweringObjectFile *getObjFileLowering() const override {
63    return TLOF.get();
64  }
65
66  void adjustPassManager(PassManagerBuilder &) override;
67
68  TargetTransformInfo getTargetTransformInfo(const Function &F) override;
69
70  bool isMachineVerifierClean() const override {
71    return false;
72  }
73}; // NVPTXTargetMachine.
74
75class NVPTXTargetMachine32 : public NVPTXTargetMachine {
76  virtual void anchor();
77public:
78  NVPTXTargetMachine32(const Target &T, const Triple &TT, StringRef CPU,
79                       StringRef FS, const TargetOptions &Options,
80                       Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
81                       CodeGenOpt::Level OL, bool JIT);
82};
83
84class NVPTXTargetMachine64 : public NVPTXTargetMachine {
85  virtual void anchor();
86public:
87  NVPTXTargetMachine64(const Target &T, const Triple &TT, StringRef CPU,
88                       StringRef FS, const TargetOptions &Options,
89                       Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
90                       CodeGenOpt::Level OL, bool JIT);
91};
92
93} // end namespace llvm
94
95#endif
96