119 lines
4.1 KiB
C++
119 lines
4.1 KiB
C++
//===--- AMDGPU.h - AMDGPU ToolChain Implementations ----------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H
|
|
#define LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H
|
|
|
|
#include "Gnu.h"
|
|
#include "ROCm.h"
|
|
#include "clang/Basic/TargetID.h"
|
|
#include "clang/Driver/Options.h"
|
|
#include "clang/Driver/Tool.h"
|
|
#include "clang/Driver/ToolChain.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/Support/TargetParser.h"
|
|
|
|
#include <map>
|
|
|
|
namespace clang {
|
|
namespace driver {
|
|
|
|
namespace tools {
|
|
namespace amdgpu {
|
|
|
|
class LLVM_LIBRARY_VISIBILITY Linker : public Tool {
|
|
public:
|
|
Linker(const ToolChain &TC) : Tool("amdgpu::Linker", "ld.lld", TC) {}
|
|
bool isLinkJob() const override { return true; }
|
|
bool hasIntegratedCPP() const override { return false; }
|
|
void ConstructJob(Compilation &C, const JobAction &JA,
|
|
const InputInfo &Output, const InputInfoList &Inputs,
|
|
const llvm::opt::ArgList &TCArgs,
|
|
const char *LinkingOutput) const override;
|
|
};
|
|
|
|
void getAMDGPUTargetFeatures(const Driver &D, const llvm::Triple &Triple,
|
|
const llvm::opt::ArgList &Args,
|
|
std::vector<StringRef> &Features);
|
|
|
|
} // end namespace amdgpu
|
|
} // end namespace tools
|
|
|
|
namespace toolchains {
|
|
|
|
class LLVM_LIBRARY_VISIBILITY AMDGPUToolChain : public Generic_ELF {
|
|
protected:
|
|
const std::map<options::ID, const StringRef> OptionsDefault;
|
|
|
|
Tool *buildLinker() const override;
|
|
const StringRef getOptionDefault(options::ID OptID) const {
|
|
auto opt = OptionsDefault.find(OptID);
|
|
assert(opt != OptionsDefault.end() && "No Default for Option");
|
|
return opt->second;
|
|
}
|
|
|
|
public:
|
|
AMDGPUToolChain(const Driver &D, const llvm::Triple &Triple,
|
|
const llvm::opt::ArgList &Args);
|
|
unsigned GetDefaultDwarfVersion() const override { return 4; }
|
|
bool IsIntegratedAssemblerDefault() const override { return true; }
|
|
bool IsMathErrnoDefault() const override { return false; }
|
|
|
|
llvm::opt::DerivedArgList *
|
|
TranslateArgs(const llvm::opt::DerivedArgList &Args, StringRef BoundArch,
|
|
Action::OffloadKind DeviceOffloadKind) const override;
|
|
|
|
void addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
|
|
llvm::opt::ArgStringList &CC1Args,
|
|
Action::OffloadKind DeviceOffloadKind) const override;
|
|
|
|
/// Return whether denormals should be flushed, and treated as 0 by default
|
|
/// for the subtarget.
|
|
static bool getDefaultDenormsAreZeroForTarget(llvm::AMDGPU::GPUKind GPUKind);
|
|
|
|
llvm::DenormalMode getDefaultDenormalModeForType(
|
|
const llvm::opt::ArgList &DriverArgs, const JobAction &JA,
|
|
const llvm::fltSemantics *FPType = nullptr) const override;
|
|
|
|
static bool isWave64(const llvm::opt::ArgList &DriverArgs,
|
|
llvm::AMDGPU::GPUKind Kind);
|
|
/// Needed for using lto.
|
|
bool HasNativeLLVMSupport() const override {
|
|
return true;
|
|
}
|
|
|
|
/// Needed for translating LTO options.
|
|
const char *getDefaultLinker() const override { return "ld.lld"; }
|
|
|
|
/// Should skip argument.
|
|
bool shouldSkipArgument(const llvm::opt::Arg *Arg) const;
|
|
|
|
protected:
|
|
/// Check and diagnose invalid target ID specified by -mcpu.
|
|
void checkTargetID(const llvm::opt::ArgList &DriverArgs) const;
|
|
|
|
/// Get GPU arch from -mcpu without checking.
|
|
StringRef getGPUArch(const llvm::opt::ArgList &DriverArgs) const;
|
|
};
|
|
|
|
class LLVM_LIBRARY_VISIBILITY ROCMToolChain : public AMDGPUToolChain {
|
|
public:
|
|
ROCMToolChain(const Driver &D, const llvm::Triple &Triple,
|
|
const llvm::opt::ArgList &Args);
|
|
void
|
|
addClangTargetOptions(const llvm::opt::ArgList &DriverArgs,
|
|
llvm::opt::ArgStringList &CC1Args,
|
|
Action::OffloadKind DeviceOffloadKind) const override;
|
|
};
|
|
|
|
} // end namespace toolchains
|
|
} // end namespace driver
|
|
} // end namespace clang
|
|
|
|
#endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_AMDGPU_H
|