mirror of
https://github.com/The-OpenROAD-Project/OpenSTA.git
synced 2026-05-30 00:24:12 +08:00
7ac4a47db1
I'm looking to try to have unified thread pool with OpenSTA and OpenROAD and having this method would make it easier to understand how much parallelism is available in the pool. Signed-off-by: Ethan Mahintorabi <ethanmoon@google.com>
53 lines
1.3 KiB
C++
53 lines
1.3 KiB
C++
// Author Phillip Johnston
|
|
// Licensed under CC0 1.0 Universal
|
|
// https://github.com/embeddedartistry/embedded-resources/blob/master/examples/cpp/dispatch.cpp
|
|
// https://embeddedartistry.com/blog/2017/2/1/dispatch-queues?rq=dispatch
|
|
|
|
#pragma once
|
|
|
|
#include <thread>
|
|
#include <functional>
|
|
#include <vector>
|
|
#include <cstdint>
|
|
#include <queue>
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include <atomic>
|
|
|
|
namespace sta {
|
|
|
|
class DispatchQueue
|
|
{
|
|
typedef std::function<void(int thread)> fp_t;
|
|
|
|
public:
|
|
DispatchQueue(size_t thread_cnt);
|
|
~DispatchQueue();
|
|
void setThreadCount(size_t thread_count);
|
|
size_t getThreadCount() const;
|
|
// Dispatch and copy.
|
|
void dispatch(const fp_t& op);
|
|
// Dispatch and move.
|
|
void dispatch(fp_t&& op);
|
|
void finishTasks();
|
|
|
|
// Deleted operations
|
|
DispatchQueue(const DispatchQueue& rhs) = delete;
|
|
DispatchQueue& operator=(const DispatchQueue& rhs) = delete;
|
|
DispatchQueue(DispatchQueue&& rhs) = delete;
|
|
DispatchQueue& operator=(DispatchQueue&& rhs) = delete;
|
|
|
|
private:
|
|
void dispatch_thread_handler(size_t i);
|
|
void terminateThreads();
|
|
|
|
std::mutex lock_;
|
|
std::vector<std::thread> threads_;
|
|
std::queue<fp_t> q_;
|
|
std::condition_variable cv_;
|
|
std::atomic<size_t> pending_task_count_;
|
|
bool quit_ = false;
|
|
};
|
|
|
|
} // namespace
|