1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
/*
* 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
|
