godotengine/godot
1
0
Fork 0
mirror of https://github.com/godotengine/godot.git synced 2025-11-05 12:10:55 +00:00
Code Releases Wiki Activity

Merge pull request #104237 from Ivorforce/optimize-thread-pool-waiting

Browse Source 
Optimize thread pools by avoiding needless locks and unlocks of the `task_mutex`.
This commit is contained in:
Rémi Verschelde
2025-03-21 13:03:37 +01:00
parent f51ea67fb5 9d30732c5b
commit 1f682b6554
1 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
core/object/worker_thread_pool.cpp
View File

@@ -186,28 +186,36 @@ void WorkerThreadPool::_thread_function(void *p_user) {
while (true) {
Task *task_to_process = nullptr;
{
// Create the lock outside the inner loop so it isn't needlessly unlocked and relocked
// when no task was found to process, and the loop is re-entered.
MutexLock lock(thread_data->pool->task_mutex);
while (true) {
bool exit = thread_data->pool->_handle_runlevel(thread_data, lock);
if (unlikely(exit)) {
break;
return;
}
thread_data->signaled = false;
if (thread_data->pool->task_queue.first()) {
if (!thread_data->pool->task_queue.first()) {
// There wasn't a task available yet.
// Let's wait for the next notification, then recheck.
thread_data->cond_var.wait(lock);
continue;
}
// Got a task to process! Remove it from the queue, then break into the task handling section.
task_to_process = thread_data->pool->task_queue.first()->self();
thread_data->pool->task_queue.remove(thread_data->pool->task_queue.first());
} else {
thread_data->cond_var.wait(lock);
break;
}
}
if (task_to_process) {
DEV_ASSERT(task_to_process);
thread_data->pool->_process_task(task_to_process);
}
}
}
void WorkerThreadPool::_post_tasks(Task **p_tasks, uint32_t p_count, bool p_high_priority, MutexLock<BinaryMutex> &p_lock) {
// Fall back to processing on the calling thread if there are no worker threads.