[SYCL][UR] Link UR PI against UR Loader (#8865)

Compile UR L0 adapter as a standalone library

Signed-off-by: Brandon Yates <[email protected]>

Author: bmyates

sycl/plugins/unified_runtime/CMakeLists.txt

@@ -37,28 +37,23 @@ if (NOT DEFINED UNIFIED_RUNTIME_LIBRARY OR NOT DEFINED UNIFIED_RUNTIME_INCLUDE_D
   # Restore original flags
   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS_BAK}")
 
-  add_library(${PROJECT_NAME}::ur_loader ALIAS loader)
-
-  set(UNIFIED_RUNTIME_LIBRARY ${PROJECT_NAME}::ur_loader)
+  add_library(UnifiedRuntimeLoader ALIAS loader)
 
   set(UNIFIED_RUNTIME_SOURCE_DIR
     ${unified-runtime_SOURCE_DIR} CACHE PATH "Path to Unified Runtime Headers")
   set(UNIFIED_RUNTIME_INCLUDE_DIR "${UNIFIED_RUNTIME_SOURCE_DIR}/include")
 endif()
 
+
 add_library (UnifiedRuntime-Headers INTERFACE)
+
 target_include_directories(UnifiedRuntime-Headers
   INTERFACE
     "${UNIFIED_RUNTIME_INCLUDE_DIR}"
 )
 
 find_package(Threads REQUIRED)
 
-#
-# NOTE: the Unified Runtime doesn't have the loader [yet].
-# So what we really build is the Unified Runtime with Level Zero Adapter
-# together.
-#
 add_sycl_plugin(unified_runtime
   SOURCES
     # These are short-term shared with Unified Runtime
@@ -69,9 +64,25 @@ add_sycl_plugin(unified_runtime
     "ur_bindings.hpp"
     "pi2ur.hpp"
     "pi2ur.cpp"
+    # These below belong to Unified Runtime PI Plugin only
+    "pi_unified_runtime.hpp"
+    "pi_unified_runtime.cpp"
+    LIBRARIES
+    Threads::Threads
+    UnifiedRuntimeLoader
+    UnifiedRuntime-Headers
+)
+
+# Build level zero adapter
+add_sycl_library("ur_adapter_level_zero" SHARED
+  SOURCES
     "ur/ur.hpp"
     "ur/ur.cpp"
+    "ur/usm_allocator.hpp"
+    "ur/usm_allocator.cpp"
     "ur/adapters/level_zero/ur_level_zero.hpp"
+    "ur/adapters/level_zero/ur_level_zero.cpp"
+    "ur/adapters/level_zero/ur_loader_interface.cpp"
     "ur/adapters/level_zero/ur_level_zero_common.hpp"
     "ur/adapters/level_zero/ur_level_zero_context.hpp"
     "ur/adapters/level_zero/ur_level_zero_device.hpp"
@@ -93,18 +104,21 @@ add_sycl_plugin(unified_runtime
     "ur/adapters/level_zero/ur_level_zero_program.cpp"
     "ur/adapters/level_zero/ur_level_zero_queue.cpp"
     "ur/adapters/level_zero/ur_level_zero_sampler.cpp"
-    "ur/usm_allocator.hpp"
-    "ur/usm_allocator.cpp"
-    # These below belong to Unified Runtime PI Plugin only
-    "pi_unified_runtime.hpp"
-    "pi_unified_runtime.cpp"
-    LIBRARIES
-    Threads::Threads
+  INCLUDE_DIRS
+    ${sycl_inc_dir}
+  LIBRARIES
     UnifiedRuntime-Headers
     LevelZeroLoader-Headers
     LevelZeroLoader
+    Threads::Threads
 )
 
-if (TARGET ${PROJECT_NAME}::ur_loader)
+set_target_properties("ur_adapter_level_zero" PROPERTIES
+    VERSION "0.0.0"
+    SOVERSION "0"
+)
+
+
+if (TARGET UnifiedRuntimeLoader)
   set_target_properties(hello_world PROPERTIES EXCLUDE_FROM_ALL 1 EXCLUDE_FROM_DEFAULT_BUILD 1)
 endif()

sycl/plugins/unified_runtime/pi2ur.hpp

@@ -279,6 +279,7 @@ namespace pi2ur {
 inline pi_result piPlatformsGet(pi_uint32 num_entries, pi_platform *platforms,
                                 pi_uint32 *num_platforms) {
 
+  urInit(0);
   uint32_t Count = num_entries;
   auto phPlatforms = reinterpret_cast<ur_platform_handle_t *>(platforms);
   HANDLE_ERRORS(urPlatformGet(Count, phPlatforms, num_platforms));

sycl/plugins/unified_runtime/ur/adapters/level_zero/ur_level_zero.cpp

@@ -84,7 +84,7 @@ ur_result_t _ur_platform_handle_t::initialize() {
   return UR_RESULT_SUCCESS;
 }
 
-UR_APIEXPORT ur_result_t UR_APICALL urPlatformGet(
+ur_result_t urPlatformGet(
     uint32_t NumEntries, ///< [in] the number of platforms to be added to
                          ///< phPlatforms. If phPlatforms is not NULL, then
                          ///< NumEntries should be greater than zero, otherwise
@@ -201,7 +201,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urPlatformGet(
   return UR_RESULT_SUCCESS;
 }
 
-UR_APIEXPORT ur_result_t UR_APICALL urPlatformGetInfo(
+ur_result_t urPlatformGetInfo(
     ur_platform_handle_t Platform, ///< [in] handle of the platform
     ur_platform_info_t ParamName,  ///< [in] type of the info to retrieve
     size_t Size,      ///< [in] the number of bytes pointed to by pPlatformInfo.
@@ -255,7 +255,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urPlatformGetInfo(
   return UR_RESULT_SUCCESS;
 }
 
-UR_APIEXPORT ur_result_t UR_APICALL urDeviceGet(
+ur_result_t urDeviceGet(
     ur_platform_handle_t Platform, ///< [in] handle of the platform instance
     ur_device_type_t DeviceType,   ///< [in] the type of the devices.
     uint32_t NumEntries, ///< [in] the number of devices to be added to
@@ -332,7 +332,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGet(
   return UR_RESULT_SUCCESS;
 }
 
-UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(
+ur_result_t urDeviceGetInfo(
     ur_device_handle_t Device,  ///< [in] handle of the device instance
     ur_device_info_t ParamName, ///< [in] type of the info to retrieve
     size_t propSize,  ///< [in] the number of bytes pointed to by pDeviceInfo.
@@ -1482,7 +1482,7 @@ void ZeUSMImportExtension::doZeUSMRelease(ze_driver_handle_t DriverHandle,
   ZE_CALL_NOCHECK(zexDriverReleaseImportedPointer, (DriverHandle, HostPtr));
 }
 
-UR_APIEXPORT ur_result_t UR_APICALL urDevicePartition(
+ur_result_t urDevicePartition(
     ur_device_handle_t Device, ///< [in] handle of the device to partition.
     const ur_device_partition_property_t
         *Properties, ///< [in] null-terminated array of <$_device_partition_t
@@ -1563,3 +1563,11 @@ UR_APIEXPORT ur_result_t UR_APICALL urDevicePartition(
   }
   return UR_RESULT_SUCCESS;
 }
+
+ur_result_t urInit([[maybe_unused]] ur_device_init_flags_t device_flags) {
+  return UR_RESULT_SUCCESS;
+}
+
+ur_result_t urTearDown([[maybe_unused]] void *pParams) {
+  return UR_RESULT_SUCCESS;
+}