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