21#include "llvm/Support/Alignment.h"
22#include "llvm/Support/FileSystem.h"
23#include "llvm/Support/Path.h"
24#include "llvm/TargetParser/TargetParser.h"
32#if defined(_WIN32) || defined(_WIN64)
33#define NULL_FILE "nul"
35#define NULL_FILE "/dev/null"
39 const llvm::opt::ArgList &DriverArgs,
41 const llvm::opt::Arg *A) {
45 Option O = A->getOption();
46 if (!O.matches(options::OPT_fsanitize_EQ))
49 if (!DriverArgs.hasFlag(options::OPT_fgpu_sanitize,
50 options::OPT_fno_gpu_sanitize,
true))
57 if (K != SanitizerKind::Address)
60 llvm::StringMap<bool> FeatureMap;
63 assert(OptionalGpuArch &&
"Invalid Target ID");
64 (void)OptionalGpuArch;
65 auto Loc = FeatureMap.find(
"xnack");
66 if (Loc == FeatureMap.end() || !Loc->second) {
68 clang::diag::warn_drv_unsupported_option_for_offload_arch_req_feature)
69 << A->getAsString(DriverArgs) << TargetID <<
"xnack+";
75void AMDGCN::Linker::constructLlvmLinkCommand(
Compilation &
C,
79 const llvm::opt::ArgList &Args)
const {
82 ArgStringList LlvmLinkArgs;
84 assert(!Inputs.empty() &&
"Must have at least one input.");
87 for (
auto Input : Inputs)
88 LlvmLinkArgs.push_back(Input.getFilename());
92 auto TargetID = Args.getLastArgValue(options::OPT_mcpu_EQ);
96 const char *LlvmLink =
97 Args.MakeArgString(
getToolChain().GetProgramPath(
"llvm-link"));
99 LlvmLink, LlvmLinkArgs, Inputs,
106 const llvm::opt::ArgList &Args)
const {
109 ArgStringList LldArgs{
"-flavor",
115 "-plugin-opt=-amdgpu-internalize-symbols"};
116 if (Args.hasArg(options::OPT_hipstdpar))
117 LldArgs.push_back(
"-plugin-opt=-amdgpu-enable-hipstdpar");
119 auto &TC = getToolChain();
120 auto &D = TC.getDriver();
121 assert(!Inputs.empty() &&
"Must have at least one input.");
122 bool IsThinLTO = D.getLTOMode(
true) ==
LTOK_Thin;
123 addLTOOptions(TC, Args, LldArgs, Output, Inputs[0], IsThinLTO);
126 std::vector<llvm::StringRef> Features;
130 std::string MAttrString =
"-plugin-opt=-mattr=";
132 MAttrString.append(Args.MakeArgString(OneFeature));
133 if (OneFeature != Features.back())
134 MAttrString.append(
",");
136 if (!Features.empty())
137 LldArgs.push_back(Args.MakeArgString(MAttrString));
143 LldArgs.push_back(Args.MakeArgString(
"-plugin-opt=-force-import-all"));
145 for (
const Arg *A : Args.filtered(options::OPT_mllvm)) {
147 Args.MakeArgString(Twine(
"-plugin-opt=") + A->getValue(0)));
150 if (
C.getDriver().isSaveTempsEnabled())
151 LldArgs.push_back(
"-save-temps");
165 LldArgs.push_back(
"--whole-archive");
167 for (
auto *Arg : Args.filtered(options::OPT_Xoffload_linker)) {
168 StringRef ArgVal = Arg->getValue(1);
169 auto SplitArg = ArgVal.split(
"-mllvm=");
170 if (!SplitArg.second.empty()) {
172 Args.MakeArgString(Twine(
"-plugin-opt=") + SplitArg.second));
174 LldArgs.push_back(Args.MakeArgString(ArgVal));
180 for (
auto Input : Inputs)
181 LldArgs.push_back(Input.getFilename());
185 auto TargetID = Args.getLastArgValue(options::OPT_mcpu_EQ);
189 LldArgs.push_back(
"--no-whole-archive");
191 const char *Lld = Args.MakeArgString(getToolChain().GetProgramPath(
"lld"));
193 Lld, LldArgs, Inputs, Output));
203 const char *LinkingOutput)
const {
204 if (Inputs.size() > 0 &&
205 Inputs[0].getType() == types::TY_Image &&
206 JA.
getType() == types::TY_Object)
210 if (JA.
getType() == types::TY_HIP_FATBIN)
214 if (JA.
getType() == types::TY_LLVM_BC)
215 return constructLlvmLinkCommand(
C, JA, Inputs, Output, Args);
217 return constructLldCommand(
C, JA, Inputs, Output, Args);
221 const ToolChain &HostTC,
const ArgList &Args)
228 if (!Args.hasFlag(options::OPT_fgpu_sanitize, options::OPT_fno_gpu_sanitize,
231 for (
auto *A : Args.filtered(options::OPT_fsanitize_EQ)) {
233 if (K != SanitizerKind::Address)
234 D.
getDiags().
Report(clang::diag::warn_drv_unsupported_option_for_target)
235 << A->getAsString(Args) <<
getTriple().str();
240 const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args,
245 "Only HIP offloading kinds are supported for GPUs.");
247 CC1Args.push_back(
"-fcuda-is-device");
249 if (!DriverArgs.hasFlag(options::OPT_fgpu_rdc, options::OPT_fno_gpu_rdc,
251 CC1Args.append({
"-mllvm",
"-amdgpu-internalize-symbols"});
252 if (DriverArgs.hasArgNoClaim(options::OPT_hipstdpar))
253 CC1Args.append({
"-mllvm",
"-amdgpu-enable-hipstdpar"});
255 StringRef MaxThreadsPerBlock =
256 DriverArgs.getLastArgValue(options::OPT_gpu_max_threads_per_block_EQ);
257 if (!MaxThreadsPerBlock.empty()) {
259 (Twine(
"--gpu-max-threads-per-block=") + MaxThreadsPerBlock).str();
260 CC1Args.push_back(DriverArgs.MakeArgStringRef(ArgStr));
263 CC1Args.push_back(
"-fcuda-allow-variadic-functions");
267 if (!DriverArgs.hasArg(options::OPT_fvisibility_EQ,
268 options::OPT_fvisibility_ms_compat)) {
269 CC1Args.append({
"-fvisibility=hidden"});
270 CC1Args.push_back(
"-fapply-global-visibility-to-externs");
274 CC1Args.push_back(BCFile.ShouldInternalize ?
"-mlink-builtin-bitcode"
275 :
"-mlink-bitcode-file");
276 CC1Args.push_back(DriverArgs.MakeArgString(BCFile.Path));
280llvm::opt::DerivedArgList *
284 DerivedArgList *DAL =
287 DAL =
new DerivedArgList(Args.getBaseArgs());
291 for (Arg *A : Args) {
296 if (!BoundArch.empty()) {
297 DAL->eraseArg(options::OPT_mcpu_EQ);
298 DAL->AddJoinedArg(
nullptr, Opts.getOption(options::OPT_mcpu_EQ), BoundArch);
321 ArgStringList &CC1Args)
const {
326 const ArgList &Args, ArgStringList &CC1Args)
const {
331 ArgStringList &CC1Args)
const {
336 ArgStringList &CC1Args)
const {
354 const ArgList &Args)
const {
361 if (DriverArgs.hasArg(options::OPT_nogpulib))
363 ArgStringList LibraryPaths;
367 LibraryPaths.push_back(DriverArgs.MakeArgString(Path));
372 auto BCLibArgs = DriverArgs.getAllArgValues(options::OPT_hip_device_lib_EQ);
373 if (!BCLibArgs.empty()) {
374 llvm::for_each(BCLibArgs, [&](StringRef BCName) {
376 for (StringRef LibraryPath : LibraryPaths) {
378 llvm::sys::path::append(Path, BCName);
380 if (llvm::sys::fs::exists(FullName)) {
381 BCLibs.push_back(FullName);
393 assert(!GpuArch.empty() &&
"Must have an explicit GPU arch.");
396 if (DriverArgs.hasFlag(options::OPT_fgpu_sanitize,
397 options::OPT_fno_gpu_sanitize,
true) &&
400 if (AsanRTL.empty()) {
403 "AMDGPU address sanitizer runtime library (asanrtl) is not found. "
404 "Please install ROCm device library which supports address "
409 BCLibs.emplace_back(AsanRTL,
false);
417 BCLibs.emplace_back(N);
421 DriverArgs.getLastArgValue(options::OPT_gpu_instrument_lib_EQ);
424 if (llvm::sys::fs::exists(InstLib))
425 BCLibs.push_back(InstLib);
434 const llvm::opt::ArgList &DriverArgs)
const {
436 if (PTID.OptionalTargetID && !PTID.OptionalGPUArch) {
438 << *PTID.OptionalTargetID;
static bool shouldSkipSanitizeOption(const ToolChain &TC, const llvm::opt::ArgList &DriverArgs, StringRef TargetID, const llvm::opt::Arg *A)
DiagnosticBuilder Report(SourceLocation Loc, unsigned DiagID)
Issue the message to the client.
unsigned getCustomDiagID(Level L, const char(&FormatString)[N])
Return an ID for a diagnostic with the specified format string and level.
types::ID getType() const
Compilation - A set of tasks to perform for a single driver invocation.
Driver - Encapsulate logic for constructing compilation processes from a set of gcc-driver-like comma...
DiagnosticsEngine & getDiags() const
DiagnosticBuilder Diag(unsigned DiagID) const
const llvm::opt::OptTable & getOpts() const
The JSON file list parser is used to communicate input to InstallAPI.
std::optional< llvm::StringRef > parseTargetID(const llvm::Triple &T, llvm::StringRef OffloadArch, llvm::StringMap< bool > *FeatureMap)
Parse a target ID to get processor and feature map.
SanitizerMask parseSanitizerValue(StringRef Value, bool AllowGroups)
Parse a single value from a -fsanitize= or -fno-sanitize= value list.
static constexpr ResponseFileSupport None()
Returns a ResponseFileSupport indicating that response files are not supported.