File size: 2,281 Bytes
8207382 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
// Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "thread_pool.h"
namespace PaddlePool {
constexpr size_t ThreadPool::WAIT_SECONDS;
ThreadPool::ThreadPool() : ThreadPool(Thread::hardware_concurrency()) {}
ThreadPool::ThreadPool(size_t maxThreads)
: quit_(false), currentThreads_(0), idleThreads_(0),
maxThreads_(maxThreads) {}
ThreadPool::~ThreadPool() {
{
MutexGuard guard(mutex_);
quit_ = true;
}
cv_.notify_all();
for (auto &elem : threads_) {
assert(elem.second.joinable());
elem.second.join();
}
}
size_t ThreadPool::threadsNum() const {
MutexGuard guard(mutex_);
return currentThreads_;
}
void ThreadPool::worker() {
while (true) {
Task task;
{
UniqueLock uniqueLock(mutex_);
++idleThreads_;
auto hasTimedout =
!cv_.wait_for(uniqueLock, std::chrono::seconds(WAIT_SECONDS),
[this]() { return quit_ || !tasks_.empty(); });
--idleThreads_;
if (tasks_.empty()) {
if (quit_) {
--currentThreads_;
return;
}
if (hasTimedout) {
--currentThreads_;
joinFinishedThreads();
finishedThreadIDs_.emplace(std::this_thread::get_id());
return;
}
}
task = std::move(tasks_.front());
tasks_.pop();
}
task();
}
}
void ThreadPool::joinFinishedThreads() {
while (!finishedThreadIDs_.empty()) {
auto id = std::move(finishedThreadIDs_.front());
finishedThreadIDs_.pop();
auto iter = threads_.find(id);
assert(iter != threads_.end());
assert(iter->second.joinable());
iter->second.join();
threads_.erase(iter);
}
}
} // namespace PaddlePool
|