mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-23 11:03:02 +08:00
f84a5fbee9
* 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>
50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
/*
|
|
* Copyright (C) 2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "shared/source/helpers/non_copyable_or_moveable.h"
|
|
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <stop_token>
|
|
#include <thread>
|
|
|
|
namespace NEO {
|
|
|
|
class GraphicsAllocation;
|
|
struct HostFunctionData;
|
|
|
|
class IHostFunctionWorker : public NonCopyableAndNonMovableClass {
|
|
public:
|
|
IHostFunctionWorker(bool skipHostFunctionExecution,
|
|
const std::function<void(GraphicsAllocation &)> &downloadAllocationImpl,
|
|
GraphicsAllocation *allocation,
|
|
HostFunctionData *data);
|
|
virtual ~IHostFunctionWorker() = 0;
|
|
|
|
virtual void start() = 0;
|
|
virtual void finish() = 0;
|
|
virtual void submit() noexcept = 0;
|
|
|
|
protected:
|
|
MOCKABLE_VIRTUAL bool runHostFunction(std::stop_token st) noexcept;
|
|
std::unique_ptr<std::jthread> worker;
|
|
std::mutex workerMutex;
|
|
|
|
private:
|
|
std::function<void(GraphicsAllocation &)> downloadAllocationImpl;
|
|
GraphicsAllocation *allocation = nullptr;
|
|
HostFunctionData *data = nullptr;
|
|
bool skipHostFunctionExecution = false;
|
|
};
|
|
|
|
static_assert(NonCopyableAndNonMovable<IHostFunctionWorker>);
|
|
|
|
} // namespace NEO
|