1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
| //========- unittests/Support/TaskQueue.cpp - TaskQueue.h tests ------========//
//
// 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/Config/llvm-config.h"
#if LLVM_ENABLE_THREADS
#include "llvm/Support/TaskQueue.h"
#include "gtest/gtest.h"
using namespace llvm;
class TaskQueueTest : public testing::Test {
protected:
TaskQueueTest() {}
};
TEST_F(TaskQueueTest, OrderedFutures) {
ThreadPool TP(1);
TaskQueue TQ(TP);
std::atomic<int> X{ 0 };
std::atomic<int> Y{ 0 };
std::atomic<int> Z{ 0 };
std::mutex M1, M2, M3;
std::unique_lock<std::mutex> L1(M1);
std::unique_lock<std::mutex> L2(M2);
std::unique_lock<std::mutex> L3(M3);
std::future<void> F1 = TQ.async([&] {
std::unique_lock<std::mutex> Lock(M1);
++X;
});
std::future<void> F2 = TQ.async([&] {
std::unique_lock<std::mutex> Lock(M2);
++Y;
});
std::future<void> F3 = TQ.async([&] {
std::unique_lock<std::mutex> Lock(M3);
++Z;
});
L1.unlock();
F1.wait();
ASSERT_EQ(1, X);
ASSERT_EQ(0, Y);
ASSERT_EQ(0, Z);
L2.unlock();
F2.wait();
ASSERT_EQ(1, X);
ASSERT_EQ(1, Y);
ASSERT_EQ(0, Z);
L3.unlock();
F3.wait();
ASSERT_EQ(1, X);
ASSERT_EQ(1, Y);
ASSERT_EQ(1, Z);
}
TEST_F(TaskQueueTest, UnOrderedFutures) {
ThreadPool TP(1);
TaskQueue TQ(TP);
std::atomic<int> X{ 0 };
std::atomic<int> Y{ 0 };
std::atomic<int> Z{ 0 };
std::mutex M;
std::unique_lock<std::mutex> Lock(M);
std::future<void> F1 = TQ.async([&] { ++X; });
std::future<void> F2 = TQ.async([&] { ++Y; });
std::future<void> F3 = TQ.async([&M, &Z] {
std::unique_lock<std::mutex> Lock(M);
++Z;
});
F2.wait();
ASSERT_EQ(1, X);
ASSERT_EQ(1, Y);
ASSERT_EQ(0, Z);
Lock.unlock();
F3.wait();
ASSERT_EQ(1, X);
ASSERT_EQ(1, Y);
ASSERT_EQ(1, Z);
}
TEST_F(TaskQueueTest, FutureWithReturnValue) {
ThreadPool TP(1);
TaskQueue TQ(TP);
std::future<std::string> F1 = TQ.async([&] { return std::string("Hello"); });
std::future<int> F2 = TQ.async([&] { return 42; });
ASSERT_EQ(42, F2.get());
ASSERT_EQ("Hello", F1.get());
}
#endif
|