1 | //===--- HIPAMD.h - HIP 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_HIPAMD_H |
10 | #define LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_HIPAMD_H |
11 | |
12 | #include "AMDGPU.h" |
13 | #include "clang/Driver/Tool.h" |
14 | #include "clang/Driver/ToolChain.h" |
15 | |
16 | namespace clang { |
17 | namespace driver { |
18 | |
19 | namespace tools { |
20 | |
21 | namespace AMDGCN { |
22 | // Runs llvm-link/opt/llc/lld, which links multiple LLVM bitcode, together with |
23 | // device library, then compiles it to ISA in a shared object. |
24 | class LLVM_LIBRARY_VISIBILITY Linker final : public Tool { |
25 | public: |
26 | Linker(const ToolChain &TC) : Tool("AMDGCN::Linker" , "amdgcn-link" , TC) {} |
27 | |
28 | bool hasIntegratedCPP() const override { return false; } |
29 | |
30 | void ConstructJob(Compilation &C, const JobAction &JA, |
31 | const InputInfo &Output, const InputInfoList &Inputs, |
32 | const llvm::opt::ArgList &TCArgs, |
33 | const char *LinkingOutput) const override; |
34 | |
35 | private: |
36 | void constructLldCommand(Compilation &C, const JobAction &JA, |
37 | const InputInfoList &Inputs, const InputInfo &Output, |
38 | const llvm::opt::ArgList &Args) const; |
39 | void constructLlvmLinkCommand(Compilation &C, const JobAction &JA, |
40 | const InputInfoList &Inputs, |
41 | const InputInfo &Output, |
42 | const llvm::opt::ArgList &Args) const; |
43 | }; |
44 | |
45 | } // end namespace AMDGCN |
46 | } // end namespace tools |
47 | |
48 | namespace toolchains { |
49 | |
50 | class LLVM_LIBRARY_VISIBILITY HIPAMDToolChain final : public ROCMToolChain { |
51 | public: |
52 | HIPAMDToolChain(const Driver &D, const llvm::Triple &Triple, |
53 | const ToolChain &HostTC, const llvm::opt::ArgList &Args); |
54 | |
55 | const llvm::Triple *getAuxTriple() const override { |
56 | return &HostTC.getTriple(); |
57 | } |
58 | |
59 | llvm::opt::DerivedArgList * |
60 | TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch, |
61 | Action::OffloadKind DeviceOffloadKind) const override; |
62 | void |
63 | addClangTargetOptions(const llvm::opt::ArgList &DriverArgs, |
64 | llvm::opt::ArgStringList &CC1Args, |
65 | Action::OffloadKind DeviceOffloadKind) const override; |
66 | void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override; |
67 | CXXStdlibType GetCXXStdlibType(const llvm::opt::ArgList &Args) const override; |
68 | void |
69 | AddClangSystemIncludeArgs(const llvm::opt::ArgList &DriverArgs, |
70 | llvm::opt::ArgStringList &CC1Args) const override; |
71 | void AddClangCXXStdlibIncludeArgs( |
72 | const llvm::opt::ArgList &Args, |
73 | llvm::opt::ArgStringList &CC1Args) const override; |
74 | void AddIAMCUIncludeArgs(const llvm::opt::ArgList &DriverArgs, |
75 | llvm::opt::ArgStringList &CC1Args) const override; |
76 | void AddHIPIncludeArgs(const llvm::opt::ArgList &DriverArgs, |
77 | llvm::opt::ArgStringList &CC1Args) const override; |
78 | llvm::SmallVector<BitCodeLibraryInfo, 12> |
79 | getDeviceLibs(const llvm::opt::ArgList &Args) const override; |
80 | |
81 | SanitizerMask getSupportedSanitizers() const override; |
82 | |
83 | VersionTuple |
84 | computeMSVCVersion(const Driver *D, |
85 | const llvm::opt::ArgList &Args) const override; |
86 | |
87 | unsigned GetDefaultDwarfVersion() const override { return 5; } |
88 | |
89 | const ToolChain &HostTC; |
90 | void checkTargetID(const llvm::opt::ArgList &DriverArgs) const override; |
91 | |
92 | protected: |
93 | Tool *buildLinker() const override; |
94 | }; |
95 | |
96 | } // end namespace toolchains |
97 | } // end namespace driver |
98 | } // end namespace clang |
99 | |
100 | #endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_HIPAMD_H |
101 | |