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 20 namespace llvm { 21 22 /// NVPTXTargetMachine 23 /// 24 class NVPTXTargetMachine : public LLVMTargetMachine { 25 bool is64bit; 26 // Use 32-bit pointers for accessing const/local/short AS. 27 bool UseShortPointers; 28 std::unique_ptr<TargetLoweringObjectFile> TLOF; 29 NVPTX::DrvInterface drvInterface; 30 NVPTXSubtarget Subtarget; 31 32 // Hold Strings that can be free'd all together with NVPTXTargetMachine 33 ManagedStringPool ManagedStrPool; 34 35 public: 36 NVPTXTargetMachine(const Target &T, const Triple &TT, StringRef CPU, 37 StringRef FS, const TargetOptions &Options, 38 Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM, 39 CodeGenOpt::Level OP, bool is64bit); 40 41 ~NVPTXTargetMachine() override; 42 const NVPTXSubtarget *getSubtargetImpl(const Function &) const override { 43 return &Subtarget; 44 } 45 const NVPTXSubtarget *getSubtargetImpl() const { return &Subtarget; } 46 bool is64Bit() const { return is64bit; } 47 bool useShortPointers() const { return UseShortPointers; } 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 void adjustPassManager(PassManagerBuilder &) override; 65 void registerPassBuilderCallbacks(PassBuilder &PB) override; 66 67 TargetTransformInfo getTargetTransformInfo(const Function &F) override; 68 69 bool isMachineVerifierClean() const override { 70 return false; 71 } 72 }; // NVPTXTargetMachine. 73 74 class NVPTXTargetMachine32 : public NVPTXTargetMachine { 75 virtual void anchor(); 76 public: 77 NVPTXTargetMachine32(const Target &T, const Triple &TT, StringRef CPU, 78 StringRef FS, const TargetOptions &Options, 79 Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM, 80 CodeGenOpt::Level OL, bool JIT); 81 }; 82 83 class NVPTXTargetMachine64 : public NVPTXTargetMachine { 84 virtual void anchor(); 85 public: 86 NVPTXTargetMachine64(const Target &T, const Triple &TT, StringRef CPU, 87 StringRef FS, const TargetOptions &Options, 88 Optional<Reloc::Model> RM, Optional<CodeModel::Model> CM, 89 CodeGenOpt::Level OL, bool JIT); 90 }; 91 92 } // end namespace llvm 93 94 #endif 95