diff options
author | Dimitry Andric <dim@FreeBSD.org> | 2024-09-22 17:08:47 +0000 |
---|---|---|
committer | Dimitry Andric <dim@FreeBSD.org> | 2024-09-22 17:12:17 +0000 |
commit | 54521a2ff93ae06c95c31f79f89dc23c9b51c20b (patch) | |
tree | 339211b5d695b985d2011daa656f83ebf35129b7 /contrib/llvm-project/llvm/lib/Support | |
parent | f5a8f6f71a6845ad2eb3d8db22789ba6b1fd3fd8 (diff) |
Diffstat (limited to 'contrib/llvm-project/llvm/lib/Support')
-rw-r--r-- | contrib/llvm-project/llvm/lib/Support/Parallel.cpp | 49 |
1 files changed, 12 insertions, 37 deletions
diff --git a/contrib/llvm-project/llvm/lib/Support/Parallel.cpp b/contrib/llvm-project/llvm/lib/Support/Parallel.cpp index 9b14b05b5211..26da702969b7 100644 --- a/contrib/llvm-project/llvm/lib/Support/Parallel.cpp +++ b/contrib/llvm-project/llvm/lib/Support/Parallel.cpp @@ -12,7 +12,6 @@ #include "llvm/Support/Threading.h" #include <atomic> -#include <deque> #include <future> #include <thread> #include <vector> @@ -39,7 +38,7 @@ namespace { class Executor { public: virtual ~Executor() = default; - virtual void add(std::function<void()> func, bool Sequential = false) = 0; + virtual void add(std::function<void()> func) = 0; virtual size_t getThreadCount() const = 0; static Executor *getDefaultExecutor(); @@ -98,13 +97,10 @@ public: static void call(void *Ptr) { ((ThreadPoolExecutor *)Ptr)->stop(); } }; - void add(std::function<void()> F, bool Sequential = false) override { + void add(std::function<void()> F) override { { std::lock_guard<std::mutex> Lock(Mutex); - if (Sequential) - WorkQueueSequential.emplace_front(std::move(F)); - else - WorkQueue.emplace_back(std::move(F)); + WorkStack.push_back(std::move(F)); } Cond.notify_one(); } @@ -112,42 +108,23 @@ public: size_t getThreadCount() const override { return ThreadCount; } private: - bool hasSequentialTasks() const { - return !WorkQueueSequential.empty() && !SequentialQueueIsLocked; - } - - bool hasGeneralTasks() const { return !WorkQueue.empty(); } - void work(ThreadPoolStrategy S, unsigned ThreadID) { threadIndex = ThreadID; S.apply_thread_strategy(ThreadID); while (true) { std::unique_lock<std::mutex> Lock(Mutex); - Cond.wait(Lock, [&] { - return Stop || hasGeneralTasks() || hasSequentialTasks(); - }); + Cond.wait(Lock, [&] { return Stop || !WorkStack.empty(); }); if (Stop) break; - bool Sequential = hasSequentialTasks(); - if (Sequential) - SequentialQueueIsLocked = true; - else - assert(hasGeneralTasks()); - - auto &Queue = Sequential ? WorkQueueSequential : WorkQueue; - auto Task = std::move(Queue.back()); - Queue.pop_back(); + auto Task = std::move(WorkStack.back()); + WorkStack.pop_back(); Lock.unlock(); Task(); - if (Sequential) - SequentialQueueIsLocked = false; } } std::atomic<bool> Stop{false}; - std::atomic<bool> SequentialQueueIsLocked{false}; - std::deque<std::function<void()>> WorkQueue; - std::deque<std::function<void()>> WorkQueueSequential; + std::vector<std::function<void()>> WorkStack; std::mutex Mutex; std::condition_variable Cond; std::promise<void> ThreadsCreated; @@ -205,16 +182,14 @@ TaskGroup::~TaskGroup() { L.sync(); } -void TaskGroup::spawn(std::function<void()> F, bool Sequential) { +void TaskGroup::spawn(std::function<void()> F) { #if LLVM_ENABLE_THREADS if (Parallel) { L.inc(); - detail::Executor::getDefaultExecutor()->add( - [&, F = std::move(F)] { - F(); - L.dec(); - }, - Sequential); + detail::Executor::getDefaultExecutor()->add([&, F = std::move(F)] { + F(); + L.dec(); + }); return; } #endif |