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
|