5f2a7fa67e
This patch is extracted from D96035, it adds support for the existing DWARFLinker functionality. What is not supported yet: 1. Types deduplication(--odr mode). 2. Modules deduplication. 3. Generation of index tables. Reland2: temporarily disabled call to "--linker llvm" for tls-variable.test and location-expression.test as it does not work properly on bigendian architecture. Differential Revision: https://reviews.llvm.org/D153268
65 lines
1.9 KiB
C++
65 lines
1.9 KiB
C++
//===- llvm/unittest/DWARFLinkerParallel/StringPoolTest.cpp ---------------===//
|
|
//
|
|
// 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/DWARFLinkerParallel/StringPool.h"
|
|
#include "llvm/Support/Parallel.h"
|
|
#include "gtest/gtest.h"
|
|
#include <cstdlib>
|
|
|
|
using namespace llvm;
|
|
using namespace dwarflinker_parallel;
|
|
|
|
namespace {
|
|
|
|
TEST(StringPoolTest, TestStringPool) {
|
|
StringPool Strings;
|
|
|
|
// StringPool uses PerThreadBumpPtrAllocator which should be accessed from
|
|
// threads created by ThreadPoolExecutor. Use TaskGroup to run on
|
|
// ThreadPoolExecutor threads.
|
|
parallel::TaskGroup tg;
|
|
|
|
tg.spawn([&]() {
|
|
std::pair<StringEntry *, bool> Entry = Strings.insert("test");
|
|
EXPECT_TRUE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == "test");
|
|
|
|
StringEntry *EntryPtr = Entry.first;
|
|
|
|
Entry = Strings.insert("test");
|
|
EXPECT_FALSE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == "test");
|
|
EXPECT_TRUE(EntryPtr == Entry.first);
|
|
|
|
Entry = Strings.insert("test2");
|
|
EXPECT_TRUE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == "test2");
|
|
EXPECT_TRUE(EntryPtr != Entry.first);
|
|
});
|
|
}
|
|
|
|
TEST(StringPoolTest, TestStringPoolParallel) {
|
|
StringPool Strings;
|
|
|
|
// Add data.
|
|
parallelFor(0, 1000, [&](size_t Idx) {
|
|
std::pair<StringEntry *, bool> Entry = Strings.insert(std::to_string(Idx));
|
|
EXPECT_TRUE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == std::to_string(Idx));
|
|
});
|
|
|
|
// Check data.
|
|
parallelFor(0, 1000, [&](size_t Idx) {
|
|
std::pair<StringEntry *, bool> Entry = Strings.insert(std::to_string(Idx));
|
|
EXPECT_FALSE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == std::to_string(Idx));
|
|
});
|
|
}
|
|
|
|
} // anonymous namespace
|