Home | History | Annotate | Line # | Download | only in ToolChains 
      1 //===--- AMDGPU.h - AMDGPU 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_AMDGPU_H
     10 #define LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H
     11 
     12 #include "Gnu.h"
     13 #include "ROCm.h"
     14 #include "clang/Basic/TargetID.h"
     15 #include "clang/Driver/Options.h"
     16 #include "clang/Driver/Tool.h"
     17 #include "clang/Driver/ToolChain.h"
     18 #include "llvm/ADT/SmallString.h"
     19 #include "llvm/Support/TargetParser.h"
     20 
     21 #include <map>
     22 
     23 namespace clang {
     24 namespace driver {
     25 
     26 namespace tools {
     27 namespace amdgpu {
     28 
     29 class LLVM_LIBRARY_VISIBILITY Linker : public Tool {
     30 public:
     31   Linker(const ToolChain &TC) : Tool("amdgpu::Linker", "ld.lld", TC) {}
     32   bool isLinkJob() const override { return true; }
     33   bool hasIntegratedCPP() const override { return false; }
     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 
     40 void getAMDGPUTargetFeatures(const Driver &D, const llvm::Triple &Triple,
     41                              const llvm::opt::ArgList &Args,
     42                              std::vector<StringRef> &Features);
     43 
     44 } // end namespace amdgpu
     45 } // end namespace tools
     46 
     47 namespace toolchains {
     48 
     49 class LLVM_LIBRARY_VISIBILITY AMDGPUToolChain : public Generic_ELF {
     50 protected:
     51   const std::map<options::ID, const StringRef> OptionsDefault;
     52 
     53   Tool *buildLinker() const override;
     54   const StringRef getOptionDefault(options::ID OptID) const {
     55     auto opt = OptionsDefault.find(OptID);
     56     assert(opt != OptionsDefault.end() && "No Default for Option");
     57     return opt->second;
     58   }
     59 
     60 public:
     61   AMDGPUToolChain(const Driver &D, const llvm::Triple &Triple,
     62                   const llvm::opt::ArgList &Args);
     63   unsigned GetDefaultDwarfVersion() const override { return 4; }
     64   bool IsIntegratedAssemblerDefault() const override { return true; }
     65   bool IsMathErrnoDefault() const override { return false; }
     66 
     67   bool useIntegratedAs() const override { return true; }
     68   bool isCrossCompiling() const override { return true; }
     69   bool isPICDefault() const override { return false; }
     70   bool isPIEDefault() const override { return false; }
     71   bool isPICDefaultForced() const override { return false; }
     72   bool SupportsProfiling() const override { return false; }
     73 
     74   llvm::opt::DerivedArgList *
     75   TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch,
     76                 Action::OffloadKind DeviceOffloadKind) const override;
     77 
     78   void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
     79                              llvm::opt::ArgStringList &CC1Args,
     80                              Action::OffloadKind DeviceOffloadKind) const override;
     81 
     82   /// Return whether denormals should be flushed, and treated as 0 by default
     83   /// for the subtarget.
     84   static bool getDefaultDenormsAreZeroForTarget(llvm::AMDGPU::GPUKind GPUKind);
     85 
     86   llvm::DenormalMode getDefaultDenormalModeForType(
     87       const llvm::opt::ArgList &DriverArgs, const JobAction &JA,
     88       const llvm::fltSemantics *FPType = nullptr) const override;
     89 
     90   static bool isWave64(const llvm::opt::ArgList &DriverArgs,
     91                        llvm::AMDGPU::GPUKind Kind);
     92   /// Needed for using lto.
     93   bool HasNativeLLVMSupport() const override {
     94     return true;
     95   }
     96 
     97   /// Needed for translating LTO options.
     98   const char *getDefaultLinker() const override { return "ld.lld"; }
     99 
    100   /// Should skip argument.
    101   bool shouldSkipArgument(const llvm::opt::Arg *Arg) const;
    102 
    103   /// Uses amdgpu_arch tool to get arch of the system GPU. Will return error
    104   /// if unable to find one.
    105   llvm::Error getSystemGPUArch(const llvm::opt::ArgList &Args,
    106                                std::string &GPUArch) const;
    107 
    108 protected:
    109   /// Check and diagnose invalid target ID specified by -mcpu.
    110   void checkTargetID(const llvm::opt::ArgList &DriverArgs) const;
    111 
    112   /// Get GPU arch from -mcpu without checking.
    113   StringRef getGPUArch(const llvm::opt::ArgList &DriverArgs) const;
    114 
    115   llvm::Error detectSystemGPUs(const llvm::opt::ArgList &Args,
    116                                SmallVector<std::string, 1> &GPUArchs) const;
    117 };
    118 
    119 class LLVM_LIBRARY_VISIBILITY ROCMToolChain : public AMDGPUToolChain {
    120 public:
    121   ROCMToolChain(const Driver &D, const llvm::Triple &Triple,
    122                 const llvm::opt::ArgList &Args);
    123   void
    124   addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
    125                         llvm::opt::ArgStringList &CC1Args,
    126                         Action::OffloadKind DeviceOffloadKind) const override;
    127 };
    128 
    129 } // end namespace toolchains
    130 } // end namespace driver
    131 } // end namespace clang
    132 
    133 #endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H
    134