forked from OSchip/llvm-project
70 lines
2.2 KiB
C++
70 lines
2.2 KiB
C++
//===- 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"));
|
|
}
|