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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
/*
* Copyright (C) 2024-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "level_zero/core/source/driver/extension_function_address.h"
#include "level_zero/api/core/preview_api_entrypoints.h"
#include "level_zero/api/extensions/public/ze_exp_ext.h"
#include "level_zero/driver_experimental/mcl_ext/zex_mutable_cmdlist_ext.h"
#include "level_zero/driver_experimental/zex_api.h"
#include "level_zero/driver_experimental/zex_cmdlist.h"
#include "level_zero/driver_experimental/zex_context.h"
#include "level_zero/ze_intel_gpu.h"
#include "level_zero/zet_intel_gpu_metric.h"
#include <cstring>
namespace L0 {
void *ExtensionFunctionAddressHelper::getExtensionFunctionAddress(const std::string &functionName) {
#define RETURN_FUNC_PTR_IF_EXIST(name) \
{ \
if (functionName == #name) { \
void *ret = ((void *)(name)); \
return ret; \
} \
}
#define RETURN_L0_FUNC_PTR_IF_EXIST(name) \
{ \
if (functionName == #name) { \
void *ret = ((void *)(L0::name)); \
return ret; \
} \
}
RETURN_FUNC_PTR_IF_EXIST(zexDriverImportExternalPointer);
RETURN_FUNC_PTR_IF_EXIST(zexDriverReleaseImportedPointer);
RETURN_FUNC_PTR_IF_EXIST(zexDriverGetHostPointerBaseAddress);
RETURN_L0_FUNC_PTR_IF_EXIST(zeDriverGetDefaultContext);
RETURN_L0_FUNC_PTR_IF_EXIST(zerGetDefaultContext);
RETURN_L0_FUNC_PTR_IF_EXIST(zerGetLastErrorDescription);
RETURN_L0_FUNC_PTR_IF_EXIST(zerTranslateDeviceHandleToIdentifier);
RETURN_L0_FUNC_PTR_IF_EXIST(zerTranslateIdentifierToDeviceHandle);
RETURN_L0_FUNC_PTR_IF_EXIST(zeDeviceSynchronize);
RETURN_FUNC_PTR_IF_EXIST(zeDeviceGetPriorityLevels);
RETURN_L0_FUNC_PTR_IF_EXIST(zeCommandListAppendLaunchKernelWithArguments);
RETURN_L0_FUNC_PTR_IF_EXIST(zeCommandListAppendLaunchKernelWithParameters);
RETURN_FUNC_PTR_IF_EXIST(zexKernelGetBaseAddress);
RETURN_FUNC_PTR_IF_EXIST(zexKernelGetArgumentSize);
RETURN_FUNC_PTR_IF_EXIST(zexKernelGetArgumentType);
RETURN_FUNC_PTR_IF_EXIST(zeIntelKernelGetBinaryExp);
RETURN_FUNC_PTR_IF_EXIST(zexMemGetIpcHandles);
RETURN_FUNC_PTR_IF_EXIST(zexMemOpenIpcHandles);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendWaitOnMemory);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendWaitOnMemory64);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendWriteToMemory);
RETURN_FUNC_PTR_IF_EXIST(zexCounterBasedEventCreate);
RETURN_FUNC_PTR_IF_EXIST(zexEventGetDeviceAddress);
RETURN_FUNC_PTR_IF_EXIST(zexCounterBasedEventCreate2);
RETURN_FUNC_PTR_IF_EXIST(zexCounterBasedEventGetIpcHandle);
RETURN_FUNC_PTR_IF_EXIST(zexCounterBasedEventOpenIpcHandle);
RETURN_FUNC_PTR_IF_EXIST(zexCounterBasedEventCloseIpcHandle);
RETURN_FUNC_PTR_IF_EXIST(zexDeviceGetAggregatedCopyOffloadIncrementValue);
RETURN_FUNC_PTR_IF_EXIST(zeMemGetPitchFor2dImage);
RETURN_FUNC_PTR_IF_EXIST(zeImageGetDeviceOffsetExp);
RETURN_FUNC_PTR_IF_EXIST(zeIntelGetDriverVersionString);
RETURN_FUNC_PTR_IF_EXIST(zeIntelMediaCommunicationCreate);
RETURN_FUNC_PTR_IF_EXIST(zeIntelMediaCommunicationDestroy);
RETURN_FUNC_PTR_IF_EXIST(zexMemFreeRegisterCallbackExt);
RETURN_FUNC_PTR_IF_EXIST(zexIntelAllocateNetworkInterrupt);
RETURN_FUNC_PTR_IF_EXIST(zexIntelReleaseNetworkInterrupt);
RETURN_FUNC_PTR_IF_EXIST(zetIntelCommandListAppendMarkerExp);
RETURN_FUNC_PTR_IF_EXIST(zetDeviceEnableMetricsExp);
RETURN_FUNC_PTR_IF_EXIST(zetDeviceDisableMetricsExp);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendHostFunction);
// mutable command list extension
RETURN_FUNC_PTR_IF_EXIST(zeCommandListGetNextCommandIdExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListUpdateMutableCommandsExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListUpdateMutableCommandSignalEventExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListUpdateMutableCommandWaitEventsExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListGetNextCommandIdWithKernelsExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListUpdateMutableCommandKernelsExp);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListGetVariable);
RETURN_FUNC_PTR_IF_EXIST(zexKernelSetArgumentVariable);
RETURN_FUNC_PTR_IF_EXIST(zexVariableSetValue);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListGetLabel);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListSetLabel);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendJump);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendLoadRegVariable);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendStoreRegVariable);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListTempMemSetEleCount);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListTempMemGetSize);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListTempMemSet);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListGetNativeBinary);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListLoadNativeBinary);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendVariableLaunchKernel);
RETURN_FUNC_PTR_IF_EXIST(zexKernelSetVariableGroupSize);
RETURN_FUNC_PTR_IF_EXIST(zexVariableGetInfo);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListGetVariablesList);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendMILoadRegReg);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendMILoadRegMem);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendMILoadRegImm);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendMIStoreRegMem);
RETURN_FUNC_PTR_IF_EXIST(zexCommandListAppendMIMath);
// Metrics
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerCreateExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerDestroyExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerEnableExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerDisableExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerReadDataExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricDecoderCreateExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricDecoderDestroyExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricDecoderGetDecodableMetricsExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerDecodeExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricCalculationOperationCreateExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricCalculationOperationDestroyExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricCalculationOperationGetExcludedMetricsExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricCalculationOperationGetReportFormatExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricDecodeCalculateMultipleValuesExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricCalculateValuesExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricTracerDecodeExp2);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricScopesGetExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricScopeGetPropertiesExp);
RETURN_FUNC_PTR_IF_EXIST(zetIntelMetricSupportedScopesGetExp);
// Graphs
RETURN_FUNC_PTR_IF_EXIST(zeGraphCreateExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListBeginGraphCaptureExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListBeginCaptureIntoGraphExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListEndGraphCaptureExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListInstantiateGraphExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListAppendGraphExp);
RETURN_FUNC_PTR_IF_EXIST(zeGraphDestroyExp);
RETURN_FUNC_PTR_IF_EXIST(zeExecutableGraphDestroyExp);
RETURN_FUNC_PTR_IF_EXIST(zeCommandListIsGraphCaptureEnabledExp);
RETURN_FUNC_PTR_IF_EXIST(zeGraphIsEmptyExp);
RETURN_FUNC_PTR_IF_EXIST(zeGraphDumpContentsExp);
#undef RETURN_FUNC_PTR_IF_EXIST
#undef RETURN_L0_FUNC_PTR_IF_EXIST
return ExtensionFunctionAddressHelper::getAdditionalExtensionFunctionAddress(functionName);
}
} // namespace L0
|
