20#include "llvm/Support/FileSystem.h"
21#include "llvm/Support/Path.h"
22#include "llvm/TargetParser/TargetParser.h"
30#if defined(_WIN32) || defined(_WIN64)
31#define NULL_FILE "nul"
33#define NULL_FILE "/dev/null"
36void AMDGCN::Linker::constructLlvmLinkCommand(
Compilation &
C,
40 const llvm::opt::ArgList &Args)
const {
43 ArgStringList LlvmLinkArgs;
45 assert(!Inputs.empty() &&
"Must have at least one input.");
48 for (
auto Input : Inputs)
49 LlvmLinkArgs.push_back(Input.getFilename());
53 auto TargetID = Args.getLastArgValue(options::OPT_mcpu_EQ);
57 const char *LlvmLink =
58 Args.MakeArgString(
getToolChain().GetProgramPath(
"llvm-link"));
60 LlvmLink, LlvmLinkArgs, Inputs,
64void AMDGCN::Linker::constructLldCommand(Compilation &
C,
const JobAction &JA,
66 const InputInfo &Output,
67 const llvm::opt::ArgList &Args)
const {
70 ArgStringList LldArgs{
"-flavor",
76 "-plugin-opt=-amdgpu-internalize-symbols"};
77 if (Args.hasArg(options::OPT_hipstdpar))
78 LldArgs.push_back(
"-plugin-opt=-amdgpu-enable-hipstdpar");
80 auto &TC = getToolChain();
81 auto &D = TC.getDriver();
82 bool IsThinLTO = D.getOffloadLTOMode() ==
LTOK_Thin;
86 std::vector<llvm::StringRef> Features;
90 std::string MAttrString =
"-plugin-opt=-mattr=";
92 MAttrString.append(Args.MakeArgString(OneFeature));
93 if (OneFeature != Features.back())
94 MAttrString.append(
",");
96 if (!Features.empty())
97 LldArgs.push_back(Args.MakeArgString(MAttrString));
103 LldArgs.push_back(Args.MakeArgString(
"-plugin-opt=-force-import-all"));
104 LldArgs.push_back(Args.MakeArgString(
"-plugin-opt=-avail-extern-to-local"));
105 LldArgs.push_back(Args.MakeArgString(
106 "-plugin-opt=-avail-extern-gv-in-addrspace-to-local=3"));
109 for (
const Arg *A : Args.filtered(options::OPT_mllvm)) {
111 Args.MakeArgString(Twine(
"-plugin-opt=") + A->getValue(0)));
114 if (
C.getDriver().isSaveTempsEnabled())
115 LldArgs.push_back(
"-save-temps");
129 LldArgs.push_back(
"--whole-archive");
131 for (
auto *Arg : Args.filtered(options::OPT_Xoffload_linker)) {
132 StringRef ArgVal = Arg->getValue(1);
133 auto SplitArg = ArgVal.split(
"-mllvm=");
134 if (!SplitArg.second.empty()) {
136 Args.MakeArgString(Twine(
"-plugin-opt=") + SplitArg.second));
138 LldArgs.push_back(Args.MakeArgString(ArgVal));
144 for (
auto Input : Inputs)
145 LldArgs.push_back(Input.getFilename());
149 auto TargetID = Args.getLastArgValue(options::OPT_mcpu_EQ);
153 LldArgs.push_back(
"--no-whole-archive");
155 const char *Lld = Args.MakeArgString(getToolChain().GetProgramPath(
"lld"));
157 Lld, LldArgs, Inputs, Output));
166void AMDGCN::Linker::constructLinkAndEmitSpirvCommand(
168 const InputInfo &Output,
const llvm::opt::ArgList &Args)
const {
169 assert(!Inputs.empty() &&
"Must have at least one input.");
171 std::string LinkedBCFilePrefix(
172 Twine(llvm::sys::path::stem(Output.
getFilename()),
"-linked").str());
174 InputInfo LinkedBCFile(&JA, LinkedBCFilePath, Output.
getBaseInput());
176 constructLlvmLinkCommand(
C, JA, Inputs, LinkedBCFile, Args);
179 llvm::opt::ArgStringList TrArgs{
180 "--spirv-max-version=1.6",
182 "--spirv-allow-unknown-intrinsics",
183 "--spirv-lower-const-expr",
184 "--spirv-preserve-auxdata",
185 "--spirv-debug-info-version=nonsemantic-shader-200"};
196 const char *LinkingOutput)
const {
197 if (Inputs.size() > 0 &&
198 Inputs[0].getType() == types::TY_Image &&
199 JA.
getType() == types::TY_Object)
203 if (JA.
getType() == types::TY_HIP_FATBIN)
207 if (JA.
getType() == types::TY_LLVM_BC)
208 return constructLlvmLinkCommand(
C, JA, Inputs, Output, Args);
211 return constructLinkAndEmitSpirvCommand(
C, JA, Inputs, Output, Args);
213 return constructLldCommand(
C, JA, Inputs, Output, Args);
227 const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args,
229 HostTC.addClangTargetOptions(DriverArgs, CC1Args, DeviceOffloadingKind);
232 "Only HIP offloading kinds are supported for GPUs.");
234 CC1Args.append({
"-fcuda-is-device",
"-fno-threadsafe-statics"});
236 if (!DriverArgs.hasFlag(options::OPT_fgpu_rdc, options::OPT_fno_gpu_rdc,
238 CC1Args.append({
"-mllvm",
"-amdgpu-internalize-symbols"});
239 if (DriverArgs.hasArgNoClaim(options::OPT_hipstdpar))
240 CC1Args.append({
"-mllvm",
"-amdgpu-enable-hipstdpar"});
243 StringRef MaxThreadsPerBlock =
244 DriverArgs.getLastArgValue(options::OPT_gpu_max_threads_per_block_EQ);
245 if (!MaxThreadsPerBlock.empty()) {
247 (Twine(
"--gpu-max-threads-per-block=") + MaxThreadsPerBlock).str();
248 CC1Args.push_back(DriverArgs.MakeArgStringRef(ArgStr));
251 CC1Args.push_back(
"-fcuda-allow-variadic-functions");
255 if (!DriverArgs.hasArg(options::OPT_fvisibility_EQ,
256 options::OPT_fvisibility_ms_compat)) {
257 CC1Args.append({
"-fvisibility=hidden"});
258 CC1Args.push_back(
"-fapply-global-visibility-to-externs");
265 if (!DriverArgs.hasArg(options::OPT_fembed_bitcode_marker))
266 CC1Args.push_back(
"-fembed-bitcode=marker");
271 if (!DriverArgs.hasArg(options::OPT_disable_llvm_passes))
272 CC1Args.push_back(
"-disable-llvm-passes");
276 for (
auto BCFile :
getDeviceLibs(DriverArgs, DeviceOffloadingKind)) {
277 CC1Args.push_back(BCFile.ShouldInternalize ?
"-mlink-builtin-bitcode"
278 :
"-mlink-bitcode-file");
279 CC1Args.push_back(DriverArgs.MakeArgString(BCFile.Path));
283llvm::opt::DerivedArgList *
287 DerivedArgList *DAL =
288 HostTC.TranslateArgs(Args, BoundArch, DeviceOffloadKind);
290 DAL =
new DerivedArgList(Args.getBaseArgs());
294 for (Arg *A : Args) {
299 if (!BoundArch.empty()) {
300 DAL->eraseArg(options::OPT_mcpu_EQ);
301 DAL->AddJoinedArg(
nullptr, Opts.getOption(options::OPT_mcpu_EQ), BoundArch);
305 if (!Args.hasArg(options::OPT_flto_partitions_EQ))
306 DAL->AddJoinedArg(
nullptr, Opts.getOption(options::OPT_flto_partitions_EQ),
320 HostTC.addClangWarningOptions(CC1Args);
325 return HostTC.GetCXXStdlibType(Args);
329 ArgStringList &CC1Args)
const {
330 HostTC.AddClangSystemIncludeArgs(DriverArgs, CC1Args);
334 const ArgList &Args, ArgStringList &CC1Args)
const {
335 HostTC.AddClangCXXStdlibIncludeArgs(Args, CC1Args);
339 ArgStringList &CC1Args)
const {
340 HostTC.AddIAMCUIncludeArgs(Args, CC1Args);
344 ArgStringList &CC1Args)
const {
358 return HostTC.getSupportedSanitizers();
362 const ArgList &Args)
const {
363 return HostTC.computeMSVCVersion(D, Args);
370 if (!DriverArgs.hasFlag(options::OPT_offloadlib, options::OPT_no_offloadlib,
374 ArgStringList LibraryPaths;
378 LibraryPaths.push_back(DriverArgs.MakeArgString(Path));
383 auto BCLibArgs = DriverArgs.getAllArgValues(options::OPT_hip_device_lib_EQ);
384 if (!BCLibArgs.empty()) {
385 for (StringRef BCName : BCLibArgs) {
388 for (StringRef LibraryPath : LibraryPaths) {
390 llvm::sys::path::append(Path, BCName);
392 if (llvm::sys::fs::exists(FullName)) {
393 BCLibs.emplace_back(FullName);
407 assert(!GpuArch.empty() &&
"Must have an explicit GPU arch.");
411 DeviceOffloadingKind))
412 BCLibs.emplace_back(N);
416 DriverArgs.getLastArgValue(options::OPT_gpu_instrument_lib_EQ);
419 if (llvm::sys::fs::exists(InstLib))
420 BCLibs.emplace_back(InstLib);
429 const llvm::opt::ArgList &DriverArgs)
const {
431 if (PTID.OptionalTargetID && !PTID.OptionalGPUArch &&
432 PTID.OptionalTargetID !=
"amdgcnspirv")
434 << *PTID.OptionalTargetID;
438 const llvm::Triple &Triple,
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...
DiagnosticBuilder Diag(unsigned DiagID) const
const llvm::opt::OptTable & getOpts() const
SmallVector< InputInfo, 4 > InputInfoList
The JSON file list parser is used to communicate input to InstallAPI.
static constexpr ResponseFileSupport None()
Returns a ResponseFileSupport indicating that response files are not supported.