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
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
/*
* Copyright (C) 2023-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "level_zero/sysman/source/api/events/windows/sysman_os_events_imp.h"
#include "level_zero/sysman/source/shared/windows/sysman_kmd_sys_manager.h"
#include "level_zero/sysman/source/shared/windows/zes_os_sysman_imp.h"
namespace L0 {
namespace Sysman {
void WddmEventsImp::registerEvents(zes_event_type_flags_t eventId, uint32_t requestId) {
ze_result_t status = ZE_RESULT_SUCCESS;
EventHandler event;
KmdSysman::RequestProperty request;
KmdSysman::ResponseProperty response;
request.requestId = requestId;
request.commandId = KmdSysman::Command::RegisterEvent;
request.componentId = KmdSysman::Component::InterfaceProperties;
request.dataSize = sizeof(HANDLE);
event.requestId = requestId;
event.id = eventId;
// create an event with a default security descriptor with no name
bool manualResetObject = FALSE;
bool isEventStateSignalled = FALSE;
event.windowsHandle = createWddmEvent(NULL, manualResetObject, isEventStateSignalled, NULL);
memcpy_s(request.dataBuffer, sizeof(HANDLE), &event.windowsHandle, sizeof(HANDLE));
status = pKmdSysManager->requestSingle(request, response);
if (status != ZE_RESULT_SUCCESS) {
closeWddmHandle(event.windowsHandle);
return;
}
eventList.push_back(event);
}
void WddmEventsImp::unregisterEvents() {
ze_result_t status = ZE_RESULT_SUCCESS;
EventHandler event;
KmdSysman::RequestProperty request;
KmdSysman::ResponseProperty response;
setWddmEvent(exitHandle);
request.commandId = KmdSysman::Command::UnregisterEvent;
request.componentId = KmdSysman::Component::InterfaceProperties;
request.dataSize = sizeof(HANDLE);
for (uint32_t i = 0; i < eventList.size(); i++) {
request.requestId = eventList[i].requestId;
event.windowsHandle = eventList[i].windowsHandle;
memcpy_s(request.dataBuffer, sizeof(HANDLE), &event.windowsHandle, sizeof(HANDLE));
status = pKmdSysManager->requestSingle(request, response);
if (status == ZE_RESULT_SUCCESS) {
closeWddmHandle(event.windowsHandle);
}
}
eventList.clear();
}
ze_result_t WddmEventsImp::eventRegister(zes_event_type_flags_t events) {
unregisterEvents();
if (events & ZES_EVENT_TYPE_FLAG_ENERGY_THRESHOLD_CROSSED) {
registerEvents(ZES_EVENT_TYPE_FLAG_ENERGY_THRESHOLD_CROSSED, KmdSysman::Events::EnergyThresholdCrossed);
}
if (events & ZES_EVENT_TYPE_FLAG_DEVICE_SLEEP_STATE_ENTER) {
registerEvents(ZES_EVENT_TYPE_FLAG_DEVICE_SLEEP_STATE_ENTER, KmdSysman::Events::EnterD3);
}
if (events & ZES_EVENT_TYPE_FLAG_DEVICE_SLEEP_STATE_EXIT) {
registerEvents(ZES_EVENT_TYPE_FLAG_DEVICE_SLEEP_STATE_EXIT, KmdSysman::Events::EnterD0);
}
if (events & ZES_EVENT_TYPE_FLAG_DEVICE_DETACH) {
registerEvents(ZES_EVENT_TYPE_FLAG_DEVICE_DETACH, KmdSysman::Events::EnterTDR);
}
if (events & ZES_EVENT_TYPE_FLAG_DEVICE_ATTACH) {
registerEvents(ZES_EVENT_TYPE_FLAG_DEVICE_ATTACH, KmdSysman::Events::ExitTDR);
}
resetWddmEvent(exitHandle);
return (eventList.size() == 0) ? ZE_RESULT_ERROR_UNSUPPORTED_FEATURE : ZE_RESULT_SUCCESS;
}
bool WddmEventsImp::eventListen(zes_event_type_flags_t &pEvent, uint64_t timeout) {
HANDLE events[MAXIMUM_WAIT_OBJECTS];
pEvent = 0;
// Note: whatever happens on this function, it should return true. If that's not the case, the upper loop in sysman.cpp will
// cause an infinite loop for the case of "Infinite timeout". This may work on Linux since the implementation is poll based,
// windows uses WaitForMultipleObjects, which is a blocking call.
// no events no listen. Less than MAXIMUM_WAIT_OBJECTS - 2 to left space for the exit handle.
if (eventList.size() == 0 || (eventList.size() >= (MAXIMUM_WAIT_OBJECTS - 2))) {
pEvent = ZES_EVENT_TYPE_FLAG_FORCE_UINT32;
return true;
}
// set every handle from pos 0 onwards...
for (uint32_t i = 0; i < eventList.size(); i++) {
events[i] = eventList[i].windowsHandle;
}
events[eventList.size()] = exitHandle;
// Setting the last handle for the exit handle, then the exit handle is signaled, it breaks from the wait.
uint32_t signaledEvent = waitForMultipleWddmEvents(static_cast<uint32_t>(eventList.size() + 1), events, FALSE, static_cast<uint32_t>(timeout));
resetWddmEvent(exitHandle);
// Was a timeout, exit event loop.
if (signaledEvent == WAIT_TIMEOUT) {
return true;
}
// Was the exit event and exit event loop.
if (signaledEvent == eventList.size()) {
pEvent = ZES_EVENT_TYPE_FLAG_FORCE_UINT32;
} else {
pEvent = eventList[signaledEvent].id;
}
// Whatever reason exit the loop, WaitForMultipleObjects exited, exit from the loop must follow.
return true;
}
HANDLE WddmEventsImp::createWddmEvent(
LPSECURITY_ATTRIBUTES lpEventAttributes,
BOOL bManualReset,
BOOL bInitialState,
LPCWSTR lpName) {
return CreateEvent(lpEventAttributes, bManualReset, bInitialState, lpName);
}
void WddmEventsImp::closeWddmHandle(HANDLE exitHandle) {
CloseHandle(exitHandle);
}
void WddmEventsImp::resetWddmEvent(HANDLE resetHandle) {
ResetEvent(resetHandle);
}
void WddmEventsImp::setWddmEvent(HANDLE setHandle) {
SetEvent(setHandle);
}
uint32_t WddmEventsImp::waitForMultipleWddmEvents(_In_ DWORD nCount,
_In_reads_(nCount) CONST HANDLE *lpHandles,
_In_ BOOL bWaitAll,
_In_ DWORD dwMilliseconds) {
return WaitForMultipleObjects(nCount, lpHandles, bWaitAll, dwMilliseconds);
}
WddmEventsImp::WddmEventsImp(OsSysman *pOsSysman) {
WddmSysmanImp *pWddmSysmanImp = static_cast<WddmSysmanImp *>(pOsSysman);
pKmdSysManager = &pWddmSysmanImp->getKmdSysManager();
// create an exit event with a default security descriptor with no name
bool manualResetObject = FALSE;
bool isEventStateSignalled = FALSE;
exitHandle = WddmEventsImp::createWddmEvent(NULL, manualResetObject, isEventStateSignalled, NULL);
}
WddmEventsImp::~WddmEventsImp() {
WddmEventsImp::closeWddmHandle(exitHandle);
}
OsEvents *OsEvents::create(OsSysman *pOsSysman) {
WddmEventsImp *pWddmEventsImp = new WddmEventsImp(pOsSysman);
return static_cast<OsEvents *>(pWddmEventsImp);
}
} // namespace Sysman
} // namespace L0
|
