1 //===--- AMDGPU.h - AMDGPU ToolChain Implementations ----------*- 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 #ifndef LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H 10 #define LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H 11 12 #include "Gnu.h" 13 #include "ROCm.h" 14 #include "clang/Basic/TargetID.h" 15 #include "clang/Driver/Options.h" 16 #include "clang/Driver/Tool.h" 17 #include "clang/Driver/ToolChain.h" 18 #include "llvm/ADT/SmallString.h" 19 #include "llvm/TargetParser/TargetParser.h" 20 21 #include <map> 22 23 namespace clang { 24 namespace driver { 25 26 namespace tools { 27 namespace amdgpu { 28 29 class LLVM_LIBRARY_VISIBILITY Linker final : public Tool { 30 public: 31 Linker(const ToolChain &TC) : Tool("amdgpu::Linker", "ld.lld", TC) {} 32 bool isLinkJob() const override { return true; } 33 bool hasIntegratedCPP() const override { return false; } 34 void ConstructJob(Compilation &C, const JobAction &JA, 35 const InputInfo &Output, const InputInfoList &Inputs, 36 const llvm::opt::ArgList &TCArgs, 37 const char *LinkingOutput) const override; 38 }; 39 40 void getAMDGPUTargetFeatures(const Driver &D, const llvm::Triple &Triple, 41 const llvm::opt::ArgList &Args, 42 std::vector<StringRef> &Features); 43 44 } // end namespace amdgpu 45 } // end namespace tools 46 47 namespace toolchains { 48 49 class LLVM_LIBRARY_VISIBILITY AMDGPUToolChain : public Generic_ELF { 50 protected: 51 const std::map<options::ID, const StringRef> OptionsDefault; 52 53 Tool *buildLinker() const override; 54 StringRef getOptionDefault(options::ID OptID) const { 55 auto opt = OptionsDefault.find(OptID); 56 assert(opt != OptionsDefault.end() && "No Default for Option"); 57 return opt->second; 58 } 59 60 public: 61 AMDGPUToolChain(const Driver &D, const llvm::Triple &Triple, 62 const llvm::opt::ArgList &Args); 63 unsigned GetDefaultDwarfVersion() const override { return 5; } 64 65 bool IsMathErrnoDefault() const override { return false; } 66 bool isCrossCompiling() const override { return true; } 67 bool isPICDefault() const override { return true; } 68 bool isPIEDefault(const llvm::opt::ArgList &Args) const override { 69 return false; 70 } 71 bool isPICDefaultForced() const override { return true; } 72 bool SupportsProfiling() const override { return false; } 73 74 llvm::opt::DerivedArgList * 75 TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch, 76 Action::OffloadKind DeviceOffloadKind) const override; 77 78 void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs, 79 llvm::opt::ArgStringList &CC1Args, 80 Action::OffloadKind DeviceOffloadKind) const override; 81 82 /// Return whether denormals should be flushed, and treated as 0 by default 83 /// for the subtarget. 84 static bool getDefaultDenormsAreZeroForTarget(llvm::AMDGPU::GPUKind GPUKind); 85 86 llvm::DenormalMode getDefaultDenormalModeForType( 87 const llvm::opt::ArgList &DriverArgs, const JobAction &JA, 88 const llvm::fltSemantics *FPType = nullptr) const override; 89 90 static bool isWave64(const llvm::opt::ArgList &DriverArgs, 91 llvm::AMDGPU::GPUKind Kind); 92 /// Needed for using lto. 93 bool HasNativeLLVMSupport() const override { 94 return true; 95 } 96 97 /// Needed for translating LTO options. 98 const char *getDefaultLinker() const override { return "ld.lld"; } 99 100 /// Should skip sanitize options. 101 bool shouldSkipSanitizeOption(const ToolChain &TC, 102 const llvm::opt::ArgList &DriverArgs, 103 StringRef TargetID, 104 const llvm::opt::Arg *A) const; 105 106 /// Uses amdgpu-arch tool to get arch of the system GPU. Will return error 107 /// if unable to find one. 108 virtual Expected<SmallVector<std::string>> 109 getSystemGPUArchs(const llvm::opt::ArgList &Args) const override; 110 111 protected: 112 /// Check and diagnose invalid target ID specified by -mcpu. 113 virtual void checkTargetID(const llvm::opt::ArgList &DriverArgs) const; 114 115 /// The struct type returned by getParsedTargetID. 116 struct ParsedTargetIDType { 117 std::optional<std::string> OptionalTargetID; 118 std::optional<std::string> OptionalGPUArch; 119 std::optional<llvm::StringMap<bool>> OptionalFeatures; 120 }; 121 122 /// Get target ID, GPU arch, and target ID features if the target ID is 123 /// specified and valid. 124 ParsedTargetIDType 125 getParsedTargetID(const llvm::opt::ArgList &DriverArgs) const; 126 127 /// Get GPU arch from -mcpu without checking. 128 StringRef getGPUArch(const llvm::opt::ArgList &DriverArgs) const; 129 130 /// Common warning options shared by AMDGPU HIP, OpenCL and OpenMP toolchains. 131 /// Language specific warning options should go to derived classes. 132 void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override; 133 }; 134 135 class LLVM_LIBRARY_VISIBILITY ROCMToolChain : public AMDGPUToolChain { 136 public: 137 ROCMToolChain(const Driver &D, const llvm::Triple &Triple, 138 const llvm::opt::ArgList &Args); 139 void 140 addClangTargetOptions(const llvm::opt::ArgList &DriverArgs, 141 llvm::opt::ArgStringList &CC1Args, 142 Action::OffloadKind DeviceOffloadKind) const override; 143 144 // Returns a list of device library names shared by different languages 145 llvm::SmallVector<std::string, 12> 146 getCommonDeviceLibNames(const llvm::opt::ArgList &DriverArgs, 147 const std::string &GPUArch, 148 bool isOpenMP = false) const; 149 SanitizerMask getSupportedSanitizers() const override { 150 return SanitizerKind::Address; 151 } 152 }; 153 154 } // end namespace toolchains 155 } // end namespace driver 156 } // end namespace clang 157 158 #endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H 159