check if driver supports cb events

Author: mateuszpn

unified-runtime/source/adapters/level_zero/device.cpp

@@ -1665,6 +1665,8 @@ ur_result_t urDeviceCreateWithNativeHandle(
   ur_device_handle_t Dev = nullptr;
   for (const auto &p : GlobalAdapter->Platforms) {
     Dev = p->getDeviceFromNativeHandle(ZeDevice);
+    if (Dev)
+      break;
   }
 
   if (Dev == nullptr)

unified-runtime/source/adapters/level_zero/v2/context.cpp

@@ -86,8 +86,14 @@ ur_context_handle_t_::ur_context_handle_t_(ze_context_handle_t hContext,
             auto device = platform->getDeviceById(deviceId);
 
             // TODO: just use per-context id?
-            return std::make_unique<v2::provider_counter>(
-                platform, context, v2::QUEUE_IMMEDIATE, device, flags);
+            // Use counter-based events only if the extension is available
+            if (platform->ZeDriverEventPoolCountingEventsExtensionFound) {
+              return std::make_unique<v2::provider_counter>(
+                  platform, context, v2::QUEUE_IMMEDIATE, device, flags);
+            } else {
+              return std::make_unique<v2::provider_normal>(
+                  context, v2::QUEUE_IMMEDIATE, flags);
+            }
           }),
       eventPoolCacheRegular(this, phDevices[0]->Platform->getNumDevices(),
                             [context = this, platform = phDevices[0]->Platform](