mirror/CMake
1
0
Fork 0
mirror of https://github.com/Kitware/CMake.git synced 2026-04-23 14:48:19 -05:00
Code Issues Packages Projects Releases Wiki Activity

CUDA: Clang separable compilation

Browse Source 
For NVCC the compiler takes care of device linking when passed the "-dlink"
flag.
Clang doesn't support such magic and requires the buildsystem to do the work
that NVCC does behind the scenes.

The implementation is based on Bazel's device linking documentation:
https://github.com/tensorflow/tensorflow/blob/7cabcdf073abad8c46e9dda62bb8fa4682d2061e/third_party/nccl/build_defs.bzl.tpl#L259

Closes: #20726
This commit is contained in:
Raul Tambre
2020-09-05 19:40:02 +03:00
committed by Brad King
parent c98ec731f9
commit c63fe01835
22 changed files with 506 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Source/cmNinjaNormalTargetGenerator.h
+14 -1
View File
@@ -21,18 +21,31 @@ public:
private:
std::string LanguageLinkerRule(const std::string& config) const;
std::string LanguageLinkerDeviceRule(const std::string& config) const;
std::string LanguageLinkerCudaDeviceRule(const std::string& config) const;
std::string LanguageLinkerCudaDeviceCompileRule(
const std::string& config) const;
std::string LanguageLinkerCudaFatbinaryRule(const std::string& config) const;
const char* GetVisibleTypeName() const;
void WriteLanguagesRules(const std::string& config);
void WriteLinkRule(bool useResponseFile, const std::string& config);
void WriteDeviceLinkRule(bool useResponseFile, const std::string& config);
void WriteDeviceLinkRules(const std::string& config);
void WriteNvidiaDeviceLinkRule(bool useResponseFile,
const std::string& config);
void WriteLinkStatement(const std::string& config,
const std::string& fileConfig, bool firstForConfig);
void WriteDeviceLinkStatement(const std::string& config,
const std::string& fileConfig,
bool firstForConfig);
void WriteDeviceLinkStatements(const std::string& config,
const std::vector<std::string>& architectures,
const std::string& output);
void WriteNvidiaDeviceLinkStatement(const std::string& config,
const std::string& fileConfig,
const std::string& outputDir,
const std::string& output);
void WriteObjectLibStatement(const std::string& config);