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
|
/*
* Copyright (C) 2021-2023 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "level_zero/tools/source/debug/debug_handlers.h"
#include "shared/source/os_interface/debug_env_reader.h"
#include "level_zero/core/source/device/device_imp.h"
#include "level_zero/tools/source/debug/debug_session.h"
namespace L0 {
namespace DebugApiHandlers {
std::mutex debugSessionMutex;
ze_result_t debugAttach(zet_device_handle_t hDevice, const zet_debug_config_t *config, zet_debug_session_handle_t *phDebug) {
ze_result_t result = ZE_RESULT_SUCCESS;
if (L0::Device::fromHandle(hDevice)->getNEODevice()->isSubDevice() && !NEO::debugManager.flags.ExperimentalEnableTileAttach.get()) {
return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
}
NEO::EnvironmentVariableReader envReader;
auto affinityMask = envReader.getSetting("ZE_AFFINITY_MASK", std::string(""));
if (!affinityMask.empty()) {
NEO::printDebugString(NEO::debugManager.flags.PrintDebugMessages.get(), stdout,
"%s", "ZE_AFFINITY_MASK is not recommended while using program debug API\n");
}
auto session = L0::Device::fromHandle(hDevice)->getDebugSession(*config);
std::unique_lock<std::mutex> lock(debugSessionMutex);
auto rootSession = L0::Device::fromHandle(hDevice)->getNEODevice()->getRootDevice()->getSpecializedDevice<DeviceImp>()->getDebugSession(*config);
if (session && session->isAttached()) {
if (session->getDebugConfig().pid != config->pid) {
return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
}
}
if (rootSession) {
if (rootSession->getDebugConfig().pid != config->pid) {
return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
}
}
// If root device with active TileSessions or
// subdevice with root device attached - fail
if ((!L0::Device::fromHandle(hDevice)->getNEODevice()->isSubDevice() && session && !session->areAllTileDebugSessionDetached()) ||
(L0::Device::fromHandle(hDevice)->getNEODevice()->isSubDevice() && rootSession && rootSession->isAttached())) {
result = ZE_RESULT_ERROR_NOT_AVAILABLE;
*phDebug = nullptr;
return result;
}
if (!session) {
bool isRootAttach = !L0::Device::fromHandle(hDevice)->getNEODevice()->isSubDevice();
session = L0::Device::fromHandle(hDevice)->createDebugSession(*config, result, isRootAttach);
}
if (session) {
session->setAttached();
*phDebug = session->toHandle();
}
return result;
}
ze_result_t debugDetach(zet_debug_session_handle_t hDebug) {
auto session = L0::DebugSession::fromHandle(hDebug);
if (session) {
auto device = session->getConnectedDevice();
device->removeDebugSession();
session->closeConnection();
if (!device->getNEODevice()->isSubDevice()) {
delete session;
} else {
std::unique_lock<std::mutex> lock(debugSessionMutex);
auto rootL0Device = device->getNEODevice()->getRootDevice()->getSpecializedDevice<DeviceImp>();
zet_debug_config_t dummy = {};
auto rootSession = rootL0Device->getDebugSession(dummy);
session->setDetached();
rootSession->detachTileDebugSession(session);
if (rootSession->areAllTileDebugSessionDetached()) {
rootL0Device->removeDebugSession();
delete rootSession;
}
}
}
return ZE_RESULT_SUCCESS;
}
ze_result_t debugReadEvent(zet_debug_session_handle_t hDebug, uint64_t timeout, zet_debug_event_t *event) { return L0::DebugSession::fromHandle(hDebug)->readEvent(timeout, event); }
ze_result_t debugInterrupt(zet_debug_session_handle_t hDebug, ze_device_thread_t thread) { return L0::DebugSession::fromHandle(hDebug)->interrupt(thread); }
ze_result_t debugResume(zet_debug_session_handle_t hDebug, ze_device_thread_t thread) { return L0::DebugSession::fromHandle(hDebug)->resume(thread); }
ze_result_t debugReadMemory(zet_debug_session_handle_t hDebug, ze_device_thread_t thread, const zet_debug_memory_space_desc_t *desc, size_t size, void *buffer) {
return L0::DebugSession::fromHandle(hDebug)->readMemory(thread, desc, size, buffer);
}
ze_result_t debugWriteMemory(zet_debug_session_handle_t hDebug, ze_device_thread_t thread, const zet_debug_memory_space_desc_t *desc, size_t size, const void *buffer) {
return L0::DebugSession::fromHandle(hDebug)->writeMemory(thread, desc, size, buffer);
}
ze_result_t debugAcknowledgeEvent(zet_debug_session_handle_t hDebug, const zet_debug_event_t *event) {
return L0::DebugSession::fromHandle(hDebug)->acknowledgeEvent(event);
}
ze_result_t debugGetRegisterSetProperties(zet_device_handle_t hDevice, uint32_t *pCount, zet_debug_regset_properties_t *pRegisterSetProperties) {
return L0::DebugSession::getRegisterSetProperties(L0::Device::fromHandle(hDevice), pCount, pRegisterSetProperties);
}
ze_result_t debugGetThreadRegisterSetProperties(zet_debug_session_handle_t hDebug, ze_device_thread_t thread, uint32_t *pCount, zet_debug_regset_properties_t *pRegisterSetProperties) {
return L0::DebugSession::fromHandle(hDebug)->getThreadRegisterSetProperties(thread, pCount, pRegisterSetProperties);
}
ze_result_t debugReadRegisters(zet_debug_session_handle_t hDebug, ze_device_thread_t thread, uint32_t type, uint32_t start, uint32_t count, void *pRegisterValues) {
return L0::DebugSession::fromHandle(hDebug)->readRegisters(thread, type, start, count, pRegisterValues);
}
ze_result_t debugWriteRegisters(zet_debug_session_handle_t hDebug, ze_device_thread_t thread, uint32_t type, uint32_t start, uint32_t count, void *pRegisterValues) {
return L0::DebugSession::fromHandle(hDebug)->writeRegisters(thread, type, start, count, pRegisterValues);
}
} // namespace DebugApiHandlers
} // namespace L0
|
