mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-24 04:12:57 +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>
64 lines
2.2 KiB
C++
64 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <functional>
|
|
|
|
namespace NEO {
|
|
|
|
class LinearStream;
|
|
class CommandStreamReceiver;
|
|
class IHostFunctionWorker;
|
|
class GraphicsAllocation;
|
|
|
|
struct HostFunctionData {
|
|
volatile uint64_t *entry = nullptr;
|
|
volatile uint64_t *userData = nullptr;
|
|
volatile uint32_t *internalTag = nullptr;
|
|
};
|
|
|
|
enum class HostFunctionTagStatus : uint32_t {
|
|
completed = 0,
|
|
pending = 1
|
|
};
|
|
|
|
struct HostFunctionHelper {
|
|
|
|
constexpr static size_t entryOffset = offsetof(HostFunctionData, entry);
|
|
constexpr static size_t userDataOffset = offsetof(HostFunctionData, userData);
|
|
constexpr static size_t internalTagOffset = offsetof(HostFunctionData, internalTag);
|
|
|
|
template <typename GfxFamily>
|
|
static void programHostFunction(LinearStream &commandStream, const HostFunctionData &hostFunctionData, uint64_t userHostFunctionAddress, uint64_t userDataAddress);
|
|
|
|
template <typename GfxFamily>
|
|
static void programHostFunctionAddress(LinearStream *commandStream, void *cmdBuffer, const HostFunctionData &hostFunctionData, uint64_t userHostFunctionAddress);
|
|
|
|
template <typename GfxFamily>
|
|
static void programHostFunctionUserData(LinearStream *commandStream, void *cmdBuffer, const HostFunctionData &hostFunctionData, uint64_t userDataAddress);
|
|
|
|
template <typename GfxFamily>
|
|
static void programSignalHostFunctionStart(LinearStream *commandStream, void *cmdBuffer, const HostFunctionData &hostFunctionData);
|
|
|
|
template <typename GfxFamily>
|
|
static void programWaitForHostFunctionCompletion(LinearStream *commandStream, void *cmdBuffer, const HostFunctionData &hostFunctionData);
|
|
};
|
|
|
|
namespace HostFunctionFactory {
|
|
IHostFunctionWorker *createHostFunctionWorker(int32_t hostFunctionWorkerMode,
|
|
bool isAubMode,
|
|
const std::function<void(GraphicsAllocation &)> &downloadAllocationImpl,
|
|
GraphicsAllocation *allocation,
|
|
HostFunctionData *data);
|
|
|
|
}
|
|
|
|
} // namespace NEO
|