mirror of
https://github.com/Kitware/CMake.git
synced 2025-05-09 23:08:18 +08:00
fd0523a215
The nvcc device linker is designed so that each static library with device symbols only needs to be listed once as it doesn't care about link order. If you provide the same static library multiple times it will error out. To make sure this occurs we find the unique set of link items.
107 lines
3.0 KiB
C++
107 lines
3.0 KiB
C++
/* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
|
|
file Copyright.txt or https://cmake.org/licensing for details. */
|
|
|
|
#include "cmLinkLineDeviceComputer.h"
|
|
|
|
#include <set>
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
#include "cmAlgorithms.h"
|
|
#include "cmComputeLinkInformation.h"
|
|
#include "cmGeneratorTarget.h"
|
|
#include "cmGlobalNinjaGenerator.h"
|
|
#include "cmStateTypes.h"
|
|
|
|
class cmOutputConverter;
|
|
|
|
cmLinkLineDeviceComputer::cmLinkLineDeviceComputer(
|
|
cmOutputConverter* outputConverter, cmStateDirectory const& stateDir)
|
|
: cmLinkLineComputer(outputConverter, stateDir)
|
|
{
|
|
}
|
|
|
|
cmLinkLineDeviceComputer::~cmLinkLineDeviceComputer()
|
|
{
|
|
}
|
|
|
|
std::string cmLinkLineDeviceComputer::ComputeLinkLibraries(
|
|
cmComputeLinkInformation& cli, std::string const& stdLibString)
|
|
{
|
|
// Write the library flags to the build rule.
|
|
std::ostringstream fout;
|
|
|
|
// Generate the unique set of link items when device linking.
|
|
// The nvcc device linker is designed so that each static library
|
|
// with device symbols only needs to be listed once as it doesn't
|
|
// care about link order.
|
|
std::set<std::string> emitted;
|
|
typedef cmComputeLinkInformation::ItemVector ItemVector;
|
|
ItemVector const& items = cli.GetItems();
|
|
std::string config = cli.GetConfig();
|
|
for (auto const& item : items) {
|
|
if (item.Target) {
|
|
bool skip = false;
|
|
switch (item.Target->GetType()) {
|
|
case cmStateEnums::MODULE_LIBRARY:
|
|
case cmStateEnums::INTERFACE_LIBRARY:
|
|
skip = true;
|
|
break;
|
|
case cmStateEnums::STATIC_LIBRARY:
|
|
skip = item.Target->GetPropertyAsBool("CUDA_RESOLVE_DEVICE_SYMBOLS");
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
if (skip) {
|
|
continue;
|
|
}
|
|
}
|
|
|
|
std::string out;
|
|
if (item.IsPath) {
|
|
// nvcc understands absolute paths to libraries ending in '.a' should
|
|
// be passed to nvlink. Other extensions like '.so' or '.dylib' are
|
|
// rejected by the nvcc front-end even though nvlink knows to ignore
|
|
// them. Bypass the front-end via '-Xnvlink'.
|
|
if (!cmHasLiteralSuffix(item.Value, ".a")) {
|
|
out += "-Xnvlink ";
|
|
}
|
|
out +=
|
|
this->ConvertToOutputFormat(this->ConvertToLinkReference(item.Value));
|
|
} else {
|
|
out += item.Value;
|
|
}
|
|
|
|
if (emitted.insert(out).second) {
|
|
fout << out << " ";
|
|
}
|
|
}
|
|
|
|
if (!stdLibString.empty()) {
|
|
fout << stdLibString << " ";
|
|
}
|
|
|
|
return fout.str();
|
|
}
|
|
|
|
std::string cmLinkLineDeviceComputer::GetLinkerLanguage(cmGeneratorTarget*,
|
|
std::string const&)
|
|
{
|
|
return "CUDA";
|
|
}
|
|
|
|
cmNinjaLinkLineDeviceComputer::cmNinjaLinkLineDeviceComputer(
|
|
cmOutputConverter* outputConverter, cmStateDirectory const& stateDir,
|
|
cmGlobalNinjaGenerator const* gg)
|
|
: cmLinkLineDeviceComputer(outputConverter, stateDir)
|
|
, GG(gg)
|
|
{
|
|
}
|
|
|
|
std::string cmNinjaLinkLineDeviceComputer::ConvertToLinkReference(
|
|
std::string const& lib) const
|
|
{
|
|
return GG->ConvertToNinjaPath(lib);
|
|
}
|