RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2025-01-10 22:00:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

ThreadPool unittest: reimplement concurrency test, deterministically this time.

Browse Source 
Follow-up to r256056.

From: Mehdi Amini <mehdi.amini@apple.com>
llvm-svn: 256087
This commit is contained in:
Mehdi Amini 2015-12-19 05:12:07 +00:00
parent 4165eed18e
commit 6c98bbb0ba
1 changed files with 36 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
unittests/Support/ThreadPool.cpp
View File

@ -54,6 +54,18 @@ protected:
UnsupportedArchs.push_back(Triple::ppc64le);
UnsupportedArchs.push_back(Triple::ppc64);
}
/// Make sure this thread not progress faster than the main thread.
void waitForMainThread() {
while (!MainThreadReady) {
std::unique_lock<std::mutex> LockGuard(WaitMainThreadMutex);
WaitMainThread.wait(LockGuard, [&] { return MainThreadReady; });
}
}
std::condition_variable WaitMainThread;
std::mutex WaitMainThreadMutex;
bool MainThreadReady;
};
#define CHECK_UNSUPPORTED() \
@ -68,12 +80,17 @@ TEST_F(ThreadPoolTest, AsyncBarrier) {
std::atomic_int checked_in{0};
MainThreadReady = false;
ThreadPool Pool;
for (size_t i = 0; i < 5; ++i) {
Pool.async([&checked_in, i] {
Pool.async([this, &checked_in, i] {
waitForMainThread();
++checked_in;
});
}
ASSERT_EQ(0, checked_in);
MainThreadReady = true;
WaitMainThread.notify_all();
Pool.wait();
ASSERT_EQ(5, checked_in);
}
@ -97,10 +114,15 @@ TEST_F(ThreadPoolTest, Async) {
CHECK_UNSUPPORTED();
ThreadPool Pool;
std::atomic_int i{0};
Pool.async([&i] {
MainThreadReady = false;
Pool.async([this, &i] {
waitForMainThread();
++i;
});
Pool.async([&i] { ++i; });
ASSERT_NE(2, i.load());
MainThreadReady = true;
WaitMainThread.notify_all();
Pool.wait();
ASSERT_EQ(2, i.load());
}
@ -109,11 +131,16 @@ TEST_F(ThreadPoolTest, GetFuture) {
CHECK_UNSUPPORTED();
ThreadPool Pool;
std::atomic_int i{0};
Pool.async([&i] {
MainThreadReady = false;
Pool.async([this, &i] {
waitForMainThread();
++i;
});
// Force the future using get()
Pool.async([&i] { ++i; }).get();
ASSERT_NE(2, i.load());
MainThreadReady = true;
WaitMainThread.notify_all();
Pool.wait();
ASSERT_EQ(2, i.load());
}
@ -122,14 +149,18 @@ TEST_F(ThreadPoolTest, PoolDestruction) {
CHECK_UNSUPPORTED();
// Test that we are waiting on destruction
std::atomic_int checked_in{0};
{
MainThreadReady = false;
ThreadPool Pool;
for (size_t i = 0; i < 5; ++i) {
Pool.async([&checked_in, i] {
Pool.async([this, &checked_in, i] {
waitForMainThread();
++checked_in;
});
}
ASSERT_EQ(0, checked_in);
MainThreadReady = true;
WaitMainThread.notify_all();
}
ASSERT_EQ(5, checked_in);
}