clang/lib/Driver/ToolChains/HIPAMD.h

   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