Skip to content

Commit 093fcb0

Browse files
authored
[UR][Offload] Implement logger callbacks (#18755)
1 parent 896f1b9 commit 093fcb0

File tree

2 files changed

+19
-2
lines changed

2 files changed

+19
-2
lines changed

unified-runtime/source/adapters/offload/adapter.cpp

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,3 +117,20 @@ UR_APIEXPORT ur_result_t UR_APICALL urAdapterGetLastError(ur_adapter_handle_t,
117117
// "ADAPTER_SPECIFIC", which we never do
118118
return UR_RESULT_SUCCESS;
119119
}
120+
121+
UR_APIEXPORT ur_result_t UR_APICALL urAdapterSetLoggerCallback(
122+
ur_adapter_handle_t, ur_logger_callback_t pfnLoggerCallback,
123+
void *pUserData, ur_logger_level_t level = UR_LOGGER_LEVEL_QUIET) {
124+
125+
Adapter.Logger.setCallbackSink(pfnLoggerCallback, pUserData, level);
126+
127+
return UR_RESULT_SUCCESS;
128+
}
129+
130+
UR_APIEXPORT ur_result_t UR_APICALL
131+
urAdapterSetLoggerCallbackLevel(ur_adapter_handle_t, ur_logger_level_t level) {
132+
133+
Adapter.Logger.setCallbackLevel(level);
134+
135+
return UR_RESULT_SUCCESS;
136+
}

unified-runtime/source/adapters/offload/ur_interface_loader.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,8 +207,8 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetAdapterProcAddrTable(
207207
pDdiTable->pfnRelease = urAdapterRelease;
208208
pDdiTable->pfnRetain = urAdapterRetain;
209209
pDdiTable->pfnGetLastError = urAdapterGetLastError;
210-
pDdiTable->pfnSetLoggerCallback = nullptr;
211-
pDdiTable->pfnSetLoggerCallbackLevel = nullptr;
210+
pDdiTable->pfnSetLoggerCallback = urAdapterSetLoggerCallback;
211+
pDdiTable->pfnSetLoggerCallbackLevel = urAdapterSetLoggerCallbackLevel;
212212
return UR_RESULT_SUCCESS;
213213
}
214214

0 commit comments

Comments
 (0)