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/Target/TargetMachine.h"
19 #include <utility>
20 
21 namespace llvm {
22 
23 /// NVPTXTargetMachine
24 ///
25 class NVPTXTargetMachine : public LLVMTargetMachine {
26   bool is64bit;
27   // Use 32-bit pointers for accessing const/local/short AS.
28   bool UseShortPointers;
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 
36 public:
37   NVPTXTargetMachine(const Target &T, const Triple &TT, StringRef CPU,
38                      StringRef FS, const TargetOptions &Options,
39                      Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
40                      CodeGenOpt::Level OP, 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   bool useShortPointers() const { return UseShortPointers; }
49   NVPTX::DrvInterface getDrvInterface() const { return drvInterface; }
50   ManagedStringPool *getManagedStrPool() const {
51     return const_cast<ManagedStringPool *>(&ManagedStrPool);
52   }
53 
54   TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
55 
56   // Emission of machine code through MCJIT is not supported.
57   bool addPassesToEmitMC(PassManagerBase &, MCContext *&, raw_pwrite_stream &,
58                          bool = true) override {
59     return true;
60   }
61   TargetLoweringObjectFile *getObjFileLowering() const override {
62     return TLOF.get();
63   }
64 
65   void adjustPassManager(PassManagerBuilder &) override;
66   void registerPassBuilderCallbacks(PassBuilder &PB) override;
67 
68   TargetTransformInfo getTargetTransformInfo(const Function &F) const override;
69 
70   bool isMachineVerifierClean() const override {
71     return false;
72   }
73 
74   std::pair<const Value *, unsigned>
75   getPredicatedAddrSpace(const Value *V) const override;
76 }; // NVPTXTargetMachine.
77 
78 class NVPTXTargetMachine32 : public NVPTXTargetMachine {
79   virtual void anchor();
80 public:
81   NVPTXTargetMachine32(const Target &T, const Triple &TT, StringRef CPU,
82                        StringRef FS, const TargetOptions &Options,
83                        Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
84                        CodeGenOpt::Level OL, bool JIT);
85 };
86 
87 class NVPTXTargetMachine64 : public NVPTXTargetMachine {
88   virtual void anchor();
89 public:
90   NVPTXTargetMachine64(const Target &T, const Triple &TT, StringRef CPU,
91                        StringRef FS, const TargetOptions &Options,
92                        Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM,
93                        CodeGenOpt::Level OL, bool JIT);
94 };
95 
96 } // end namespace llvm
97 
98 #endif
99