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
15namespace clang {
16namespace driver {
17
18namespace tools {
19
20namespace 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.
28class LLVM_LIBRARY_VISIBILITY Linker : public Tool {
29public:
30  Linker(const ToolChain &TC) : Tool("AMDGCN::Linker", "amdgcn-link", TC) {}
31
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
39private:
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,
62                                  bool OutputIsAsm = false) const;
63
64  void constructLldCommand(Compilation &C, const JobAction &JA,
65                           const InputInfoList &Inputs, const InputInfo &Output,
66                           const llvm::opt::ArgList &Args,
67                           const char *InputFileName) const;
68};
69
70} // end namespace AMDGCN
71} // end namespace tools
72
73namespace toolchains {
74
75class LLVM_LIBRARY_VISIBILITY HIPToolChain : public ToolChain {
76public:
77  HIPToolChain(const Driver &D, const llvm::Triple &Triple,
78                const ToolChain &HostTC, const llvm::opt::ArgList &Args);
79
80  const llvm::Triple *getAuxTriple() const override {
81    return &HostTC.getTriple();
82  }
83
84  llvm::opt::DerivedArgList *
85  TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch,
86                Action::OffloadKind DeviceOffloadKind) const override;
87  void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
88                             llvm::opt::ArgStringList &CC1Args,
89                             Action::OffloadKind DeviceOffloadKind) const override;
90
91  bool useIntegratedAs() const override { return true; }
92  bool isCrossCompiling() const override { return true; }
93  bool isPICDefault() const override { return false; }
94  bool isPIEDefault() const override { return false; }
95  bool isPICDefaultForced() const override { return false; }
96  bool SupportsProfiling() const override { return false; }
97  bool IsMathErrnoDefault() const override { return false; }
98
99  void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override;
100  CXXStdlibType GetCXXStdlibType(const llvm::opt::ArgList &Args) const override;
101  void
102  AddClangSystemIncludeArgs(const llvm::opt::ArgList &DriverArgs,
103                            llvm::opt::ArgStringList &CC1Args) const override;
104  void AddClangCXXStdlibIncludeArgs(
105      const llvm::opt::ArgList &Args,
106      llvm::opt::ArgStringList &CC1Args) const override;
107  void AddIAMCUIncludeArgs(const llvm::opt::ArgList &DriverArgs,
108                           llvm::opt::ArgStringList &CC1Args) const override;
109
110  SanitizerMask getSupportedSanitizers() const override;
111
112  VersionTuple
113  computeMSVCVersion(const Driver *D,
114                     const llvm::opt::ArgList &Args) const override;
115
116  unsigned GetDefaultDwarfVersion() const override { return 4; }
117
118  const ToolChain &HostTC;
119
120protected:
121  Tool *buildLinker() const override;
122};
123
124} // end namespace toolchains
125} // end namespace driver
126} // end namespace clang
127
128#endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_HIP_H
129