clang 23.0.0git
AMDGPU.cpp
Go to the documentation of this file.
1//===--- AMDGPU.cpp - 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#include "AMDGPU.h"
11#include "clang/Config/config.h"
17#include "llvm/ADT/StringExtras.h"
18#include "llvm/Option/ArgList.h"
19#include "llvm/Support/Error.h"
20#include "llvm/Support/LineIterator.h"
21#include "llvm/Support/Path.h"
22#include "llvm/Support/Process.h"
23#include "llvm/Support/VirtualFileSystem.h"
24#include "llvm/TargetParser/Host.h"
25#include "llvm/TargetParser/TargetParser.h"
26#include <optional>
27#include <system_error>
28
29using namespace clang::driver;
30using namespace clang::driver::tools;
31using namespace clang::driver::toolchains;
32using namespace clang;
33using namespace llvm::opt;
34
35RocmInstallationDetector::CommonBitcodeLibsPreferences::
36 CommonBitcodeLibsPreferences(const Driver &D,
37 const llvm::opt::ArgList &DriverArgs,
38 StringRef GPUArch,
39 const Action::OffloadKind DeviceOffloadingKind,
40 const bool NeedsASanRT)
41 : ABIVer(DeviceLibABIVersion::fromCodeObjectVersion(
42 tools::getAMDGPUCodeObjectVersion(D, DriverArgs))) {
43 const auto Kind = llvm::AMDGPU::parseArchAMDGCN(GPUArch);
44 const unsigned ArchAttr = llvm::AMDGPU::getArchAttrAMDGCN(Kind);
45
46 IsOpenMP = DeviceOffloadingKind == Action::OFK_OpenMP;
47
48 const bool HasWave32 = (ArchAttr & llvm::AMDGPU::FEATURE_WAVE32);
49 Wave64 =
50 !HasWave32 || DriverArgs.hasFlag(options::OPT_mwavefrontsize64,
51 options::OPT_mno_wavefrontsize64, false);
52
53 const bool IsKnownOffloading = DeviceOffloadingKind == Action::OFK_OpenMP ||
54 DeviceOffloadingKind == Action::OFK_HIP;
55
56 // Default to enabling f32 denormals on subtargets where fma is fast with
57 // denormals
58 const bool DefaultDAZ =
59 (Kind == llvm::AMDGPU::GK_NONE)
60 ? false
61 : !((ArchAttr & llvm::AMDGPU::FEATURE_FAST_FMA_F32) &&
62 (ArchAttr & llvm::AMDGPU::FEATURE_FAST_DENORMAL_F32));
63 // TODO: There are way too many flags that change this. Do we need to
64 // check them all?
65 DAZ = IsKnownOffloading
66 ? DriverArgs.hasFlag(options::OPT_fgpu_flush_denormals_to_zero,
67 options::OPT_fno_gpu_flush_denormals_to_zero,
68 DefaultDAZ)
69 : DriverArgs.hasArg(options::OPT_cl_denorms_are_zero) || DefaultDAZ;
70
71 FiniteOnly = DriverArgs.hasArg(options::OPT_cl_finite_math_only) ||
72 DriverArgs.hasFlag(options::OPT_ffinite_math_only,
73 options::OPT_fno_finite_math_only, false);
74
75 UnsafeMathOpt =
76 DriverArgs.hasArg(options::OPT_cl_unsafe_math_optimizations) ||
77 DriverArgs.hasFlag(options::OPT_funsafe_math_optimizations,
78 options::OPT_fno_unsafe_math_optimizations, false);
79
80 FastRelaxedMath = DriverArgs.hasArg(options::OPT_cl_fast_relaxed_math) ||
81 DriverArgs.hasFlag(options::OPT_ffast_math,
82 options::OPT_fno_fast_math, false);
83
84 // GPU Sanitizer currently only supports ASan and is enabled through host
85 // ASan.
86 GPUSan = (DriverArgs.hasFlag(options::OPT_fgpu_sanitize,
87 options::OPT_fno_gpu_sanitize, true) &&
88 NeedsASanRT);
89}
90
91void RocmInstallationDetector::scanLibDevicePath(llvm::StringRef Path) {
92 assert(!Path.empty());
93
94 const StringRef Suffix(".bc");
95 const StringRef Suffix2(".amdgcn.bc");
96
97 std::error_code EC;
98 for (llvm::vfs::directory_iterator LI = D.getVFS().dir_begin(Path, EC), LE;
99 !EC && LI != LE; LI = LI.increment(EC)) {
100 StringRef FilePath = LI->path();
101 StringRef FileName = llvm::sys::path::filename(FilePath);
102 if (!FileName.ends_with(Suffix))
103 continue;
104
105 StringRef BaseName;
106 if (FileName.ends_with(Suffix2))
107 BaseName = FileName.drop_back(Suffix2.size());
108 else if (FileName.ends_with(Suffix))
109 BaseName = FileName.drop_back(Suffix.size());
110
111 const StringRef ABIVersionPrefix = "oclc_abi_version_";
112 if (BaseName == "ocml") {
113 OCML = FilePath;
114 } else if (BaseName == "ockl") {
115 OCKL = FilePath;
116 } else if (BaseName == "opencl") {
117 OpenCL = FilePath;
118 } else if (BaseName == "asanrtl") {
119 AsanRTL = FilePath;
120 } else if (BaseName == "oclc_finite_only_off") {
121 FiniteOnly.Off = FilePath;
122 } else if (BaseName == "oclc_finite_only_on") {
123 FiniteOnly.On = FilePath;
124 } else if (BaseName == "oclc_unsafe_math_on") {
125 UnsafeMath.On = FilePath;
126 } else if (BaseName == "oclc_unsafe_math_off") {
127 UnsafeMath.Off = FilePath;
128 } else if (BaseName == "oclc_wavefrontsize64_on") {
129 WavefrontSize64.On = FilePath;
130 } else if (BaseName == "oclc_wavefrontsize64_off") {
131 WavefrontSize64.Off = FilePath;
132 } else if (BaseName.starts_with(ABIVersionPrefix)) {
133 unsigned ABIVersionNumber;
134 if (BaseName.drop_front(ABIVersionPrefix.size())
135 .getAsInteger(/*Redex=*/0, ABIVersionNumber))
136 continue;
137 ABIVersionMap[ABIVersionNumber] = FilePath.str();
138 } else {
139 // Process all bitcode filenames that look like
140 // ocl_isa_version_XXX.amdgcn.bc
141 const StringRef DeviceLibPrefix = "oclc_isa_version_";
142 if (!BaseName.starts_with(DeviceLibPrefix))
143 continue;
144
145 StringRef IsaVersionNumber =
146 BaseName.drop_front(DeviceLibPrefix.size());
147
148 llvm::Twine GfxName = Twine("gfx") + IsaVersionNumber;
149 SmallString<8> Tmp;
150 LibDeviceMap.insert({GfxName.toStringRef(Tmp), FilePath.str()});
151 }
152 }
153}
154
155// Parse and extract version numbers from `.hipVersion`. Return `true` if
156// the parsing fails.
157bool RocmInstallationDetector::parseHIPVersionFile(llvm::StringRef V) {
158 SmallVector<StringRef, 4> VersionParts;
159 V.split(VersionParts, '\n');
160 unsigned Major = ~0U;
161 unsigned Minor = ~0U;
162 for (auto Part : VersionParts) {
163 auto Splits = Part.rtrim().split('=');
164 if (Splits.first == "HIP_VERSION_MAJOR") {
165 if (Splits.second.getAsInteger(0, Major))
166 return true;
167 } else if (Splits.first == "HIP_VERSION_MINOR") {
168 if (Splits.second.getAsInteger(0, Minor))
169 return true;
170 } else if (Splits.first == "HIP_VERSION_PATCH")
171 VersionPatch = Splits.second.str();
172 }
173 if (Major == ~0U || Minor == ~0U)
174 return true;
175 VersionMajorMinor = llvm::VersionTuple(Major, Minor);
176 DetectedVersion =
177 (Twine(Major) + "." + Twine(Minor) + "." + VersionPatch).str();
178 return false;
179}
180
181/// \returns a list of candidate directories for ROCm installation, which is
182/// cached and populated only once.
183const SmallVectorImpl<RocmInstallationDetector::Candidate> &
184RocmInstallationDetector::getInstallationPathCandidates() {
185
186 // Return the cached candidate list if it has already been populated.
187 if (!ROCmSearchDirs.empty())
188 return ROCmSearchDirs;
189
190 auto DoPrintROCmSearchDirs = [&]() {
191 if (PrintROCmSearchDirs)
192 for (auto Cand : ROCmSearchDirs) {
193 llvm::errs() << "ROCm installation search path: " << Cand.Path << '\n';
194 }
195 };
196
197 // For candidate specified by --rocm-path we do not do strict check, i.e.,
198 // checking existence of HIP version file and device library files.
199 if (!RocmPathArg.empty()) {
200 ROCmSearchDirs.emplace_back(RocmPathArg.str());
201 DoPrintROCmSearchDirs();
202 return ROCmSearchDirs;
203 } else if (std::optional<std::string> RocmPathEnv =
204 llvm::sys::Process::GetEnv("ROCM_PATH")) {
205 if (!RocmPathEnv->empty()) {
206 ROCmSearchDirs.emplace_back(std::move(*RocmPathEnv));
207 DoPrintROCmSearchDirs();
208 return ROCmSearchDirs;
209 }
210 }
211
212 // Try to find relative to the compiler binary.
213 StringRef InstallDir = D.Dir;
214
215 // Check both a normal Unix prefix position of the clang binary, as well as
216 // the Windows-esque layout the ROCm packages use with the host architecture
217 // subdirectory of bin.
218 auto DeduceROCmPath = [](StringRef ClangPath) {
219 // Strip off directory (usually bin)
220 StringRef ParentDir = llvm::sys::path::parent_path(ClangPath);
221 StringRef ParentName = llvm::sys::path::filename(ParentDir);
222
223 // Some builds use bin/{host arch}, so go up again.
224 if (ParentName == "bin") {
225 ParentDir = llvm::sys::path::parent_path(ParentDir);
226 ParentName = llvm::sys::path::filename(ParentDir);
227 }
228
229 // Some versions of the rocm llvm package install to /opt/rocm/llvm/bin
230 // Some versions of the aomp package install to /opt/rocm/aomp/bin
231 if (ParentName == "llvm" || ParentName.starts_with("aomp")) {
232 ParentDir = llvm::sys::path::parent_path(ParentDir);
233 ParentName = llvm::sys::path::filename(ParentDir);
234
235 // Some versions of the rocm llvm package install to
236 // /opt/rocm/lib/llvm/bin, so also back up if within the lib dir still
237 if (ParentName == "lib")
238 ParentDir = llvm::sys::path::parent_path(ParentDir);
239 }
240
241 return Candidate(ParentDir.str(), /*StrictChecking=*/true);
242 };
243
244 // Deduce ROCm path by the path used to invoke clang. Do not resolve symbolic
245 // link of clang itself.
246 ROCmSearchDirs.emplace_back(DeduceROCmPath(InstallDir));
247
248 // Deduce ROCm path by the real path of the invoked clang, resolving symbolic
249 // link of clang itself.
250 llvm::SmallString<256> RealClangPath;
251 llvm::sys::fs::real_path(D.getClangProgramPath(), RealClangPath);
252 auto ParentPath = llvm::sys::path::parent_path(RealClangPath);
253 if (ParentPath != InstallDir)
254 ROCmSearchDirs.emplace_back(DeduceROCmPath(ParentPath));
255
256 // Device library may be installed in clang or resource directory.
257 auto ClangRoot = llvm::sys::path::parent_path(InstallDir);
258 auto RealClangRoot = llvm::sys::path::parent_path(ParentPath);
259 ROCmSearchDirs.emplace_back(ClangRoot.str(), /*StrictChecking=*/true);
260 if (RealClangRoot != ClangRoot)
261 ROCmSearchDirs.emplace_back(RealClangRoot.str(), /*StrictChecking=*/true);
262 ROCmSearchDirs.emplace_back(D.ResourceDir,
263 /*StrictChecking=*/true);
264
265 ROCmSearchDirs.emplace_back(D.SysRoot + "/opt/rocm",
266 /*StrictChecking=*/true);
267
268 // Find the latest /opt/rocm-{release} directory.
269 std::error_code EC;
270 std::string LatestROCm;
271 llvm::VersionTuple LatestVer;
272 // Get ROCm version from ROCm directory name.
273 auto GetROCmVersion = [](StringRef DirName) {
274 llvm::VersionTuple V;
275 std::string VerStr = DirName.drop_front(strlen("rocm-")).str();
276 // The ROCm directory name follows the format of
277 // rocm-{major}.{minor}.{subMinor}[-{build}]
278 llvm::replace(VerStr, '-', '.');
279 V.tryParse(VerStr);
280 return V;
281 };
282 for (llvm::vfs::directory_iterator
283 File = D.getVFS().dir_begin(D.SysRoot + "/opt", EC),
284 FileEnd;
285 File != FileEnd && !EC; File.increment(EC)) {
286 llvm::StringRef FileName = llvm::sys::path::filename(File->path());
287 if (!FileName.starts_with("rocm-"))
288 continue;
289 if (LatestROCm.empty()) {
290 LatestROCm = FileName.str();
291 LatestVer = GetROCmVersion(LatestROCm);
292 continue;
293 }
294 auto Ver = GetROCmVersion(FileName);
295 if (LatestVer < Ver) {
296 LatestROCm = FileName.str();
297 LatestVer = Ver;
298 }
299 }
300 if (!LatestROCm.empty())
301 ROCmSearchDirs.emplace_back(D.SysRoot + "/opt/" + LatestROCm,
302 /*StrictChecking=*/true);
303
304 ROCmSearchDirs.emplace_back(D.SysRoot + "/usr/local",
305 /*StrictChecking=*/true);
306 ROCmSearchDirs.emplace_back(D.SysRoot + "/usr",
307 /*StrictChecking=*/true);
308
309 DoPrintROCmSearchDirs();
310 return ROCmSearchDirs;
311}
312
314 const Driver &D, const llvm::Triple &HostTriple,
315 const llvm::opt::ArgList &Args, bool DetectHIPRuntime)
316 : D(D) {
317 Verbose = Args.hasArg(options::OPT_v);
318 RocmPathArg = Args.getLastArgValue(options::OPT_rocm_path_EQ);
319 PrintROCmSearchDirs = Args.hasArg(options::OPT_print_rocm_search_dirs);
320 RocmDeviceLibPathArg =
321 Args.getAllArgValues(options::OPT_rocm_device_lib_path_EQ);
322 HIPPathArg = Args.getLastArgValue(options::OPT_hip_path_EQ);
323 HIPStdParPathArg = Args.getLastArgValue(options::OPT_hipstdpar_path_EQ);
324 HasHIPStdParLibrary =
325 !HIPStdParPathArg.empty() && D.getVFS().exists(HIPStdParPathArg +
326 "/hipstdpar_lib.hpp");
327 HIPRocThrustPathArg =
328 Args.getLastArgValue(options::OPT_hipstdpar_thrust_path_EQ);
329 HasRocThrustLibrary = !HIPRocThrustPathArg.empty() &&
330 D.getVFS().exists(HIPRocThrustPathArg + "/thrust");
331 HIPRocPrimPathArg = Args.getLastArgValue(options::OPT_hipstdpar_prim_path_EQ);
332 HasRocPrimLibrary = !HIPRocPrimPathArg.empty() &&
333 D.getVFS().exists(HIPRocPrimPathArg + "/rocprim");
334
335 if (auto *A = Args.getLastArg(options::OPT_hip_version_EQ)) {
336 HIPVersionArg = A->getValue();
337 unsigned Major = ~0U;
338 unsigned Minor = ~0U;
339 SmallVector<StringRef, 3> Parts;
340 HIPVersionArg.split(Parts, '.');
341 if (!Parts.empty())
342 Parts[0].getAsInteger(0, Major);
343 if (Parts.size() > 1)
344 Parts[1].getAsInteger(0, Minor);
345 if (Parts.size() > 2)
346 VersionPatch = Parts[2].str();
347 if (VersionPatch.empty())
348 VersionPatch = "0";
349 if (Major != ~0U && Minor == ~0U)
350 Minor = 0;
351 if (Major == ~0U || Minor == ~0U)
352 D.Diag(diag::err_drv_invalid_value)
353 << A->getAsString(Args) << HIPVersionArg;
354
355 VersionMajorMinor = llvm::VersionTuple(Major, Minor);
356 DetectedVersion =
357 (Twine(Major) + "." + Twine(Minor) + "." + VersionPatch).str();
358 } else {
359 VersionPatch = DefaultVersionPatch;
360 VersionMajorMinor =
361 llvm::VersionTuple(DefaultVersionMajor, DefaultVersionMinor);
362 DetectedVersion = (Twine(DefaultVersionMajor) + "." +
363 Twine(DefaultVersionMinor) + "." + VersionPatch)
364 .str();
365 }
366
367 if (DetectHIPRuntime)
369}
370
372 assert(LibDevicePath.empty());
373
374 if (!RocmDeviceLibPathArg.empty())
375 LibDevicePath = RocmDeviceLibPathArg.back();
376 else if (std::optional<std::string> LibPathEnv =
377 llvm::sys::Process::GetEnv("HIP_DEVICE_LIB_PATH"))
378 LibDevicePath = std::move(*LibPathEnv);
379
380 auto &FS = D.getVFS();
381 if (!LibDevicePath.empty()) {
382 // Maintain compatability with HIP flag/envvar pointing directly at the
383 // bitcode library directory. This points directly at the library path instead
384 // of the rocm root installation.
385 if (!FS.exists(LibDevicePath))
386 return;
387
388 scanLibDevicePath(LibDevicePath);
389 HasDeviceLibrary = allGenericLibsValid() && !LibDeviceMap.empty();
390 return;
391 }
392
393 // Check device library exists at the given path.
394 auto CheckDeviceLib = [&](StringRef Path, bool StrictChecking) {
395 bool CheckLibDevice = (!NoBuiltinLibs || StrictChecking);
396 if (CheckLibDevice && !FS.exists(Path))
397 return false;
398
399 scanLibDevicePath(Path);
400
401 if (!NoBuiltinLibs) {
402 // Check that the required non-target libraries are all available.
403 if (!allGenericLibsValid())
404 return false;
405
406 // Check that we have found at least one libdevice that we can link in
407 // if -nobuiltinlib hasn't been specified.
408 if (LibDeviceMap.empty())
409 return false;
410 }
411 return true;
412 };
413
414 // Find device libraries in <LLVM_DIR>/lib/clang/<ver>/lib/amdgcn/bitcode
415 LibDevicePath = D.ResourceDir;
416 llvm::sys::path::append(LibDevicePath, CLANG_INSTALL_LIBDIR_BASENAME,
417 "amdgcn", "bitcode");
418 HasDeviceLibrary = CheckDeviceLib(LibDevicePath, true);
419 if (HasDeviceLibrary)
420 return;
421
422 // Find device libraries in a legacy ROCm directory structure
423 // ${ROCM_ROOT}/amdgcn/bitcode/*
424 auto &ROCmDirs = getInstallationPathCandidates();
425 for (const auto &Candidate : ROCmDirs) {
426 LibDevicePath = Candidate.Path;
427 llvm::sys::path::append(LibDevicePath, "amdgcn", "bitcode");
428 HasDeviceLibrary = CheckDeviceLib(LibDevicePath, Candidate.StrictChecking);
429 if (HasDeviceLibrary)
430 return;
431 }
432}
433
435 SmallVector<Candidate, 4> HIPSearchDirs;
436 if (!HIPPathArg.empty())
437 HIPSearchDirs.emplace_back(HIPPathArg.str());
438 else if (std::optional<std::string> HIPPathEnv =
439 llvm::sys::Process::GetEnv("HIP_PATH")) {
440 if (!HIPPathEnv->empty())
441 HIPSearchDirs.emplace_back(std::move(*HIPPathEnv));
442 }
443 if (HIPSearchDirs.empty())
444 HIPSearchDirs.append(getInstallationPathCandidates());
445 auto &FS = D.getVFS();
446
447 for (const auto &Candidate : HIPSearchDirs) {
448 InstallPath = Candidate.Path;
449 if (InstallPath.empty() || !FS.exists(InstallPath))
450 continue;
451
452 BinPath = InstallPath;
453 llvm::sys::path::append(BinPath, "bin");
454 IncludePath = InstallPath;
455 llvm::sys::path::append(IncludePath, "include");
456 LibPath = InstallPath;
457 llvm::sys::path::append(LibPath, "lib");
458 SharePath = InstallPath;
459 llvm::sys::path::append(SharePath, "share");
460
461 // Get parent of InstallPath and append "share"
462 SmallString<0> ParentSharePath = llvm::sys::path::parent_path(InstallPath);
463 llvm::sys::path::append(ParentSharePath, "share");
464
465 auto Append = [](SmallString<0> &path, const Twine &a, const Twine &b = "",
466 const Twine &c = "", const Twine &d = "") {
467 SmallString<0> newpath = path;
468 llvm::sys::path::append(newpath, a, b, c, d);
469 return newpath;
470 };
471 // If HIP version file can be found and parsed, use HIP version from there.
472 std::vector<SmallString<0>> VersionFilePaths = {
473 Append(SharePath, "hip", "version"),
474 InstallPath != D.SysRoot + "/usr/local"
475 ? Append(ParentSharePath, "hip", "version")
476 : SmallString<0>(),
477 Append(BinPath, ".hipVersion")};
478
479 for (const auto &VersionFilePath : VersionFilePaths) {
480 if (VersionFilePath.empty())
481 continue;
482 llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> VersionFile =
483 FS.getBufferForFile(VersionFilePath);
484 if (!VersionFile)
485 continue;
486 if (HIPVersionArg.empty() && VersionFile)
487 if (parseHIPVersionFile((*VersionFile)->getBuffer()))
488 continue;
489
490 HasHIPRuntime = true;
491 return;
492 }
493 // Otherwise, if -rocm-path is specified (no strict checking), use the
494 // default HIP version or specified by --hip-version.
495 if (!Candidate.StrictChecking) {
496 HasHIPRuntime = true;
497 return;
498 }
499 }
500 HasHIPRuntime = false;
501}
502
503void RocmInstallationDetector::print(raw_ostream &OS) const {
504 if (hasHIPRuntime())
505 OS << "Found HIP installation: " << InstallPath << ", version "
506 << DetectedVersion << '\n';
507}
508
509void RocmInstallationDetector::AddHIPIncludeArgs(const ArgList &DriverArgs,
510 ArgStringList &CC1Args) const {
511 bool UsesRuntimeWrapper = VersionMajorMinor > llvm::VersionTuple(3, 5) &&
512 !DriverArgs.hasArg(options::OPT_nohipwrapperinc);
513 bool HasHipStdPar = DriverArgs.hasArg(options::OPT_hipstdpar);
514
515 if (!DriverArgs.hasArg(options::OPT_nobuiltininc)) {
516 // HIP header includes standard library wrapper headers under clang
517 // cuda_wrappers directory. Since these wrapper headers include_next
518 // standard C++ headers, whereas libc++ headers include_next other clang
519 // headers. The include paths have to follow this order:
520 // - wrapper include path
521 // - standard C++ include path
522 // - other clang include path
523 // Since standard C++ and other clang include paths are added in other
524 // places after this function, here we only need to make sure wrapper
525 // include path is added.
526 //
527 // ROCm 3.5 does not fully support the wrapper headers. Therefore it needs
528 // a workaround.
529 SmallString<128> P(D.ResourceDir);
530 if (UsesRuntimeWrapper)
531 llvm::sys::path::append(P, "include", "cuda_wrappers");
532 CC1Args.push_back("-internal-isystem");
533 CC1Args.push_back(DriverArgs.MakeArgString(P));
534 }
535
536 const auto HandleHipStdPar = [=, &DriverArgs, &CC1Args]() {
537 StringRef Inc = getIncludePath();
538 auto &FS = D.getVFS();
539
540 if (!hasHIPStdParLibrary())
541 if (!HIPStdParPathArg.empty() ||
542 !FS.exists(Inc + "/thrust/system/hip/hipstdpar/hipstdpar_lib.hpp")) {
543 D.Diag(diag::err_drv_no_hipstdpar_lib);
544 return;
545 }
546 if (!HasRocThrustLibrary && !FS.exists(Inc + "/thrust")) {
547 D.Diag(diag::err_drv_no_hipstdpar_thrust_lib);
548 return;
549 }
550 if (!HasRocPrimLibrary && !FS.exists(Inc + "/rocprim")) {
551 D.Diag(diag::err_drv_no_hipstdpar_prim_lib);
552 return;
553 }
554 const char *ThrustPath;
555 if (HasRocThrustLibrary)
556 ThrustPath = DriverArgs.MakeArgString(HIPRocThrustPathArg);
557 else
558 ThrustPath = DriverArgs.MakeArgString(Inc + "/thrust");
559
560 const char *HIPStdParPath;
562 HIPStdParPath = DriverArgs.MakeArgString(HIPStdParPathArg);
563 else
564 HIPStdParPath = DriverArgs.MakeArgString(StringRef(ThrustPath) +
565 "/system/hip/hipstdpar");
566
567 const char *PrimPath;
568 if (HasRocPrimLibrary)
569 PrimPath = DriverArgs.MakeArgString(HIPRocPrimPathArg);
570 else
571 PrimPath = DriverArgs.MakeArgString(getIncludePath() + "/rocprim");
572
573 CC1Args.append({"-idirafter", ThrustPath, "-idirafter", PrimPath,
574 "-idirafter", HIPStdParPath, "-include",
575 "hipstdpar_lib.hpp"});
576 };
577
578 if (!DriverArgs.hasFlag(options::OPT_offload_inc, options::OPT_no_offload_inc,
579 true)) {
580 if (HasHipStdPar)
581 HandleHipStdPar();
582
583 return;
584 }
585
586 if (!hasHIPRuntime()) {
587 D.Diag(diag::err_drv_no_hip_runtime);
588 return;
589 }
590
591 CC1Args.push_back("-idirafter");
592 CC1Args.push_back(DriverArgs.MakeArgString(getIncludePath()));
593 if (UsesRuntimeWrapper)
594 CC1Args.append({"-include", "__clang_hip_runtime_wrapper.h"});
595 if (HasHipStdPar)
596 HandleHipStdPar();
597}
598
600 const InputInfo &Output,
601 const InputInfoList &Inputs,
602 const ArgList &Args,
603 const char *LinkingOutput) const {
604 std::string Linker = getToolChain().GetLinkerPath();
605 ArgStringList CmdArgs;
606 if (!Args.hasArg(options::OPT_r)) {
607 CmdArgs.push_back("--no-undefined");
608 CmdArgs.push_back("-shared");
609 }
610
611 if (C.getDriver().isUsingLTO()) {
612 const bool ThinLTO = (C.getDriver().getLTOMode() == LTOK_Thin);
613 addLTOOptions(getToolChain(), Args, CmdArgs, Output, Inputs, ThinLTO);
614 } else if (Args.hasArg(options::OPT_mcpu_EQ)) {
615 CmdArgs.push_back(Args.MakeArgString(
616 "-plugin-opt=mcpu=" +
618 Args.getLastArgValue(options::OPT_mcpu_EQ))));
619 }
621 getToolChain().AddFilePathLibArgs(Args, CmdArgs);
622 Args.AddAllArgs(CmdArgs, options::OPT_L);
623 AddLinkerInputs(getToolChain(), Inputs, Args, CmdArgs, JA);
624
625 // Always pass the target-id features to the LTO job.
626 std::vector<StringRef> Features;
627 getAMDGPUTargetFeatures(C.getDriver(), getToolChain().getTriple(), Args,
628 Features);
629 if (!Features.empty()) {
630 CmdArgs.push_back(
631 Args.MakeArgString("-plugin-opt=-mattr=" + llvm::join(Features, ",")));
632 }
633
634 getToolChain().addProfileRTLibs(Args, CmdArgs);
635
636 if (Args.hasArg(options::OPT_stdlib))
637 CmdArgs.append({"-lc", "-lm"});
638 if (Args.hasArg(options::OPT_startfiles)) {
639 std::optional<std::string> IncludePath = getToolChain().getStdlibPath();
640 if (!IncludePath)
641 IncludePath = "/lib";
642 SmallString<128> P(*IncludePath);
643 llvm::sys::path::append(P, "crt1.o");
644 CmdArgs.push_back(Args.MakeArgString(P));
645 }
646
647 CmdArgs.push_back("-o");
648 CmdArgs.push_back(Output.getFilename());
649 C.addCommand(std::make_unique<Command>(
650 JA, *this, ResponseFileSupport::AtFileCurCP(), Args.MakeArgString(Linker),
651 CmdArgs, Inputs, Output));
652}
653
655 const llvm::Triple &Triple,
656 const llvm::opt::ArgList &Args,
657 std::vector<StringRef> &Features) {
658 // Add target ID features to -target-feature options. No diagnostics should
659 // be emitted here since invalid target ID is diagnosed at other places.
660 StringRef TargetID;
661 if (Args.hasArg(options::OPT_mcpu_EQ))
662 TargetID = Args.getLastArgValue(options::OPT_mcpu_EQ);
663 else if (Args.hasArg(options::OPT_march_EQ))
664 TargetID = Args.getLastArgValue(options::OPT_march_EQ);
665 if (!TargetID.empty()) {
666 llvm::StringMap<bool> FeatureMap;
667 auto OptionalGpuArch = parseTargetID(Triple, TargetID, &FeatureMap);
668 if (OptionalGpuArch) {
669 StringRef GpuArch = *OptionalGpuArch;
670 // Iterate through all possible target ID features for the given GPU.
671 // If it is mapped to true, add +feature.
672 // If it is mapped to false, add -feature.
673 // If it is not in the map (default), do not add it
674 for (auto &&Feature : getAllPossibleTargetIDFeatures(Triple, GpuArch)) {
675 auto Pos = FeatureMap.find(Feature);
676 if (Pos == FeatureMap.end())
677 continue;
678 Features.push_back(Args.MakeArgStringRef(
679 (Twine(Pos->second ? "+" : "-") + Feature).str()));
680 }
681 }
682 }
683
684 if (Args.hasFlag(options::OPT_mwavefrontsize64,
685 options::OPT_mno_wavefrontsize64, false))
686 Features.push_back("+wavefrontsize64");
687
688 if (Args.hasFlag(options::OPT_mamdgpu_precise_memory_op,
689 options::OPT_mno_amdgpu_precise_memory_op, false))
690 Features.push_back("+precise-memory");
691
692 handleTargetFeaturesGroup(D, Triple, Args, Features,
693 options::OPT_m_amdgpu_Features_Group);
694}
695
696/// AMDGPU Toolchain
697AMDGPUToolChain::AMDGPUToolChain(const Driver &D, const llvm::Triple &Triple,
698 const ArgList &Args)
699 : Generic_ELF(D, Triple, Args),
701 {{options::OPT_O, "3"}, {options::OPT_cl_std_EQ, "CL1.2"}}) {
702 // Check code object version options. Emit warnings for legacy options
703 // and errors for the last invalid code object version options.
704 // It is done here to avoid repeated warning or error messages for
705 // each tool invocation.
707}
708
710 return new tools::amdgpu::Linker(*this);
711}
712
713DerivedArgList *
714AMDGPUToolChain::TranslateArgs(const DerivedArgList &Args, StringRef BoundArch,
715 Action::OffloadKind DeviceOffloadKind) const {
716
717 DerivedArgList *DAL =
718 Generic_ELF::TranslateArgs(Args, BoundArch, DeviceOffloadKind);
719
720 const OptTable &Opts = getDriver().getOpts();
721
722 if (!DAL)
723 DAL = new DerivedArgList(Args.getBaseArgs());
724
725 for (Arg *A : Args)
726 DAL->append(A);
727
728 // Replace -mcpu=native with detected GPU.
729 Arg *LastMCPUArg = DAL->getLastArg(options::OPT_mcpu_EQ);
730 if (LastMCPUArg && StringRef(LastMCPUArg->getValue()) == "native") {
731 DAL->eraseArg(options::OPT_mcpu_EQ);
732 auto GPUsOrErr = getSystemGPUArchs(Args);
733 if (!GPUsOrErr) {
734 getDriver().Diag(diag::err_drv_undetermined_gpu_arch)
735 << llvm::Triple::getArchTypeName(getArch())
736 << llvm::toString(GPUsOrErr.takeError()) << "-mcpu";
737 } else {
738 auto &GPUs = *GPUsOrErr;
739 if (GPUs.size() > 1) {
740 getDriver().Diag(diag::warn_drv_multi_gpu_arch)
741 << llvm::Triple::getArchTypeName(getArch())
742 << llvm::join(GPUs, ", ") << "-mcpu";
743 }
744 DAL->AddJoinedArg(nullptr, Opts.getOption(options::OPT_mcpu_EQ),
745 Args.MakeArgString(GPUs.front()));
746 }
747 }
748
749 checkTargetID(*DAL);
750
751 if (Args.getLastArgValue(options::OPT_x) != "cl")
752 return DAL;
753
754 // Phase 1 (.cl -> .bc)
755 if (Args.hasArg(options::OPT_c) && Args.hasArg(options::OPT_emit_llvm)) {
756 DAL->AddFlagArg(nullptr, Opts.getOption(getTriple().isArch64Bit()
757 ? options::OPT_m64
758 : options::OPT_m32));
759
760 // Have to check OPT_O4, OPT_O0 & OPT_Ofast separately
761 // as they defined that way in Options.td
762 if (!Args.hasArg(options::OPT_O, options::OPT_O0, options::OPT_O4,
763 options::OPT_Ofast))
764 DAL->AddJoinedArg(nullptr, Opts.getOption(options::OPT_O),
765 getOptionDefault(options::OPT_O));
766 }
767
768 return DAL;
769}
770
772 llvm::AMDGPU::GPUKind Kind) {
773
774 // Assume nothing without a specific target.
775 if (Kind == llvm::AMDGPU::GK_NONE)
776 return false;
777
778 const unsigned ArchAttr = llvm::AMDGPU::getArchAttrAMDGCN(Kind);
779
780 // Default to enabling f32 denormals by default on subtargets where fma is
781 // fast with denormals
782 const bool BothDenormAndFMAFast =
783 (ArchAttr & llvm::AMDGPU::FEATURE_FAST_FMA_F32) &&
784 (ArchAttr & llvm::AMDGPU::FEATURE_FAST_DENORMAL_F32);
785 return !BothDenormAndFMAFast;
786}
787
789 const llvm::opt::ArgList &DriverArgs, const JobAction &JA,
790 const llvm::fltSemantics *FPType) const {
791 // Denormals should always be enabled for f16 and f64.
792 if (!FPType || FPType != &llvm::APFloat::IEEEsingle())
793 return llvm::DenormalMode::getIEEE();
794
798 auto Kind = llvm::AMDGPU::parseArchAMDGCN(Arch);
799 if (FPType && FPType == &llvm::APFloat::IEEEsingle() &&
800 DriverArgs.hasFlag(options::OPT_fgpu_flush_denormals_to_zero,
801 options::OPT_fno_gpu_flush_denormals_to_zero,
803 return llvm::DenormalMode::getPreserveSign();
804
805 return llvm::DenormalMode::getIEEE();
806 }
807
808 const StringRef GpuArch = getGPUArch(DriverArgs);
809 auto Kind = llvm::AMDGPU::parseArchAMDGCN(GpuArch);
810
811 // TODO: There are way too many flags that change this. Do we need to check
812 // them all?
813 bool DAZ = DriverArgs.hasArg(options::OPT_cl_denorms_are_zero) ||
815
816 // Outputs are flushed to zero (FTZ), preserving sign. Denormal inputs are
817 // also implicit treated as zero (DAZ).
818 return DAZ ? llvm::DenormalMode::getPreserveSign() :
819 llvm::DenormalMode::getIEEE();
820}
821
822bool AMDGPUToolChain::isWave64(const llvm::opt::ArgList &DriverArgs,
823 llvm::AMDGPU::GPUKind Kind) {
824 const unsigned ArchAttr = llvm::AMDGPU::getArchAttrAMDGCN(Kind);
825 bool HasWave32 = (ArchAttr & llvm::AMDGPU::FEATURE_WAVE32);
826
827 return !HasWave32 || DriverArgs.hasFlag(
828 options::OPT_mwavefrontsize64, options::OPT_mno_wavefrontsize64, false);
829}
830
831
832/// ROCM Toolchain
833ROCMToolChain::ROCMToolChain(const Driver &D, const llvm::Triple &Triple,
834 const ArgList &Args)
835 : AMDGPUToolChain(D, Triple, Args) {
836 if (Triple.getEnvironment() != llvm::Triple::LLVM)
837 RocmInstallation->detectDeviceLibrary();
838}
839
841 const llvm::opt::ArgList &DriverArgs,
842 llvm::opt::ArgStringList &CC1Args,
843 Action::OffloadKind DeviceOffloadingKind) const {
844 // Default to "hidden" visibility, as object level linking will not be
845 // supported for the foreseeable future.
846 if (!DriverArgs.hasArg(options::OPT_fvisibility_EQ,
847 options::OPT_fvisibility_ms_compat)) {
848 CC1Args.push_back("-fvisibility=hidden");
849 CC1Args.push_back("-fapply-global-visibility-to-externs");
850 }
851
852 // For SPIR-V we want to retain the pristine output of Clang CodeGen, since
853 // optimizations might lose structure / information that is necessary for
854 // generating optimal concrete AMDGPU code.
855 // TODO: using the below option is a temporary placeholder until Clang
856 // provides the required functionality, which essentially boils down to
857 // -O0 being refactored / reworked to not imply optnone / remove TBAA.
858 // Once that is added, we should pivot to that functionality, being
859 // mindful to not corrupt the user provided and subsequently embedded
860 // command-line (i.e. if the user asks for -O3 this is what the
861 // finalisation should use).
862 if (getTriple().isSPIRV() &&
863 !DriverArgs.hasArg(options::OPT_disable_llvm_optzns))
864 CC1Args.push_back("-disable-llvm-optzns");
865
866 if (DeviceOffloadingKind == Action::OFK_None)
867 addOpenCLBuiltinsLib(getDriver(), getTriple(), DriverArgs, CC1Args);
868}
869
870void AMDGPUToolChain::addClangWarningOptions(ArgStringList &CC1Args) const {
871 // AMDGPU does not support atomic lib call. Treat atomic alignment
872 // warnings as errors.
873 CC1Args.push_back("-Werror=atomic-alignment");
874}
875
876void AMDGPUToolChain::AddClangSystemIncludeArgs(const ArgList &DriverArgs,
877 ArgStringList &CC1Args) const {
878 if (DriverArgs.hasArg(options::OPT_nostdinc) ||
879 DriverArgs.hasArg(options::OPT_nostdlibinc))
880 return;
881
882 if (std::optional<std::string> Path = getStdlibIncludePath())
883 addSystemInclude(DriverArgs, CC1Args, *Path);
884}
885
886StringRef
887AMDGPUToolChain::getGPUArch(const llvm::opt::ArgList &DriverArgs) const {
889 getTriple(), DriverArgs.getLastArgValue(options::OPT_mcpu_EQ));
890}
891
893AMDGPUToolChain::getParsedTargetID(const llvm::opt::ArgList &DriverArgs) const {
894 StringRef TargetID = DriverArgs.getLastArgValue(options::OPT_mcpu_EQ);
895 if (TargetID.empty())
896 return {std::nullopt, std::nullopt, std::nullopt};
897
898 llvm::StringMap<bool> FeatureMap;
899 auto OptionalGpuArch = parseTargetID(getTriple(), TargetID, &FeatureMap);
900 if (!OptionalGpuArch)
901 return {TargetID.str(), std::nullopt, std::nullopt};
902
903 return {TargetID.str(), OptionalGpuArch->str(), FeatureMap};
904}
905
907 const llvm::opt::ArgList &DriverArgs) const {
908 auto PTID = getParsedTargetID(DriverArgs);
909 if (PTID.OptionalTargetID && !PTID.OptionalGPUArch) {
910 getDriver().Diag(clang::diag::err_drv_bad_target_id)
911 << *PTID.OptionalTargetID;
912 }
913}
914
916AMDGPUToolChain::getSystemGPUArchs(const ArgList &Args) const {
917 // Detect AMD GPUs availible on the system.
918 std::string Program;
919 if (Arg *A = Args.getLastArg(options::OPT_offload_arch_tool_EQ))
920 Program = A->getValue();
921 else
922 Program = GetProgramPath("amdgpu-arch");
923
924 auto StdoutOrErr = getDriver().executeProgram({Program});
925 if (!StdoutOrErr)
926 return StdoutOrErr.takeError();
927
929 for (StringRef Arch : llvm::split((*StdoutOrErr)->getBuffer(), "\n"))
930 if (!Arch.empty())
931 GPUArchs.push_back(Arch.str());
932
933 if (GPUArchs.empty())
934 return llvm::createStringError(std::error_code(),
935 "No AMD GPU detected in the system");
936
937 return std::move(GPUArchs);
938}
939
941 const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args,
942 Action::OffloadKind DeviceOffloadingKind) const {
943 AMDGPUToolChain::addClangTargetOptions(DriverArgs, CC1Args,
944 DeviceOffloadingKind);
945
946 // For the OpenCL case where there is no offload target, accept -nostdlib to
947 // disable bitcode linking.
948 if (DeviceOffloadingKind == Action::OFK_None &&
949 DriverArgs.hasArg(options::OPT_nostdlib))
950 return;
951
952 if (!DriverArgs.hasFlag(options::OPT_offloadlib, options::OPT_no_offloadlib,
953 true))
954 return;
955
956 // For SPIR-V (SPIRVAMDToolChain) we must not link any device libraries so we
957 // skip it.
958 const llvm::Triple &TT = this->getEffectiveTriple();
959 if (TT.isSPIRV())
960 return;
961
962 // With an LLVM environment, only use libraries provided by the resource
963 // directory.
964 if (TT.getEnvironment() == llvm::Triple::LLVM)
965 return;
966
967 // Get the device name and canonicalize it
968 const StringRef GpuArch = getGPUArch(DriverArgs);
969 auto Kind = llvm::AMDGPU::parseArchAMDGCN(GpuArch);
970 const StringRef CanonArch = llvm::AMDGPU::getArchNameAMDGCN(Kind);
971 StringRef LibDeviceFile = RocmInstallation->getLibDeviceFile(CanonArch);
974 if (!RocmInstallation->checkCommonBitcodeLibs(CanonArch, LibDeviceFile,
975 ABIVer))
976 return;
977
978 // Add the OpenCL specific bitcode library.
980 BCLibs.emplace_back(RocmInstallation->getOpenCLPath().str());
981
982 // Add the generic set of libraries.
983 BCLibs.append(RocmInstallation->getCommonBitcodeLibs(
984 DriverArgs, LibDeviceFile, GpuArch, DeviceOffloadingKind,
985 getSanitizerArgs(DriverArgs).needsAsanRt()));
986
987 for (auto [BCFile, Internalize] : BCLibs) {
988 if (Internalize)
989 CC1Args.push_back("-mlink-builtin-bitcode");
990 else
991 CC1Args.push_back("-mlink-bitcode-file");
992 CC1Args.push_back(DriverArgs.MakeArgString(BCFile));
993 }
994}
995
997 StringRef GPUArch, StringRef LibDeviceFile,
998 DeviceLibABIVersion ABIVer) const {
999 if (!hasDeviceLibrary()) {
1000 D.Diag(diag::err_drv_no_rocm_device_lib) << 0;
1001 return false;
1002 }
1003 if (LibDeviceFile.empty()) {
1004 D.Diag(diag::err_drv_no_rocm_device_lib) << 1 << GPUArch;
1005 return false;
1006 }
1007 if (ABIVer.requiresLibrary() && getABIVersionPath(ABIVer).empty()) {
1008 // Starting from COV6, we will report minimum ROCm version requirement in
1009 // the error message.
1010 if (ABIVer.getAsCodeObjectVersion() < 6)
1011 D.Diag(diag::err_drv_no_rocm_device_lib) << 2 << ABIVer.toString() << 0;
1012 else
1013 D.Diag(diag::err_drv_no_rocm_device_lib)
1014 << 2 << ABIVer.toString() << 1 << "6.3";
1015 return false;
1016 }
1017 return true;
1018}
1019
1022 const llvm::opt::ArgList &DriverArgs, StringRef LibDeviceFile,
1023 StringRef GPUArch, const Action::OffloadKind DeviceOffloadingKind,
1024 const bool NeedsASanRT) const {
1026
1027 CommonBitcodeLibsPreferences Pref{D, DriverArgs, GPUArch,
1028 DeviceOffloadingKind, NeedsASanRT};
1029
1030 auto AddBCLib = [&](ToolChain::BitCodeLibraryInfo BCLib,
1031 bool Internalize = true) {
1032 if (!BCLib.Path.empty()) {
1033 BCLib.ShouldInternalize = Internalize;
1034 BCLibs.emplace_back(BCLib);
1035 }
1036 };
1037 auto AddSanBCLibs = [&]() {
1038 if (Pref.GPUSan)
1039 AddBCLib(getAsanRTLPath(), false);
1040 };
1041
1042 AddSanBCLibs();
1043 AddBCLib(getOCMLPath());
1044 if (!Pref.IsOpenMP)
1045 AddBCLib(getOCKLPath());
1046 else if (Pref.GPUSan && Pref.IsOpenMP)
1047 AddBCLib(getOCKLPath());
1048 AddBCLib(getUnsafeMathPath(Pref.UnsafeMathOpt || Pref.FastRelaxedMath));
1049 AddBCLib(getFiniteOnlyPath(Pref.FiniteOnly || Pref.FastRelaxedMath));
1050 AddBCLib(getWavefrontSize64Path(Pref.Wave64));
1051 AddBCLib(LibDeviceFile);
1052 auto ABIVerPath = getABIVersionPath(Pref.ABIVer);
1053 if (!ABIVerPath.empty())
1054 AddBCLib(ABIVerPath);
1055
1056 return BCLibs;
1057}
1058
1061 const llvm::opt::ArgList &DriverArgs, const std::string &GPUArch,
1062 Action::OffloadKind DeviceOffloadingKind) const {
1063 auto Kind = llvm::AMDGPU::parseArchAMDGCN(GPUArch);
1064 const StringRef CanonArch = llvm::AMDGPU::getArchNameAMDGCN(Kind);
1065
1066 StringRef LibDeviceFile = RocmInstallation->getLibDeviceFile(CanonArch);
1068 getAMDGPUCodeObjectVersion(getDriver(), DriverArgs));
1069 if (!RocmInstallation->checkCommonBitcodeLibs(CanonArch, LibDeviceFile,
1070 ABIVer))
1071 return {};
1072
1073 return RocmInstallation->getCommonBitcodeLibs(
1074 DriverArgs, LibDeviceFile, GPUArch, DeviceOffloadingKind,
1075 getSanitizerArgs(DriverArgs).needsAsanRt());
1076}
1077
1079 const ToolChain &TC, const llvm::opt::ArgList &DriverArgs,
1080 StringRef TargetID, const llvm::opt::Arg *A) const {
1081 auto &Diags = TC.getDriver().getDiags();
1082 bool IsExplicitDevice =
1083 A->getBaseArg().getOption().matches(options::OPT_Xarch_device);
1084
1085 // Check 'xnack+' availability by default
1086 llvm::StringRef Processor =
1087 getProcessorFromTargetID(TC.getTriple(), TargetID);
1088 auto ProcKind = TC.getTriple().isAMDGCN()
1089 ? llvm::AMDGPU::parseArchAMDGCN(Processor)
1090 : llvm::AMDGPU::parseArchR600(Processor);
1091 auto Features = TC.getTriple().isAMDGCN()
1092 ? llvm::AMDGPU::getArchAttrAMDGCN(ProcKind)
1093 : llvm::AMDGPU::getArchAttrR600(ProcKind);
1094 if (Features & llvm::AMDGPU::FEATURE_XNACK_ALWAYS)
1095 return false;
1096
1097 // Look for the xnack feature in TargetID
1098 llvm::StringMap<bool> FeatureMap;
1099 auto OptionalGpuArch = parseTargetID(TC.getTriple(), TargetID, &FeatureMap);
1100 assert(OptionalGpuArch && "Invalid Target ID");
1101 (void)OptionalGpuArch;
1102 auto Loc = FeatureMap.find("xnack");
1103 if (Loc == FeatureMap.end() || !Loc->second) {
1104 if (IsExplicitDevice) {
1105 Diags.Report(
1106 clang::diag::err_drv_unsupported_option_for_offload_arch_req_feature)
1107 << A->getAsString(DriverArgs) << TargetID << "xnack+";
1108 } else {
1109 Diags.Report(
1110 clang::diag::warn_drv_unsupported_option_for_offload_arch_req_feature)
1111 << A->getAsString(DriverArgs) << TargetID << "xnack+";
1112 }
1113 return true;
1114 }
1115
1116 return false;
1117}
#define V(N, I)
static StringRef getTriple(const Command &Job)
static void Append(char *Start, char *End, char *&Buffer, unsigned &BufferSize, unsigned &BufferCapacity)
__device__ __2f16 b
__device__ __2f16 float c
const char * getOffloadingArch() const
Definition Action.h:213
OffloadKind getOffloadingDeviceKind() const
Definition Action.h:212
Compilation - A set of tasks to perform for a single driver invocation.
Definition Compilation.h:45
Driver - Encapsulate logic for constructing compilation processes from a set of gcc-driver-like comma...
Definition Driver.h:99
DiagnosticsEngine & getDiags() const
Definition Driver.h:419
llvm::Expected< std::unique_ptr< llvm::MemoryBuffer > > executeProgram(llvm::ArrayRef< llvm::StringRef > Args) const
Definition Driver.cpp:392
DiagnosticBuilder Diag(unsigned DiagID) const
Definition Driver.h:169
const llvm::opt::OptTable & getOpts() const
Definition Driver.h:417
InputInfo - Wrapper for information about an input source.
Definition InputInfo.h:22
const char * getFilename() const
Definition InputInfo.h:83
StringRef getIncludePath() const
Get the detected path to Rocm's bin directory.
StringRef getAsanRTLPath() const
Returns empty string of Asan runtime library is not available.
RocmInstallationDetector(const Driver &D, const llvm::Triple &HostTriple, const llvm::opt::ArgList &Args, bool DetectHIPRuntime=true)
Definition AMDGPU.cpp:313
bool hasDeviceLibrary() const
Check whether we detected a valid ROCm device library.
bool checkCommonBitcodeLibs(StringRef GPUArch, StringRef LibDeviceFile, DeviceLibABIVersion ABIVer) const
Check file paths of default bitcode libraries common to AMDGPU based toolchains.
Definition AMDGPU.cpp:996
bool hasHIPStdParLibrary() const
Check whether we detected a valid HIP STDPAR Acceleration library.
StringRef getABIVersionPath(DeviceLibABIVersion ABIVer) const
llvm::SmallVector< ToolChain::BitCodeLibraryInfo, 12 > getCommonBitcodeLibs(const llvm::opt::ArgList &DriverArgs, StringRef LibDeviceFile, StringRef GPUArch, const Action::OffloadKind DeviceOffloadingKind, const bool NeedsASanRT) const
Get file paths of default bitcode libraries common to AMDGPU based toolchains.
Definition AMDGPU.cpp:1021
bool hasHIPRuntime() const
Check whether we detected a valid HIP runtime.
void AddHIPIncludeArgs(const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args) const
Definition AMDGPU.cpp:509
StringRef getWavefrontSize64Path(bool Enabled) const
void print(raw_ostream &OS) const
Print information about the detected ROCm installation.
Definition AMDGPU.cpp:503
static void addSystemInclude(const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args, const Twine &Path)
Utility function to add a system include directory to CC1 arguments.
llvm::Triple::ArchType getArch() const
Definition ToolChain.h:277
const Driver & getDriver() const
Definition ToolChain.h:261
ToolChain(const Driver &D, const llvm::Triple &T, const llvm::opt::ArgList &Args)
Definition ToolChain.cpp:89
virtual llvm::opt::DerivedArgList * TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch, Action::OffloadKind DeviceOffloadKind) const
TranslateArgs - Create a new derived argument list for any argument translations this ToolChain may w...
Definition ToolChain.h:367
const llvm::Triple & getEffectiveTriple() const
Get the toolchain's effective clang triple.
Definition ToolChain.h:291
const llvm::Triple & getTriple() const
Definition ToolChain.h:263
std::string GetProgramPath(const char *Name) const
std::optional< std::string > getStdlibIncludePath() const
SanitizerArgs getSanitizerArgs(const llvm::opt::ArgList &JobArgs) const
Tool - Information on a specific compilation tool.
Definition Tool.h:32
const ToolChain & getToolChain() const
Definition Tool.h:52
llvm::DenormalMode getDefaultDenormalModeForType(const llvm::opt::ArgList &DriverArgs, const JobAction &JA, const llvm::fltSemantics *FPType=nullptr) const override
Returns the output denormal handling type in the default floating point environment for the given FPT...
Definition AMDGPU.cpp:788
llvm::opt::DerivedArgList * TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch, Action::OffloadKind DeviceOffloadKind) const override
TranslateArgs - Create a new derived argument list for any argument translations this ToolChain may w...
Definition AMDGPU.cpp:714
static bool getDefaultDenormsAreZeroForTarget(llvm::AMDGPU::GPUKind GPUKind)
Return whether denormals should be flushed, and treated as 0 by default for the subtarget.
Definition AMDGPU.cpp:771
StringRef getGPUArch(const llvm::opt::ArgList &DriverArgs) const
Get GPU arch from -mcpu without checking.
Definition AMDGPU.cpp:887
bool shouldSkipSanitizeOption(const ToolChain &TC, const llvm::opt::ArgList &DriverArgs, StringRef TargetID, const llvm::opt::Arg *A) const
Should skip sanitize option.
Definition AMDGPU.cpp:1078
virtual void checkTargetID(const llvm::opt::ArgList &DriverArgs) const
Check and diagnose invalid target ID specified by -mcpu.
Definition AMDGPU.cpp:906
Tool * buildLinker() const override
Definition AMDGPU.cpp:709
static bool isWave64(const llvm::opt::ArgList &DriverArgs, llvm::AMDGPU::GPUKind Kind)
Definition AMDGPU.cpp:822
void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override
Common warning options shared by AMDGPU HIP, OpenCL and OpenMP toolchains.
Definition AMDGPU.cpp:870
AMDGPUToolChain(const Driver &D, const llvm::Triple &Triple, const llvm::opt::ArgList &Args)
AMDGPU Toolchain.
Definition AMDGPU.cpp:697
ParsedTargetIDType getParsedTargetID(const llvm::opt::ArgList &DriverArgs) const
Get target ID, GPU arch, and target ID features if the target ID is specified and valid.
Definition AMDGPU.cpp:893
const std::map< options::ID, const StringRef > OptionsDefault
Definition AMDGPU.h:52
void AddClangSystemIncludeArgs(const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args) const override
Add the clang cc1 arguments for system include paths.
Definition AMDGPU.cpp:876
StringRef getOptionDefault(options::ID OptID) const
Definition AMDGPU.h:55
void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args, Action::OffloadKind DeviceOffloadKind) const override
Add options that need to be passed to cc1 for this target.
Definition AMDGPU.cpp:840
virtual Expected< SmallVector< std::string > > getSystemGPUArchs(const llvm::opt::ArgList &Args) const override
Uses amdgpu-arch tool to get arch of the system GPU.
Definition AMDGPU.cpp:916
Generic_ELF(const Driver &D, const llvm::Triple &Triple, const llvm::opt::ArgList &Args)
Definition Gnu.h:439
LazyDetector< RocmInstallationDetector > RocmInstallation
Definition Gnu.h:359
void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args, Action::OffloadKind DeviceOffloadKind) const override
Add options that need to be passed to cc1 for this target.
Definition AMDGPU.cpp:940
llvm::SmallVector< BitCodeLibraryInfo, 12 > getCommonDeviceLibNames(const llvm::opt::ArgList &DriverArgs, const std::string &GPUArch, Action::OffloadKind DeviceOffloadingKind) const
Definition AMDGPU.cpp:1060
ROCMToolChain(const Driver &D, const llvm::Triple &Triple, const llvm::opt::ArgList &Args)
ROCM Toolchain.
Definition AMDGPU.cpp:833
Linker(const ToolChain &TC)
Definition AMDGPU.h:30
void ConstructJob(Compilation &C, const JobAction &JA, const InputInfo &Output, const InputInfoList &Inputs, const llvm::opt::ArgList &TCArgs, const char *LinkingOutput) const override
ConstructJob - Construct jobs to perform the action JA, writing to Output and with Inputs,...
Definition AMDGPU.cpp:599
AMDGPU builtins.
void getAMDGPUTargetFeatures(const Driver &D, const llvm::Triple &Triple, const llvm::opt::ArgList &Args, std::vector< StringRef > &Features)
Definition AMDGPU.cpp:654
void handleTargetFeaturesGroup(const Driver &D, const llvm::Triple &Triple, const llvm::opt::ArgList &Args, std::vector< StringRef > &Features, llvm::opt::OptSpecifier Group)
Iterate Args and convert -mxxx to +xxx and -mno-xxx to -xxx and append it to Features.
void checkAMDGPUCodeObjectVersion(const Driver &D, const llvm::opt::ArgList &Args)
void addLTOOptions(const ToolChain &ToolChain, const llvm::opt::ArgList &Args, llvm::opt::ArgStringList &CmdArgs, const InputInfo &Output, const InputInfoList &Inputs, bool IsThinLTO)
void addLinkerCompressDebugSectionsOption(const ToolChain &TC, const llvm::opt::ArgList &Args, llvm::opt::ArgStringList &CmdArgs)
void AddLinkerInputs(const ToolChain &TC, const InputInfoList &Inputs, const llvm::opt::ArgList &Args, llvm::opt::ArgStringList &CmdArgs, const JobAction &JA)
unsigned getAMDGPUCodeObjectVersion(const Driver &D, const llvm::opt::ArgList &Args)
void addOpenCLBuiltinsLib(const Driver &D, const llvm::Triple &TT, const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args)
SmallVector< InputInfo, 4 > InputInfoList
Definition Driver.h:50
The JSON file list parser is used to communicate input to InstallAPI.
if(T->getSizeExpr()) TRY_TO(TraverseStmt(const_cast< Expr * >(T -> getSizeExpr())))
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.
Definition TargetID.cpp:106
llvm::StringRef getProcessorFromTargetID(const llvm::Triple &T, llvm::StringRef OffloadArch)
Get processor name from target ID.
Definition TargetID.cpp:57
llvm::SmallVector< llvm::StringRef, 4 > getAllPossibleTargetIDFeatures(const llvm::Triple &T, llvm::StringRef Processor)
Get all feature strings that can be used in target ID for Processor.
Definition TargetID.cpp:41
Diagnostic wrappers for TextAPI types for error reporting.
Definition Dominators.h:30
static DeviceLibABIVersion fromCodeObjectVersion(unsigned CodeObjectVersion)
bool requiresLibrary()
Whether ABI version bc file is requested.
static constexpr ResponseFileSupport AtFileCurCP()
Definition Job.h:92
The struct type returned by getParsedTargetID.
Definition AMDGPU.h:120