initial commit, 4.5 stable
Some checks failed
🔗 GHA / 📊 Static checks (push) Has been cancelled
🔗 GHA / 🤖 Android (push) Has been cancelled
🔗 GHA / 🍏 iOS (push) Has been cancelled
🔗 GHA / 🐧 Linux (push) Has been cancelled
🔗 GHA / 🍎 macOS (push) Has been cancelled
🔗 GHA / 🏁 Windows (push) Has been cancelled
🔗 GHA / 🌐 Web (push) Has been cancelled
Some checks failed
🔗 GHA / 📊 Static checks (push) Has been cancelled
🔗 GHA / 🤖 Android (push) Has been cancelled
🔗 GHA / 🍏 iOS (push) Has been cancelled
🔗 GHA / 🐧 Linux (push) Has been cancelled
🔗 GHA / 🍎 macOS (push) Has been cancelled
🔗 GHA / 🏁 Windows (push) Has been cancelled
🔗 GHA / 🌐 Web (push) Has been cancelled
This commit is contained in:
175
tests/core/threads/test_worker_thread_pool.h
Normal file
175
tests/core/threads/test_worker_thread_pool.h
Normal file
@@ -0,0 +1,175 @@
|
||||
/**************************************************************************/
|
||||
/* test_worker_thread_pool.h */
|
||||
/**************************************************************************/
|
||||
/* This file is part of: */
|
||||
/* GODOT ENGINE */
|
||||
/* https://godotengine.org */
|
||||
/**************************************************************************/
|
||||
/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
|
||||
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
|
||||
/* */
|
||||
/* Permission is hereby granted, free of charge, to any person obtaining */
|
||||
/* a copy of this software and associated documentation files (the */
|
||||
/* "Software"), to deal in the Software without restriction, including */
|
||||
/* without limitation the rights to use, copy, modify, merge, publish, */
|
||||
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
||||
/* permit persons to whom the Software is furnished to do so, subject to */
|
||||
/* the following conditions: */
|
||||
/* */
|
||||
/* The above copyright notice and this permission notice shall be */
|
||||
/* included in all copies or substantial portions of the Software. */
|
||||
/* */
|
||||
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
||||
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
||||
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
|
||||
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
||||
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
||||
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
||||
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
||||
/**************************************************************************/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "core/object/worker_thread_pool.h"
|
||||
|
||||
#include "tests/test_macros.h"
|
||||
|
||||
namespace TestWorkerThreadPool {
|
||||
|
||||
static LocalVector<SafeNumeric<int>> counter;
|
||||
static SafeFlag exit;
|
||||
|
||||
static void static_test(void *p_arg) {
|
||||
counter[(uint64_t)p_arg].increment();
|
||||
counter[0].add(2);
|
||||
}
|
||||
static void static_callable_test() {
|
||||
counter[0].sub(2);
|
||||
}
|
||||
TEST_CASE("[WorkerThreadPool] Process threads using individual tasks") {
|
||||
for (int iterations = 0; iterations < 500; iterations++) {
|
||||
const int count = Math::pow(2.0f, Math::random(0.0f, 5.0f));
|
||||
const bool low_priority = Math::rand() % 2;
|
||||
|
||||
LocalVector<WorkerThreadPool::TaskID> tasks1;
|
||||
LocalVector<WorkerThreadPool::TaskID> tasks2;
|
||||
tasks1.resize(count);
|
||||
tasks2.resize(count);
|
||||
|
||||
counter.clear();
|
||||
counter.resize(count);
|
||||
for (int i = 0; i < count; i++) {
|
||||
tasks1[i] = WorkerThreadPool::get_singleton()->add_native_task(static_test, (void *)(uintptr_t)i, low_priority);
|
||||
tasks2[i] = WorkerThreadPool::get_singleton()->add_task(callable_mp_static(static_callable_test), !low_priority);
|
||||
}
|
||||
for (int i = 0; i < count; i++) {
|
||||
WorkerThreadPool::get_singleton()->wait_for_task_completion(tasks1[i]);
|
||||
WorkerThreadPool::get_singleton()->wait_for_task_completion(tasks2[i]);
|
||||
}
|
||||
|
||||
bool all_run_once = true;
|
||||
for (int i = 0; i < count; i++) {
|
||||
//Reduce number of check messages
|
||||
all_run_once &= counter[i].get() == 1;
|
||||
}
|
||||
CHECK(all_run_once);
|
||||
}
|
||||
}
|
||||
|
||||
static void static_group_test(void *p_arg, uint32_t p_index) {
|
||||
counter[p_index].increment();
|
||||
counter[0].add((uintptr_t)p_arg);
|
||||
}
|
||||
static void static_callable_group_test(uint32_t p_index) {
|
||||
counter[p_index].increment();
|
||||
counter[0].sub(2);
|
||||
}
|
||||
TEST_CASE("[WorkerThreadPool] Process elements using group tasks") {
|
||||
for (int iterations = 0; iterations < 500; iterations++) {
|
||||
const int count = Math::pow(2.0f, Math::random(0.0f, 5.0f));
|
||||
const int tasks = Math::pow(2.0f, Math::random(0.0f, 5.0f));
|
||||
const bool low_priority = Math::rand() % 2;
|
||||
|
||||
counter.clear();
|
||||
counter.resize(count);
|
||||
WorkerThreadPool::GroupID group1 = WorkerThreadPool::get_singleton()->add_native_group_task(static_group_test, (void *)2, count, tasks, !low_priority);
|
||||
WorkerThreadPool::GroupID group2 = WorkerThreadPool::get_singleton()->add_group_task(callable_mp_static(static_callable_group_test), count, tasks, low_priority);
|
||||
WorkerThreadPool::get_singleton()->wait_for_group_task_completion(group1);
|
||||
WorkerThreadPool::get_singleton()->wait_for_group_task_completion(group2);
|
||||
|
||||
bool all_run_once = true;
|
||||
for (int i = 0; i < count; i++) {
|
||||
//Reduce number of check messages
|
||||
all_run_once &= counter[i].get() == 2;
|
||||
}
|
||||
CHECK(all_run_once);
|
||||
}
|
||||
}
|
||||
|
||||
static void static_test_daemon(void *p_arg) {
|
||||
while (!exit.is_set()) {
|
||||
counter[0].add(1);
|
||||
WorkerThreadPool::get_singleton()->yield();
|
||||
}
|
||||
}
|
||||
|
||||
static void static_busy_task(void *p_arg) {
|
||||
while (!exit.is_set()) {
|
||||
OS::get_singleton()->delay_usec(1);
|
||||
}
|
||||
}
|
||||
|
||||
static void static_legit_task(void *p_arg) {
|
||||
*((bool *)p_arg) = counter[0].get() > 0;
|
||||
counter[1].add(1);
|
||||
}
|
||||
|
||||
TEST_CASE("[WorkerThreadPool] Run a yielding daemon as the only hope for other tasks to run") {
|
||||
exit.clear();
|
||||
counter.clear();
|
||||
counter.resize(2);
|
||||
|
||||
WorkerThreadPool::TaskID daemon_task_id = WorkerThreadPool::get_singleton()->add_native_task(static_test_daemon, nullptr, true);
|
||||
|
||||
int num_threads = WorkerThreadPool::get_singleton()->get_thread_count();
|
||||
|
||||
// Keep all the other threads busy.
|
||||
LocalVector<WorkerThreadPool::TaskID> task_ids;
|
||||
for (int i = 0; i < num_threads - 1; i++) {
|
||||
task_ids.push_back(WorkerThreadPool::get_singleton()->add_native_task(static_busy_task, nullptr, true));
|
||||
}
|
||||
|
||||
LocalVector<WorkerThreadPool::TaskID> legit_task_ids;
|
||||
LocalVector<bool> legit_task_needed_yield;
|
||||
int legit_tasks_count = num_threads * 4;
|
||||
legit_task_needed_yield.resize(legit_tasks_count);
|
||||
for (int i = 0; i < legit_tasks_count; i++) {
|
||||
legit_task_needed_yield[i] = false;
|
||||
task_ids.push_back(WorkerThreadPool::get_singleton()->add_native_task(static_legit_task, &legit_task_needed_yield[i], i >= legit_tasks_count / 2));
|
||||
}
|
||||
|
||||
while (counter[1].get() != legit_tasks_count) {
|
||||
OS::get_singleton()->delay_usec(1);
|
||||
}
|
||||
|
||||
exit.set();
|
||||
for (uint32_t i = 0; i < task_ids.size(); i++) {
|
||||
WorkerThreadPool::get_singleton()->wait_for_task_completion(task_ids[i]);
|
||||
}
|
||||
WorkerThreadPool::get_singleton()->notify_yield_over(daemon_task_id);
|
||||
WorkerThreadPool::get_singleton()->wait_for_task_completion(daemon_task_id);
|
||||
|
||||
CHECK_MESSAGE(counter[0].get() > 0, "Daemon task should have looped at least once.");
|
||||
CHECK_MESSAGE(counter[1].get() == legit_tasks_count, "All legit tasks should have been able to run.");
|
||||
|
||||
bool all_needed_yield = true;
|
||||
for (int i = 0; i < legit_tasks_count; i++) {
|
||||
if (!legit_task_needed_yield[i]) {
|
||||
all_needed_yield = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
CHECK_MESSAGE(all_needed_yield, "All legit tasks should have needed the daemon yielding to run.");
|
||||
}
|
||||
|
||||
} // namespace TestWorkerThreadPool
|
Reference in New Issue
Block a user