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
|
//===- ModuleUtilsTest.cpp - Unit tests for Module utility ----===//
//
// 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
//
//===----------------------------------------------------------------------===//
#include "llvm/Transforms/Utils/ModuleUtils.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/AsmParser/Parser.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/Module.h"
#include "llvm/Support/SourceMgr.h"
#include "gtest/gtest.h"
using namespace llvm;
static std::unique_ptr<Module> parseIR(LLVMContext &C, const char *IR) {
SMDiagnostic Err;
std::unique_ptr<Module> Mod = parseAssemblyString(IR, Err, C);
if (!Mod)
Err.print("ModuleUtilsTest", errs());
return Mod;
}
static int getUsedListSize(Module &M, StringRef Name) {
auto *UsedList = M.getGlobalVariable(Name);
if (!UsedList)
return 0;
auto *UsedListBaseArrayType = cast<ArrayType>(UsedList->getValueType());
return UsedListBaseArrayType->getNumElements();
}
TEST(ModuleUtils, AppendToUsedList1) {
LLVMContext C;
std::unique_ptr<Module> M = parseIR(
C, R"(@x = addrspace(4) global [2 x i32] zeroinitializer, align 4)");
SmallVector<GlobalValue *, 2> Globals;
for (auto &G : M->globals()) {
Globals.push_back(&G);
}
EXPECT_EQ(0, getUsedListSize(*M, "llvm.compiler.used"));
appendToCompilerUsed(*M, Globals);
EXPECT_EQ(1, getUsedListSize(*M, "llvm.compiler.used"));
EXPECT_EQ(0, getUsedListSize(*M, "llvm.used"));
appendToUsed(*M, Globals);
EXPECT_EQ(1, getUsedListSize(*M, "llvm.used"));
}
TEST(ModuleUtils, AppendToUsedList2) {
LLVMContext C;
std::unique_ptr<Module> M =
parseIR(C, R"(@x = global [2 x i32] zeroinitializer, align 4)");
SmallVector<GlobalValue *, 2> Globals;
for (auto &G : M->globals()) {
Globals.push_back(&G);
}
EXPECT_EQ(0, getUsedListSize(*M, "llvm.compiler.used"));
appendToCompilerUsed(*M, Globals);
EXPECT_EQ(1, getUsedListSize(*M, "llvm.compiler.used"));
EXPECT_EQ(0, getUsedListSize(*M, "llvm.used"));
appendToUsed(*M, Globals);
EXPECT_EQ(1, getUsedListSize(*M, "llvm.used"));
}
|
