You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.6 KiB
124 lines
3.6 KiB
/* 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()
|
|
{
|
|
}
|
|
|
|
static bool cmLinkItemValidForDevice(std::string const& item)
|
|
{
|
|
// Valid items are:
|
|
// * Non-flags (does not start in '-')
|
|
// * Specific flags --library, --library-path, -l, -L
|
|
// For example:
|
|
// * 'cublas_device' => pass-along
|
|
// * '--library pthread' => pass-along
|
|
// * '-lpthread' => pass-along
|
|
// * '-pthread' => drop
|
|
// * '-a' => drop
|
|
return (!cmHasLiteralPrefix(item, "-") || //
|
|
cmHasLiteralPrefix(item, "-l") || //
|
|
cmHasLiteralPrefix(item, "-L") || //
|
|
cmHasLiteralPrefix(item, "--library"));
|
|
}
|
|
|
|
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' or '.lib'.
|
|
// These should be passed to nvlink. Other extensions need to be left
|
|
// out because nvlink may not understand or need them. Even though it
|
|
// can tolerate '.so' or '.dylib' it cannot tolerate '.so.1'.
|
|
if (cmHasLiteralSuffix(item.Value, ".a") ||
|
|
cmHasLiteralSuffix(item.Value, ".lib")) {
|
|
out += this->ConvertToOutputFormat(
|
|
this->ConvertToLinkReference(item.Value));
|
|
}
|
|
} else if (cmLinkItemValidForDevice(item.Value)) {
|
|
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);
|
|
}
|