dba558b474
This patch moves `LazyDetector` and target specific (Cuda, Hip, SYCL) installation detectors to clang's include directory. It was problematic for downstream to use headers from clang's lib dir. The use of lib headers could lead to subtle errors, as some of the symbols there are annotated with `LLVM_LIBRARY_VISIBILITY`. For instance [`ROCMToolChain::getCommonDeviceLibNames`](https://github.com/jchlanda/llvm-project/blob/jakub/installation_detectors/clang/lib/Driver/ToolChains/AMDGPU.h#L147) is c++ public, but because of the annotation it ends up as ELF hidden symbol, which causes errors when accessed from another shared library.
187 lines
6.9 KiB
C++
187 lines
6.9 KiB
C++
//===--- Cuda.h - Cuda 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_CUDA_H
|
|
#define LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_CUDA_H
|
|
|
|
#include "clang/Basic/Cuda.h"
|
|
#include "clang/Driver/Action.h"
|
|
#include "clang/Driver/CudaInstallationDetector.h"
|
|
#include "clang/Driver/Multilib.h"
|
|
#include "clang/Driver/Tool.h"
|
|
#include "clang/Driver/ToolChain.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
#include "llvm/Support/VersionTuple.h"
|
|
#include <bitset>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
namespace clang {
|
|
namespace driver {
|
|
namespace tools {
|
|
namespace NVPTX {
|
|
|
|
// Run ptxas, the NVPTX assembler.
|
|
class LLVM_LIBRARY_VISIBILITY Assembler final : public Tool {
|
|
public:
|
|
Assembler(const ToolChain &TC) : Tool("NVPTX::Assembler", "ptxas", TC) {}
|
|
|
|
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;
|
|
};
|
|
|
|
// Runs fatbinary, which combines GPU object files ("cubin" files) and/or PTX
|
|
// assembly into a single output file.
|
|
class LLVM_LIBRARY_VISIBILITY FatBinary : public Tool {
|
|
public:
|
|
FatBinary(const ToolChain &TC) : Tool("NVPTX::Linker", "fatbinary", TC) {}
|
|
|
|
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;
|
|
};
|
|
|
|
// Runs nvlink, which links GPU object files ("cubin" files) into a single file.
|
|
class LLVM_LIBRARY_VISIBILITY Linker final : public Tool {
|
|
public:
|
|
Linker(const ToolChain &TC) : Tool("NVPTX::Linker", "nvlink", TC) {}
|
|
|
|
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 getNVPTXTargetFeatures(const Driver &D, const llvm::Triple &Triple,
|
|
const llvm::opt::ArgList &Args,
|
|
std::vector<StringRef> &Features);
|
|
|
|
} // end namespace NVPTX
|
|
} // end namespace tools
|
|
|
|
namespace toolchains {
|
|
|
|
class LLVM_LIBRARY_VISIBILITY NVPTXToolChain : public ToolChain {
|
|
public:
|
|
NVPTXToolChain(const Driver &D, const llvm::Triple &Triple,
|
|
const llvm::Triple &HostTriple,
|
|
const llvm::opt::ArgList &Args);
|
|
|
|
NVPTXToolChain(const Driver &D, const llvm::Triple &Triple,
|
|
const llvm::opt::ArgList &Args);
|
|
|
|
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;
|
|
|
|
// Never try to use the integrated assembler with CUDA; always fork out to
|
|
// ptxas.
|
|
bool useIntegratedAs() const override { return false; }
|
|
bool isCrossCompiling() const override { return true; }
|
|
bool isPICDefault() const override { return false; }
|
|
bool isPIEDefault(const llvm::opt::ArgList &Args) const override {
|
|
return false;
|
|
}
|
|
bool HasNativeLLVMSupport() const override { return true; }
|
|
bool isPICDefaultForced() const override { return false; }
|
|
bool SupportsProfiling() const override { return false; }
|
|
|
|
bool IsMathErrnoDefault() const override { return false; }
|
|
|
|
bool supportsDebugInfoOption(const llvm::opt::Arg *A) const override;
|
|
void adjustDebugInfoKind(llvm::codegenoptions::DebugInfoKind &DebugInfoKind,
|
|
const llvm::opt::ArgList &Args) const override;
|
|
|
|
// NVPTX supports only DWARF2.
|
|
unsigned GetDefaultDwarfVersion() const override { return 2; }
|
|
unsigned getMaxDwarfVersion() const override { return 2; }
|
|
|
|
/// Uses nvptx-arch tool to get arch of the system GPU. Will return error
|
|
/// if unable to find one.
|
|
virtual Expected<SmallVector<std::string>>
|
|
getSystemGPUArchs(const llvm::opt::ArgList &Args) const override;
|
|
|
|
CudaInstallationDetector CudaInstallation;
|
|
|
|
protected:
|
|
Tool *buildAssembler() const override; // ptxas.
|
|
Tool *buildLinker() const override; // nvlink.
|
|
};
|
|
|
|
class LLVM_LIBRARY_VISIBILITY CudaToolChain : public NVPTXToolChain {
|
|
public:
|
|
CudaToolChain(const Driver &D, const llvm::Triple &Triple,
|
|
const ToolChain &HostTC, const llvm::opt::ArgList &Args);
|
|
|
|
const llvm::Triple *getAuxTriple() const override {
|
|
return &HostTC.getTriple();
|
|
}
|
|
|
|
bool HasNativeLLVMSupport() const override { return false; }
|
|
|
|
std::string getInputFilename(const InputInfo &Input) const override;
|
|
|
|
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;
|
|
|
|
llvm::DenormalMode getDefaultDenormalModeForType(
|
|
const llvm::opt::ArgList &DriverArgs, const JobAction &JA,
|
|
const llvm::fltSemantics *FPType = nullptr) const override;
|
|
|
|
void AddCudaIncludeArgs(const llvm::opt::ArgList &DriverArgs,
|
|
llvm::opt::ArgStringList &CC1Args) const override;
|
|
|
|
void addClangWarningOptions(llvm::opt::ArgStringList &CC1Args) const override;
|
|
CXXStdlibType GetCXXStdlibType(const llvm::opt::ArgList &Args) const override;
|
|
void
|
|
AddClangSystemIncludeArgs(const llvm::opt::ArgList &DriverArgs,
|
|
llvm::opt::ArgStringList &CC1Args) const override;
|
|
void AddClangCXXStdlibIncludeArgs(
|
|
const llvm::opt::ArgList &Args,
|
|
llvm::opt::ArgStringList &CC1Args) const override;
|
|
void AddIAMCUIncludeArgs(const llvm::opt::ArgList &DriverArgs,
|
|
llvm::opt::ArgStringList &CC1Args) const override;
|
|
|
|
SanitizerMask getSupportedSanitizers() const override;
|
|
|
|
VersionTuple
|
|
computeMSVCVersion(const Driver *D,
|
|
const llvm::opt::ArgList &Args) const override;
|
|
|
|
const ToolChain &HostTC;
|
|
|
|
protected:
|
|
Tool *buildAssembler() const override; // ptxas
|
|
Tool *buildLinker() const override; // fatbinary (ok, not really a linker)
|
|
};
|
|
|
|
} // end namespace toolchains
|
|
} // end namespace driver
|
|
} // end namespace clang
|
|
|
|
#endif // LLVM_CLANG_LIB_DRIVER_TOOLCHAINS_CUDA_H
|