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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
//===- unittests/Threading.cpp - Thread 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/Support/Threading.h"
#include "llvm/Config/llvm-config.h" // for LLVM_ENABLE_THREADS
#include "llvm/Support/thread.h"
#include "llvm/TargetParser/Host.h"
#include "llvm/TargetParser/Triple.h"
#include "gtest/gtest.h"
#include <atomic>
#include <condition_variable>
using namespace llvm;
namespace {
static bool isThreadingSupportedArchAndOS() {
#if LLVM_ENABLE_THREADS
Triple Host(Triple::normalize(sys::getProcessTriple()));
// Initially this is only testing detection of the number of
// physical cores, which is currently only supported/tested on
// some systems.
return (Host.isOSWindows() && llvm_is_multithreaded()) || Host.isOSDarwin() ||
(Host.isX86() && Host.isOSLinux()) ||
(Host.isOSLinux() && !Host.isAndroid()) ||
(Host.isSystemZ() && Host.isOSzOS()) || Host.isOSAIX();
#else
return false;
#endif
}
TEST(Threading, PhysicalConcurrency) {
auto Num = heavyweight_hardware_concurrency();
// Since Num is unsigned this will also catch us trying to
// return -1.
ASSERT_LE(Num.compute_thread_count(),
hardware_concurrency().compute_thread_count());
}
TEST(Threading, NumPhysicalCoresSupported) {
if (!isThreadingSupportedArchAndOS())
GTEST_SKIP();
int Num = get_physical_cores();
ASSERT_GT(Num, 0);
}
TEST(Threading, NumPhysicalCoresUnsupported) {
if (isThreadingSupportedArchAndOS())
GTEST_SKIP();
int Num = get_physical_cores();
ASSERT_EQ(Num, -1);
}
#if LLVM_ENABLE_THREADS
class Notification {
public:
void notify() {
{
std::lock_guard<std::mutex> Lock(M);
Notified = true;
// Broadcast with the lock held, so it's safe to destroy the Notification
// after wait() returns.
CV.notify_all();
}
}
bool wait() {
std::unique_lock<std::mutex> Lock(M);
using steady_clock = std::chrono::steady_clock;
auto Deadline = steady_clock::now() +
std::chrono::duration_cast<steady_clock::duration>(
std::chrono::duration<double>(5));
return CV.wait_until(Lock, Deadline, [this] { return Notified; });
}
private:
bool Notified = false;
mutable std::condition_variable CV;
mutable std::mutex M;
};
TEST(Threading, RunOnThreadSyncAsync) {
Notification ThreadStarted, ThreadAdvanced, ThreadFinished;
auto ThreadFunc = [&] {
ThreadStarted.notify();
ASSERT_TRUE(ThreadAdvanced.wait());
ThreadFinished.notify();
};
llvm::thread Thread(ThreadFunc);
Thread.detach();
ASSERT_TRUE(ThreadStarted.wait());
ThreadAdvanced.notify();
ASSERT_TRUE(ThreadFinished.wait());
}
TEST(Threading, RunOnThreadSync) {
std::atomic_bool Executed(false);
llvm::thread Thread(
[](void *Arg) { *static_cast<std::atomic_bool *>(Arg) = true; },
&Executed);
Thread.join();
ASSERT_EQ(Executed, true);
}
#if defined(__APPLE__)
TEST(Threading, AppleStackSize) {
llvm::thread Thread([] {
volatile unsigned char Var[8 * 1024 * 1024 - 10240];
Var[0] = 0xff;
ASSERT_EQ(Var[0], 0xff);
});
Thread.join();
}
#endif
#endif
} // namespace
|
