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 105