xref: /freebsd/contrib/llvm-project/clang/lib/Driver/ToolChains/HIPAMD.h (revision 2e3507c25e42292b45a5482e116d278f5515d04d)
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 : 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