213666f804
This type has been moved up into the llvm::orc::shared namespace. This type was originally put in the detail:: namespace on the assumption that few (if any) LLVM source files would need to use it. In practice it has been needed in many places, and will continue to be needed until/unless OrcTargetProcess is fully merged into the ORC runtime.
85 lines
2.9 KiB
C++
85 lines
2.9 KiB
C++
//===---------------- SimpleExecutorMemoryManagerTest.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/ExecutionEngine/Orc/TargetProcess/SimpleExecutorMemoryManager.h"
|
|
#include "llvm/Testing/Support/Error.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <limits>
|
|
#include <vector>
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::orc;
|
|
using namespace llvm::orc::shared;
|
|
using namespace llvm::orc::rt_bootstrap;
|
|
|
|
namespace {
|
|
|
|
orc::shared::CWrapperFunctionResult incrementWrapper(const char *ArgData,
|
|
size_t ArgSize) {
|
|
return WrapperFunction<void(SPSExecutorAddr)>::handle(
|
|
ArgData, ArgSize, [](ExecutorAddr A) { *A.toPtr<int *>() += 1; })
|
|
.release();
|
|
}
|
|
|
|
TEST(SimpleExecutorMemoryManagerTest, AllocFinalizeFree) {
|
|
SimpleExecutorMemoryManager MemMgr;
|
|
|
|
constexpr unsigned AllocSize = 16384;
|
|
auto Mem = MemMgr.allocate(AllocSize);
|
|
EXPECT_THAT_ERROR(Mem.takeError(), Succeeded());
|
|
|
|
std::string HW = "Hello, world!";
|
|
|
|
int FinalizeCounter = 0;
|
|
auto FinalizeCounterAddrArgBuffer =
|
|
orc::shared::detail::serializeViaSPSToWrapperFunctionResult<
|
|
SPSArgList<SPSExecutorAddr>>(ExecutorAddr::fromPtr(&FinalizeCounter));
|
|
|
|
int DeallocateCounter = 0;
|
|
auto DeallocateCounterAddrArgBuffer =
|
|
orc::shared::detail::serializeViaSPSToWrapperFunctionResult<
|
|
SPSArgList<SPSExecutorAddr>>(
|
|
ExecutorAddr::fromPtr(&DeallocateCounter));
|
|
|
|
tpctypes::FinalizeRequest FR;
|
|
FR.Segments.push_back(
|
|
tpctypes::SegFinalizeRequest{tpctypes::WPF_Read | tpctypes::WPF_Write,
|
|
*Mem,
|
|
AllocSize,
|
|
{HW.data(), HW.size() + 1}});
|
|
FR.Actions.push_back(
|
|
{/* Finalize: */
|
|
{ExecutorAddr::fromPtr(incrementWrapper),
|
|
{ExecutorAddr::fromPtr(FinalizeCounterAddrArgBuffer.data()),
|
|
ExecutorAddrDiff(FinalizeCounterAddrArgBuffer.size())}},
|
|
/* Deallocate: */
|
|
{ExecutorAddr::fromPtr(incrementWrapper),
|
|
{ExecutorAddr::fromPtr(DeallocateCounterAddrArgBuffer.data()),
|
|
ExecutorAddrDiff(DeallocateCounterAddrArgBuffer.size())}}});
|
|
|
|
EXPECT_EQ(FinalizeCounter, 0);
|
|
EXPECT_EQ(DeallocateCounter, 0);
|
|
|
|
auto FinalizeErr = MemMgr.finalize(FR);
|
|
EXPECT_THAT_ERROR(std::move(FinalizeErr), Succeeded());
|
|
|
|
EXPECT_EQ(FinalizeCounter, 1);
|
|
EXPECT_EQ(DeallocateCounter, 0);
|
|
|
|
EXPECT_EQ(HW, std::string(Mem->toPtr<const char *>()));
|
|
|
|
auto DeallocateErr = MemMgr.deallocate({*Mem});
|
|
EXPECT_THAT_ERROR(std::move(DeallocateErr), Succeeded());
|
|
|
|
EXPECT_EQ(FinalizeCounter, 1);
|
|
EXPECT_EQ(DeallocateCounter, 1);
|
|
}
|
|
|
|
} // namespace
|