[UR] Pull in change to make multi device compile extension core.

Author: aarongreig

Diff for: sycl/plugins/unified_runtime/CMakeLists.txt

@@ -56,14 +56,8 @@ endif()
 if(SYCL_PI_UR_USE_FETCH_CONTENT)
   include(FetchContent)
 
-  set(UNIFIED_RUNTIME_REPO "https://github.com/oneapi-src/unified-runtime")
-  # commit 3225b822b5d8cbfa85d7fc1bd5a5bf96e5bb8c1a
-  # Merge: edb281f3 5fc41099
-  # Author: Kenneth Benzie (Benie) <[email protected]>
-  # Date:   Tue Jan 30 12:31:44 2024 +0000
-  #     Merge pull request #1168 from Seanst98/sean/unique-addr-mode-per-dim-adapters
-  #     [Bindless][CUDA] Unique addressing modes per dimension
-  set(UNIFIED_RUNTIME_TAG 3225b822b5d8cbfa85d7fc1bd5a5bf96e5bb8c1a)
+  set(UNIFIED_RUNTIME_REPO "https://github.com/aarongreig/unified-runtime")
+  set(UNIFIED_RUNTIME_TAG aaron/makeDeviceCompileExtCore)
 
   if(SYCL_PI_UR_OVERRIDE_FETCH_CONTENT_REPO)
     set(UNIFIED_RUNTIME_REPO "${SYCL_PI_UR_OVERRIDE_FETCH_CONTENT_REPO}")

Diff for: sycl/plugins/unified_runtime/pi2ur.hpp

@@ -1999,14 +1999,10 @@ piProgramLink(pi_context Context, pi_uint32 NumDevices,
   auto UrDevices = reinterpret_cast<ur_device_handle_t *>(
       const_cast<pi_device *>(DeviceList));
 
-  auto urResult =
-      urProgramLinkExp(UrContext, NumDevices, UrDevices, NumInputPrograms,
-                       UrInputPrograms, Options, UrProgram);
-  if (urResult == UR_RESULT_ERROR_UNSUPPORTED_FEATURE) {
-    urResult = urProgramLink(UrContext, NumInputPrograms, UrInputPrograms,
-                             Options, UrProgram);
-  }
-  return ur2piResult(urResult);
+  HANDLE_ERRORS(urProgramLink(UrContext, NumDevices, UrDevices,
+                              NumInputPrograms, UrInputPrograms, Options,
+                              UrProgram));
+  return PI_SUCCESS;
 }
 
 inline pi_result piProgramCompile(
@@ -2038,12 +2034,8 @@ inline pi_result piProgramCompile(
   auto UrDevices = reinterpret_cast<ur_device_handle_t *>(
       const_cast<pi_device *>(DeviceList));
 
-  auto urResult =
-      urProgramCompileExp(UrProgram, NumDevices, UrDevices, Options);
-  if (urResult == UR_RESULT_ERROR_UNSUPPORTED_FEATURE) {
-    urResult = urProgramCompile(UrContext, UrProgram, Options);
-  }
-  return ur2piResult(urResult);
+  HANDLE_ERRORS(urProgramCompile(UrProgram, NumDevices, UrDevices, Options));
+  return PI_SUCCESS;
 }
 
 inline pi_result
@@ -2068,12 +2060,8 @@ piProgramBuild(pi_program Program, pi_uint32 NumDevices,
 
   auto UrDevices = reinterpret_cast<ur_device_handle_t *>(
       const_cast<pi_device *>(DeviceList));
-
-  auto urResult = urProgramBuildExp(UrProgram, NumDevices, UrDevices, Options);
-  if (urResult == UR_RESULT_ERROR_UNSUPPORTED_FEATURE) {
-    urResult = urProgramBuild(UrContext, UrProgram, Options);
-  }
-  return ur2piResult(urResult);
+  HANDLE_ERRORS(urProgramBuild(UrProgram, NumDevices, UrDevices, Options));
+  return PI_SUCCESS;
 }
 
 inline pi_result piextProgramSetSpecializationConstant(pi_program Program,