6ab43f9b87
PerThreadBumpPtrAllocator allows separating allocations by thread id. That makes allocations race free. It is possible because ThreadPoolExecutor class creates threads, keeps them until the destructor of ThreadPoolExecutor is called, and assigns ids to the threads. Thus PerThreadBumpPtrAllocator should be used with only threads created by ThreadPoolExecutor. This allocator is useful when thread safe BumpPtrAllocator is needed. Reviewed By: MaskRay, dexonsmith, andrewng Differential Revision: https://reviews.llvm.org/D142318
70 lines
2.1 KiB
C++
70 lines
2.1 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");
|
|
EXPECT_TRUE(Entry.first->second == nullptr);
|
|
|
|
StringEntry *EntryPtr = Entry.first;
|
|
|
|
Entry = Strings.insert("test");
|
|
EXPECT_FALSE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == "test");
|
|
EXPECT_TRUE(Entry.first->second == nullptr);
|
|
EXPECT_TRUE(EntryPtr == Entry.first);
|
|
|
|
Entry = Strings.insert("test2");
|
|
EXPECT_TRUE(Entry.second);
|
|
EXPECT_TRUE(Entry.first->getKey() == "test2");
|
|
EXPECT_TRUE(Entry.first->second == nullptr);
|
|
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));
|
|
EXPECT_TRUE(Entry.first->second == nullptr);
|
|
});
|
|
|
|
// 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));
|
|
EXPECT_TRUE(Entry.first->second == nullptr);
|
|
});
|
|
}
|
|
|
|
} // anonymous namespace
|