mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-26 15:03:02 +08:00
feature: add host functions workers
* add common host function worker interface * add worker as a single thread per csr with 3 modes * add logic for waiting on internal tag, check gpu hang * if tag is in pending state, read callback data, run callback and signal completion * threads will exit the work loop once stop request is called in finish * add multi thread unit tests Related-To: NEO-14577 Signed-off-by: Kamil Kopryk <kamil.kopryk@intel.com>
This commit is contained in:
committed by
Compute-Runtime-Automation
parent
ae05e0551e
commit
f84a5fbee9
68
shared/source/command_stream/host_function_worker_atomic.cpp
Normal file
68
shared/source/command_stream/host_function_worker_atomic.cpp
Normal file
@@ -0,0 +1,68 @@
|
||||
/*
|
||||
* Copyright (C) 2025 Intel Corporation
|
||||
*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*
|
||||
*/
|
||||
|
||||
#include "shared/source/command_stream/host_function_worker_atomic.h"
|
||||
|
||||
#include "shared/source/command_stream/host_function.h"
|
||||
|
||||
namespace NEO {
|
||||
HostFunctionWorkerAtomic::HostFunctionWorkerAtomic(bool skipHostFunctionExecution,
|
||||
const std::function<void(GraphicsAllocation &)> &downloadAllocationImpl,
|
||||
GraphicsAllocation *allocation,
|
||||
HostFunctionData *data)
|
||||
: IHostFunctionWorker(skipHostFunctionExecution, downloadAllocationImpl, allocation, data) {
|
||||
}
|
||||
|
||||
HostFunctionWorkerAtomic::~HostFunctionWorkerAtomic() = default;
|
||||
|
||||
void HostFunctionWorkerAtomic::start() {
|
||||
|
||||
std::lock_guard<std::mutex> lg{workerMutex};
|
||||
if (!worker) {
|
||||
worker = std::make_unique<std::jthread>([this](std::stop_token st) {
|
||||
this->workerLoop(st);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
void HostFunctionWorkerAtomic::finish() {
|
||||
std::lock_guard<std::mutex> lg{workerMutex};
|
||||
if (worker) {
|
||||
worker->request_stop();
|
||||
pending.fetch_add(1u);
|
||||
pending.notify_one();
|
||||
worker.reset(nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
void HostFunctionWorkerAtomic::submit() noexcept {
|
||||
pending.fetch_add(1, std::memory_order_release);
|
||||
pending.notify_one();
|
||||
}
|
||||
|
||||
void HostFunctionWorkerAtomic::workerLoop(std::stop_token st) noexcept {
|
||||
|
||||
while (true) {
|
||||
|
||||
while (pending.load(std::memory_order_acquire) == 0) {
|
||||
pending.wait(0, std::memory_order_acquire);
|
||||
}
|
||||
|
||||
if (st.stop_requested()) {
|
||||
return;
|
||||
}
|
||||
|
||||
pending.fetch_sub(1, std::memory_order_acq_rel);
|
||||
|
||||
bool sucess = this->runHostFunction(st);
|
||||
if (!sucess) [[unlikely]] {
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace NEO
|
||||
Reference in New Issue
Block a user