[L0] Add support for multi-device kernel compilation (#12032)

pre-commit Pr for
oneapi-src/unified-runtime#1135

---------

Signed-off-by: Spruit, Neil R <[email protected]>
Co-authored-by: Kenneth Benzie (Benie) <[email protected]>

Author: nrspruit

sycl/plugins/unified_runtime/CMakeLists.txt

@@ -57,11 +57,13 @@ if(SYCL_PI_UR_USE_FETCH_CONTENT)
   include(FetchContent)
 
   set(UNIFIED_RUNTIME_REPO "https://github.com/oneapi-src/unified-runtime.git")
-  # commit 2d27a62757ba2f2f4a47ed0a0e86254081634419
+  # commit 7c58060ffdd86e1e342e9bea12834e49aa2cb187
+  # Merge: 4f800802 1b2cd5b3
   # Author: Kenneth Benzie (Benie) <[email protected]>
-  # Date:   Mon Jan 22 12:31:47 2024 +0000
-  #     Set version to v0.8.3
-  set(UNIFIED_RUNTIME_TAG v0.8.3)
+  # Date:   Tue Jan 23 11:29:19 2024 +0000
+  #     Merge pull request #1135 from nrspruit/multi_device_kernel_compilation_main
+  #     [L0] Add support for multi-device kernel compilation
+  set(UNIFIED_RUNTIME_TAG 7c58060ffdd86e1e342e9bea12834e49aa2cb187)
 
   if(SYCL_PI_UR_OVERRIDE_FETCH_CONTENT_REPO)
     set(UNIFIED_RUNTIME_REPO "${SYCL_PI_UR_OVERRIDE_FETCH_CONTENT_REPO}")

sycl/plugins/unified_runtime/pi2ur.hpp

@@ -1879,11 +1879,6 @@ inline pi_result piProgramCreateWithBinary(
   PI_ASSERT(Binaries && Lengths, PI_ERROR_INVALID_VALUE);
   PI_ASSERT(Program, PI_ERROR_INVALID_PROGRAM);
 
-  // For now we support only one device.
-  if (NumDevices != 1) {
-    die("piProgramCreateWithBinary: level_zero supports only one device.");
-    return PI_ERROR_INVALID_VALUE;
-  }
   if (!Binaries[0] || !Lengths[0]) {
     if (BinaryStatus)
       *BinaryStatus = PI_ERROR_INVALID_VALUE;
@@ -1985,11 +1980,6 @@ piProgramLink(pi_context Context, pi_uint32 NumDevices,
               pi_uint32 NumInputPrograms, const pi_program *InputPrograms,
               void (*PFnNotify)(pi_program Program, void *UserData),
               void *UserData, pi_program *RetProgram) {
-  // We only support one device with Level Zero currently.
-  if (NumDevices != 1) {
-    die("piProgramLink: level_zero supports only one device.");
-    return PI_ERROR_INVALID_VALUE;
-  }
 
   // Validate input parameters.
   PI_ASSERT(DeviceList, PI_ERROR_INVALID_DEVICE);
@@ -2064,14 +2054,6 @@ piProgramBuild(pi_program Program, pi_uint32 NumDevices,
     return PI_ERROR_INVALID_VALUE;
   }
 
-  // We only support build to one device with Level Zero now.
-  // TODO: we should eventually build to the possibly multiple root
-  // devices in the context.
-  if (NumDevices != 1) {
-    die("piProgramBuild: level_zero supports only one device.");
-    return PI_ERROR_INVALID_VALUE;
-  }
-
   // These aren't supported.
   PI_ASSERT(!PFnNotify && !UserData, PI_ERROR_INVALID_VALUE);