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   void constructLinkAndEmitSpirvCommand(Compilation &C, const JobAction &JA,
  44                                         const InputInfoList &Inputs,
  45                                         const InputInfo &Output,
  46                                         const llvm::opt::ArgList &Args) const;
  47 };
  48
  49 } // end namespace AMDGCN
  50 } // end namespace tools
  51
  52 namespace toolchains {
  53
  54 class LLVM_LIBRARY_VISIBILITY HIPAMDToolChain final : public ROCMToolChain {
  55 public:
  56   HIPAMDToolChain(const Driver &D, const llvm::Triple &Triple,
  57                   const ToolChain &HostTC, const llvm::opt::ArgList &Args);
  58
  59   const llvm::Triple *getAuxTriple() const override {
  60     return &HostTC.getTriple();
  61   }
  62
  63   llvm::opt::DerivedArgList *
  64   TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch,
  65                 Action::OffloadKind DeviceOffloadKind) const override;
  66   void
  67   addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
  68                         llvm::opt::ArgStringList &CC1Args,
  69                         Action::OffloadKind DeviceOffloadKind) const override;
  70   void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override;
  71   CXXStdlibType GetCXXStdlibType(const llvm::opt::ArgList &Args) const override;
  72   void
  73   AddClangSystemIncludeArgs(const llvm::opt::ArgList &DriverArgs,
  74                             llvm::opt::ArgStringList &CC1Args) const override;
  75   void AddClangCXXStdlibIncludeArgs(
  76       const llvm::opt::ArgList &Args,
  77       llvm::opt::ArgStringList &CC1Args) const override;
  78   void AddIAMCUIncludeArgs(const llvm::opt::ArgList &DriverArgs,
  79                            llvm::opt::ArgStringList &CC1Args) const override;
  80   void AddHIPIncludeArgs(const llvm::opt::ArgList &DriverArgs,
  81                          llvm::opt::ArgStringList &CC1Args) const override;
  82   llvm::SmallVector<BitCodeLibraryInfo, 12>
  83   getDeviceLibs(const llvm::opt::ArgList &Args) const override;
  84
  85   SanitizerMask getSupportedSanitizers() const override;
  86
  87   VersionTuple
  88   computeMSVCVersion(const Driver *D,
  89                      const llvm::opt::ArgList &Args) const override;
  90
  91   unsigned GetDefaultDwarfVersion() const override { return 5; }
  92
  93   const ToolChain &HostTC;
  94   void checkTargetID(const llvm::opt::ArgList &DriverArgs) const override;
  95
  96 protected:
  97   Tool *buildLinker() const override;
  98 };
  99
 100 } // end namespace toolchains
 101 } // end namespace driver
 102 } // end namespace clang
 103
 104 #endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_HIPAMD_H