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
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
//===- PerThreadBumpPtrAllocatorTest.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/Support/PerThreadBumpPtrAllocator.h"
|
|
#include "llvm/Support/Parallel.h"
|
|
#include "gtest/gtest.h"
|
|
#include <cstdlib>
|
|
|
|
using namespace llvm;
|
|
using namespace parallel;
|
|
|
|
namespace {
|
|
|
|
TEST(PerThreadBumpPtrAllocatorTest, Simple) {
|
|
PerThreadBumpPtrAllocator Allocator;
|
|
|
|
parallel::TaskGroup tg;
|
|
|
|
tg.spawn([&]() {
|
|
uint64_t *Var =
|
|
(uint64_t *)Allocator.Allocate(sizeof(uint64_t), alignof(uint64_t));
|
|
*Var = 0xFE;
|
|
EXPECT_EQ(0xFEul, *Var);
|
|
EXPECT_EQ(sizeof(uint64_t), Allocator.getBytesAllocated());
|
|
EXPECT_TRUE(Allocator.getBytesAllocated() <= Allocator.getTotalMemory());
|
|
|
|
PerThreadBumpPtrAllocator Allocator2(std::move(Allocator));
|
|
|
|
EXPECT_EQ(sizeof(uint64_t), Allocator2.getBytesAllocated());
|
|
EXPECT_TRUE(Allocator2.getBytesAllocated() <= Allocator2.getTotalMemory());
|
|
|
|
EXPECT_EQ(0xFEul, *Var);
|
|
});
|
|
}
|
|
|
|
TEST(PerThreadBumpPtrAllocatorTest, ParallelAllocation) {
|
|
PerThreadBumpPtrAllocator Allocator;
|
|
|
|
static size_t constexpr NumAllocations = 1000;
|
|
|
|
parallelFor(0, NumAllocations, [&](size_t Idx) {
|
|
uint64_t *ptr =
|
|
(uint64_t *)Allocator.Allocate(sizeof(uint64_t), alignof(uint64_t));
|
|
*ptr = Idx;
|
|
});
|
|
|
|
EXPECT_EQ(sizeof(uint64_t) * NumAllocations, Allocator.getBytesAllocated());
|
|
EXPECT_EQ(Allocator.getNumberOfAllocators(), parallel::getThreadCount());
|
|
}
|
|
|
|
} // anonymous namespace
|