Set loader callback and add adapter implementations.

Author: martygrant

scripts/templates/ldrddi.cpp.mako

@@ -51,6 +51,12 @@ namespace ur_loader
         add_local = False
     %>
 
+        %if re.match(r"\w+SetLoggerCallback$", th.make_func_name(n, tags, obj)):
+ 
+        ur_loader::urSetLoggerCallback(${obj['params'][0]['name']}, ${obj['params'][1]['name']}, ${obj['params'][2]['name']});
+
+        %endif
+
         %if re.match(r"\w+AdapterGet$", th.make_func_name(n, tags, obj)):
 
         size_t adapterIndex = 0;

source/adapters/cuda/adapter.cpp

@@ -114,3 +114,14 @@ UR_APIEXPORT ur_result_t UR_APICALL urAdapterGetInfo(ur_adapter_handle_t,
 
   return UR_RESULT_SUCCESS;
 }
+
+UR_APIEXPORT ur_result_t UR_APICALL urSetLoggerCallback(
+    ur_adapter_handle_t, ur_logger_output_callback_t pfnLoggerCallback,
+    void *pUserData) {
+
+  if (adapter) {
+    adapter->log.setSinkCallback(pfnLoggerCallback, pUserData);
+  }
+
+  return UR_RESULT_SUCCESS;
+}

source/adapters/hip/adapter.cpp

@@ -102,3 +102,14 @@ UR_APIEXPORT ur_result_t UR_APICALL urAdapterGetInfo(ur_adapter_handle_t,
 
   return UR_RESULT_SUCCESS;
 }
+
+UR_APIEXPORT ur_result_t UR_APICALL urSetLoggerCallback(
+    ur_adapter_handle_t, ur_logger_output_callback_t pfnLoggerCallback,
+    void *pUserData) {
+
+  if (adapter) {
+    adapter->log.setSinkCallback(pfnLoggerCallback, pUserData);
+  }
+
+  return UR_RESULT_SUCCESS;
+}

source/adapters/level_zero/adapter.cpp

@@ -336,3 +336,14 @@ UR_APIEXPORT ur_result_t UR_APICALL urAdapterGetInfo(ur_adapter_handle_t,
 
   return UR_RESULT_SUCCESS;
 }
+
+UR_APIEXPORT ur_result_t UR_APICALL urSetLoggerCallback(
+    ur_adapter_handle_t, ur_logger_output_callback_t pfnLoggerCallback,
+    void *pUserData) {
+
+  if (GlobalAdapter) {
+    GlobalAdapter->log.setSinkCallback(pfnLoggerCallback, pUserData);
+  }
+
+  return UR_RESULT_SUCCESS;
+}

source/adapters/native_cpu/adapter.cpp

@@ -63,3 +63,12 @@ UR_APIEXPORT ur_result_t UR_APICALL urAdapterGetInfo(ur_adapter_handle_t,
 
   return UR_RESULT_SUCCESS;
 }
+
+UR_APIEXPORT ur_result_t UR_APICALL urSetLoggerCallback(
+    ur_adapter_handle_t, ur_logger_output_callback_t pfnLoggerCallback,
+    void *pUserData) {
+
+  Adapter.logger.setSinkCallback(pfnLoggerCallback, pUserData);
+
+  return UR_RESULT_SUCCESS;
+}

source/adapters/opencl/adapter.cpp

@@ -101,3 +101,14 @@ UR_APIEXPORT ur_result_t UR_APICALL urAdapterGetInfo(ur_adapter_handle_t,
 
   return UR_RESULT_SUCCESS;
 }
+
+UR_APIEXPORT ur_result_t UR_APICALL urSetLoggerCallback(
+    ur_adapter_handle_t, ur_logger_output_callback_t pfnLoggerCallback,
+    void *pUserData) {
+
+  if (adapter) {
+    adapter->log.setSinkCallback(pfnLoggerCallback, pUserData);
+  }
+
+  return UR_RESULT_SUCCESS;
+}

source/loader/ur_ldrddi.cpp

@@ -201,6 +201,8 @@ __urdlllocal ur_result_t UR_APICALL urSetLoggerCallback(
 ) {
     ur_result_t result = UR_RESULT_SUCCESS;
 
+    ur_loader::urSetLoggerCallback(hAdapter, pfnLoggerCallback, pUserData);
+
     // extract platform's function pointer table
     auto dditable = reinterpret_cast<ur_adapter_object_t *>(hAdapter)->dditable;
     auto pfnSetLoggerCallback = dditable->ur.Global.pfnSetLoggerCallback;