Manually check error code from zeDeviceGetRootDevice calls.

Author: maarquitos14

source/adapters/level_zero/device.cpp

@@ -88,7 +88,16 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGet(
     if (!isCombinedMode) {
       ze_device_handle_t RootDev = nullptr;
       // Query Root Device
-      ZE2UR_CALL(zeDeviceGetRootDevice, (D->ZeDevice, &RootDev));
+      // We cannot use ZE2UR_CALL because under some circumstances this call may
+      // return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE, and ZE2UR_CALL will abort
+      // because it's not UR_RESULT_SUCCESS. Instead, we use ZE_CALL_NOCHECK and
+      // we check manually that the result is either ZE_RESULT_SUCCESS or
+      // ZE_RESULT_ERROR_UNSUPPORTED_FEATURE.
+      auto errc =
+          ZE_CALL_NOCHECK(zeDeviceGetRootDevice, (D->ZeDevice, &RootDev));
+      if (errc != ZE_RESULT_SUCCESS &&
+          errc != ZE_RESULT_ERROR_UNSUPPORTED_FEATURE)
+        return ze2urResult(errc);
       // For COMPOSITE and FLAT modes, RootDev will always be nullptr. Thus a
       // single device returning RootDev != nullptr means we are in COMBINED
       // mode.

source/adapters/level_zero/platform.cpp

@@ -423,8 +423,18 @@ ur_result_t ur_platform_handle_t_::populateDeviceCacheIfNeeded() {
       // through zeDeviceGetRootDevice. We need to cache the card device
       // handle too, such that it is readily visible to the
       // urDeviceCreateWithNativeHandle.
-      ze_device_handle_t RootDevice;
-      ZE2UR_CALL(zeDeviceGetRootDevice, (Device->ZeDevice, &RootDevice));
+      ze_device_handle_t RootDevice = nullptr;
+      // We cannot use ZE2UR_CALL because under some circumstances this call may
+      // return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE, and ZE2UR_CALL will abort
+      // because it's not UR_RESULT_SUCCESS. Instead, we use ZE_CALL_NOCHECK and
+      // we check manually that the result is either ZE_RESULT_SUCCESS or
+      // ZE_RESULT_ERROR_UNSUPPORTED_FEATURE.
+      auto errc = ZE_CALL_NOCHECK(zeDeviceGetRootDevice,
+                                  (Device->ZeDevice, &RootDevice));
+      if (errc != ZE_RESULT_SUCCESS &&
+          errc != ZE_RESULT_ERROR_UNSUPPORTED_FEATURE)
+        return ze2urResult(errc);
+
       if (RootDevice) {
         if (std::find_if(URDevicesCache.begin(), URDevicesCache.end(),
                          [&](auto &Dev) {