c59ebe4c4c
These calls were left out of 4d7cea3d2e8. In the InPlaceDispatcher test case the operation is a no-op, but it's good form to include it. In the DynamicThreadPoolTaskDispatcher test the shutdown call is required to ensure that we don't exit the test (and tear down the dispatcher) before the thread running the dispatch has completed.
36 lines
1.1 KiB
C++
36 lines
1.1 KiB
C++
//===----------- TaskDispatchTest.cpp - Test TaskDispatch APIs ------------===//
|
|
//
|
|
// 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/TaskDispatch.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <future>
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::orc;
|
|
|
|
TEST(InPlaceTaskDispatchTest, GenericNamedTask) {
|
|
auto D = std::make_unique<InPlaceTaskDispatcher>();
|
|
bool B = false;
|
|
D->dispatch(makeGenericNamedTask([&]() { B = true; }));
|
|
EXPECT_TRUE(B);
|
|
D->shutdown();
|
|
}
|
|
|
|
#if LLVM_ENABLE_THREADS
|
|
TEST(DynamicThreadPoolDispatchTest, GenericNamedTask) {
|
|
auto D = std::make_unique<DynamicThreadPoolTaskDispatcher>();
|
|
std::promise<bool> P;
|
|
auto F = P.get_future();
|
|
D->dispatch(makeGenericNamedTask(
|
|
[P = std::move(P)]() mutable { P.set_value(true); }));
|
|
EXPECT_TRUE(F.get());
|
|
D->shutdown();
|
|
}
|
|
#endif
|