diff --git a/include/sta/DispatchQueue.hh b/include/sta/DispatchQueue.hh index a8f6d60e..f28d94c5 100644 --- a/include/sta/DispatchQueue.hh +++ b/include/sta/DispatchQueue.hh @@ -24,6 +24,7 @@ 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. diff --git a/util/DispatchQueue.cc b/util/DispatchQueue.cc index c5f85e62..4e828c23 100644 --- a/util/DispatchQueue.cc +++ b/util/DispatchQueue.cc @@ -49,6 +49,12 @@ DispatchQueue::setThreadCount(size_t thread_count) } } +size_t +DispatchQueue::getThreadCount() const +{ + return threads_.size(); +} + void DispatchQueue::finishTasks() {