1 //===--- HIP.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_HIP_H
10 #define LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_HIP_H
11 
12 #include "clang/Driver/ToolChain.h"
13 #include "clang/Driver/Tool.h"
14 
15 namespace clang {
16 namespace driver {
17 
18 namespace tools {
19 
20 namespace AMDGCN {
21   // Construct command for creating HIP fatbin.
22   void constructHIPFatbinCommand(Compilation &C, const JobAction &JA,
23                   StringRef OutputFileName, const InputInfoList &Inputs,
24                   const llvm::opt::ArgList &TCArgs, const Tool& T);
25 
26 // Runs llvm-link/opt/llc/lld, which links multiple LLVM bitcode, together with
27 // device library, then compiles it to ISA in a shared object.
28 class LLVM_LIBRARY_VISIBILITY Linker : public Tool {
29 public:
Linker(const ToolChain & TC)30   Linker(const ToolChain &TC) : Tool("AMDGCN::Linker", "amdgcn-link", TC) {}
31 
hasIntegratedCPP()32   bool hasIntegratedCPP() const override { return false; }
33 
34   void ConstructJob(Compilation &C, const JobAction &JA,
35                     const InputInfo &Output, const InputInfoList &Inputs,
36                     const llvm::opt::ArgList &TCArgs,
37                     const char *LinkingOutput) const override;
38 
39 private:
40   /// \return llvm-link output file name.
41   const char *constructLLVMLinkCommand(Compilation &C, const JobAction &JA,
42                                        const InputInfoList &Inputs,
43                                        const llvm::opt::ArgList &Args,
44                                        llvm::StringRef SubArchName,
45                                        llvm::StringRef OutputFilePrefix) const;
46 
47   /// \return opt output file name.
48   const char *constructOptCommand(Compilation &C, const JobAction &JA,
49                                   const InputInfoList &Inputs,
50                                   const llvm::opt::ArgList &Args,
51                                   llvm::StringRef SubArchName,
52                                   llvm::StringRef OutputFilePrefix,
53                                   const char *InputFileName) const;
54 
55   /// \return llc output file name.
56   const char *constructLlcCommand(Compilation &C, const JobAction &JA,
57                                   const InputInfoList &Inputs,
58                                   const llvm::opt::ArgList &Args,
59                                   llvm::StringRef SubArchName,
60                                   llvm::StringRef OutputFilePrefix,
61                                   const char *InputFileName) const;
62 
63   void constructLldCommand(Compilation &C, const JobAction &JA,
64                            const InputInfoList &Inputs, const InputInfo &Output,
65                            const llvm::opt::ArgList &Args,
66                            const char *InputFileName) const;
67 };
68 
69 } // end namespace AMDGCN
70 } // end namespace tools
71 
72 namespace toolchains {
73 
74 class LLVM_LIBRARY_VISIBILITY HIPToolChain : public ToolChain {
75 public:
76   HIPToolChain(const Driver &D, const llvm::Triple &Triple,
77                 const ToolChain &HostTC, const llvm::opt::ArgList &Args);
78 
getAuxTriple()79   const llvm::Triple *getAuxTriple() const override {
80     return &HostTC.getTriple();
81   }
82 
83   llvm::opt::DerivedArgList *
84   TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch,
85                 Action::OffloadKind DeviceOffloadKind) const override;
86   void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
87                              llvm::opt::ArgStringList &CC1Args,
88                              Action::OffloadKind DeviceOffloadKind) const override;
89 
useIntegratedAs()90   bool useIntegratedAs() const override { return true; }
isCrossCompiling()91   bool isCrossCompiling() const override { return true; }
isPICDefault()92   bool isPICDefault() const override { return false; }
isPIEDefault()93   bool isPIEDefault() const override { return false; }
isPICDefaultForced()94   bool isPICDefaultForced() const override { return false; }
SupportsProfiling()95   bool SupportsProfiling() const override { return false; }
IsMathErrnoDefault()96   bool IsMathErrnoDefault() const override { return false; }
97 
98   void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override;
99   CXXStdlibType GetCXXStdlibType(const llvm::opt::ArgList &Args) const override;
100   void
101   AddClangSystemIncludeArgs(const llvm::opt::ArgList &DriverArgs,
102                             llvm::opt::ArgStringList &CC1Args) const override;
103   void AddClangCXXStdlibIncludeArgs(
104       const llvm::opt::ArgList &Args,
105       llvm::opt::ArgStringList &CC1Args) const override;
106   void AddIAMCUIncludeArgs(const llvm::opt::ArgList &DriverArgs,
107                            llvm::opt::ArgStringList &CC1Args) const override;
108 
109   SanitizerMask getSupportedSanitizers() const override;
110 
111   VersionTuple
112   computeMSVCVersion(const Driver *D,
113                      const llvm::opt::ArgList &Args) const override;
114 
GetDefaultDwarfVersion()115   unsigned GetDefaultDwarfVersion() const override { return 2; }
116 
117   const ToolChain &HostTC;
118 
119 protected:
120   Tool *buildLinker() const override;
121 };
122 
123 } // end namespace toolchains
124 } // end namespace driver
125 } // end namespace clang
126 
127 #endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_HIP_H
128