aboutsummaryrefslogtreecommitdiff
path: root/contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp')
-rw-r--r--contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp b/contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp
index 46a1990cd719..81926d8071b2 100644
--- a/contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp
+++ b/contrib/llvm-project/llvm/lib/Support/ThreadPool.cpp
@@ -72,6 +72,14 @@ void ThreadPool::wait() {
CompletionCondition.wait(LockGuard, [&] { return workCompletedUnlocked(); });
}
+bool ThreadPool::isWorkerThread() const {
+ llvm::thread::id CurrentThreadId = llvm::this_thread::get_id();
+ for (const llvm::thread &Thread : Threads)
+ if (CurrentThreadId == Thread.get_id())
+ return true;
+ return false;
+}
+
std::shared_future<void> ThreadPool::asyncImpl(TaskTy Task) {
/// Wrap the Task in a packaged_task to return a future object.
PackagedTaskTy PackagedTask(std::move(Task));