1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
//===- ROCm.cpp -------------------------------------------------*- C++ -*-===//
//
// This file is licensed 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
//
//===----------------------------------------------------------------------===//
// The purpose of this test is to showcase a more singular usage of LLD as a
// library, where only one LLD driver is being used (and linked in the target
// application). We also expect that linking twice the same object files
// would yield a successfull result. When used as library, LLD always cleans its
// internal memory context after each linker call.
//===----------------------------------------------------------------------===//
// When this flag is on, we actually need the MinGW driver library, not the
// ELF one. Here our test only covers the case where the ELF driver is linked
// into the unit test binary.
#ifndef LLD_DEFAULT_LD_LLD_IS_MINGW
#include "lld/Common/Driver.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/FileUtilities.h"
#include "llvm/Support/Path.h"
#include "gmock/gmock.h"
#include <algorithm>
static std::string expand(const char *path) {
if (!llvm::StringRef(path).contains("%"))
return std::string(path);
llvm::SmallString<256> thisPath;
thisPath.append(getenv("LLD_SRC_DIR"));
llvm::sys::path::append(thisPath, "unittests", "AsLibELF");
std::string expanded(path);
expanded.replace(expanded.find("%S"), 2, thisPath.data(), thisPath.size());
return expanded;
}
LLD_HAS_DRIVER(elf)
static bool lldInvoke(const char *inPath, const char *outPath) {
std::vector<const char *> args{"ld.lld", "-shared", inPath, "-o", outPath};
lld::Result s = lld::lldMain(args, llvm::outs(), llvm::errs(),
{{lld::Gnu, &lld::elf::link}});
return !s.retCode && s.canRunAgain;
}
static bool runLinker(const char *path) {
// Create a temp file for HSA code object.
int tempHsacoFD = -1;
llvm::SmallString<128> tempHsacoFilename;
if (llvm::sys::fs::createTemporaryFile("kernel", "hsaco", tempHsacoFD,
tempHsacoFilename)) {
return false;
}
llvm::FileRemover cleanupHsaco(tempHsacoFilename);
// Invoke lld. Expect a true return value from lld.
std::string expandedPath = expand(path);
if (!lldInvoke(expandedPath.data(), tempHsacoFilename.c_str())) {
llvm::errs() << "Failed to link: " << expandedPath << "\n";
return false;
}
return true;
}
TEST(AsLib, ROCm) {
EXPECT_TRUE(runLinker("%S/Inputs/kernel1.o"));
EXPECT_TRUE(runLinker("%S/Inputs/kernel2.o"));
EXPECT_TRUE(runLinker("%S/Inputs/kernel1.o"));
EXPECT_TRUE(runLinker("%S/Inputs/kernel2.o"));
}
#endif
|
