1 //===-- AMDGPUTargetMachine.h - AMDGPU TargetMachine Interface --*- 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 /// \file 10 /// The AMDGPU TargetMachine interface definition for hw codegen targets. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETMACHINE_H 15 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETMACHINE_H 16 17 #include "GCNSubtarget.h" 18 #include "llvm/CodeGen/TargetPassConfig.h" 19 #include "llvm/Target/TargetMachine.h" 20 #include <optional> 21 #include <utility> 22 23 namespace llvm { 24 25 //===----------------------------------------------------------------------===// 26 // AMDGPU Target Machine (R600+) 27 //===----------------------------------------------------------------------===// 28 29 class AMDGPUTargetMachine : public LLVMTargetMachine { 30 protected: 31 std::unique_ptr<TargetLoweringObjectFile> TLOF; 32 33 StringRef getGPUName(const Function &F) const; 34 StringRef getFeatureString(const Function &F) const; 35 36 public: 37 static bool EnableLateStructurizeCFG; 38 static bool EnableFunctionCalls; 39 static bool EnableLowerModuleLDS; 40 41 AMDGPUTargetMachine(const Target &T, const Triple &TT, StringRef CPU, 42 StringRef FS, TargetOptions Options, 43 std::optional<Reloc::Model> RM, 44 std::optional<CodeModel::Model> CM, CodeGenOpt::Level OL); 45 ~AMDGPUTargetMachine() override; 46 47 const TargetSubtargetInfo *getSubtargetImpl() const; 48 const TargetSubtargetInfo *getSubtargetImpl(const Function &) const override = 0; 49 50 TargetLoweringObjectFile *getObjFileLowering() const override { 51 return TLOF.get(); 52 } 53 54 void registerPassBuilderCallbacks(PassBuilder &PB) override; 55 void registerDefaultAliasAnalyses(AAManager &) override; 56 57 /// Get the integer value of a null pointer in the given address space. 58 static int64_t getNullPointerValue(unsigned AddrSpace); 59 60 bool isNoopAddrSpaceCast(unsigned SrcAS, unsigned DestAS) const override; 61 62 unsigned getAssumedAddrSpace(const Value *V) const override; 63 64 std::pair<const Value *, unsigned> 65 getPredicatedAddrSpace(const Value *V) const override; 66 67 unsigned getAddressSpaceForPseudoSourceKind(unsigned Kind) const override; 68 }; 69 70 //===----------------------------------------------------------------------===// 71 // GCN Target Machine (SI+) 72 //===----------------------------------------------------------------------===// 73 74 class GCNTargetMachine final : public AMDGPUTargetMachine { 75 private: 76 mutable StringMap<std::unique_ptr<GCNSubtarget>> SubtargetMap; 77 78 public: 79 GCNTargetMachine(const Target &T, const Triple &TT, StringRef CPU, 80 StringRef FS, TargetOptions Options, 81 std::optional<Reloc::Model> RM, 82 std::optional<CodeModel::Model> CM, CodeGenOpt::Level OL, 83 bool JIT); 84 85 TargetPassConfig *createPassConfig(PassManagerBase &PM) override; 86 87 const TargetSubtargetInfo *getSubtargetImpl(const Function &) const override; 88 89 TargetTransformInfo getTargetTransformInfo(const Function &F) const override; 90 91 bool useIPRA() const override { 92 return true; 93 } 94 95 MachineFunctionInfo * 96 createMachineFunctionInfo(BumpPtrAllocator &Allocator, const Function &F, 97 const TargetSubtargetInfo *STI) const override; 98 99 yaml::MachineFunctionInfo *createDefaultFuncInfoYAML() const override; 100 yaml::MachineFunctionInfo * 101 convertFuncInfoToYAML(const MachineFunction &MF) const override; 102 bool parseMachineFunctionInfo(const yaml::MachineFunctionInfo &, 103 PerFunctionMIParsingState &PFS, 104 SMDiagnostic &Error, 105 SMRange &SourceRange) const override; 106 }; 107 108 //===----------------------------------------------------------------------===// 109 // AMDGPU Pass Setup 110 //===----------------------------------------------------------------------===// 111 112 class AMDGPUPassConfig : public TargetPassConfig { 113 public: 114 AMDGPUPassConfig(LLVMTargetMachine &TM, PassManagerBase &PM); 115 116 AMDGPUTargetMachine &getAMDGPUTargetMachine() const { 117 return getTM<AMDGPUTargetMachine>(); 118 } 119 120 ScheduleDAGInstrs * 121 createMachineScheduler(MachineSchedContext *C) const override; 122 123 void addEarlyCSEOrGVNPass(); 124 void addStraightLineScalarOptimizationPasses(); 125 void addIRPasses() override; 126 void addCodeGenPrepare() override; 127 bool addPreISel() override; 128 bool addInstSelector() override; 129 bool addGCPasses() override; 130 131 std::unique_ptr<CSEConfigBase> getCSEConfig() const override; 132 133 /// Check if a pass is enabled given \p Opt option. The option always 134 /// overrides defaults if explicitly used. Otherwise its default will 135 /// be used given that a pass shall work at an optimization \p Level 136 /// minimum. 137 bool isPassEnabled(const cl::opt<bool> &Opt, 138 CodeGenOpt::Level Level = CodeGenOpt::Default) const { 139 if (Opt.getNumOccurrences()) 140 return Opt; 141 if (TM->getOptLevel() < Level) 142 return false; 143 return Opt; 144 } 145 }; 146 147 } // end namespace llvm 148 149 #endif // LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETMACHINE_H 150