tmp: use zeInit

Author: igchor

src/provider/provider_level_zero.c

@@ -162,6 +162,7 @@ typedef struct ze_memory_provider_t {
 } ze_memory_provider_t;
 
 typedef struct ze_ops_t {
+    ze_result_t (*zeInit)(ze_init_flags_t);
     ze_result_t (*zeInitDrivers)(uint32_t *, ze_driver_handle_t *,
                                  ze_init_driver_type_desc_t *);
     ze_result_t (*zeMemAllocHost)(ze_context_handle_t,
@@ -218,25 +219,25 @@ static umf_result_t ze2umf_result(ze_result_t result) {
 }
 
 static umf_result_t ze_init_drivers() {
-    ze_init_driver_type_desc_t desc = {
-        .stype = ZE_STRUCTURE_TYPE_INIT_DRIVER_TYPE_DESC,
-        .pNext = NULL,
-        .flags = UINT32_MAX};
-    uint32_t driverCount = 0;
-    ze_result_t result = g_ze_ops.zeInitDrivers(&driverCount, NULL, &desc);
-    if (result != ZE_RESULT_SUCCESS) {
-        return ze2umf_result(result);
-    }
-
-    assert(zeAllDrivers == NULL);
-    zeAllDrivers =
-        umf_ba_global_alloc(sizeof(ze_driver_handle_t) * driverCount);
-    result = g_ze_ops.zeInitDrivers(&driverCount, zeAllDrivers, &desc);
-    if (result != ZE_RESULT_SUCCESS) {
-        return ze2umf_result(result);
-    }
-
-    return UMF_RESULT_SUCCESS;
+    // ze_init_driver_type_desc_t desc = {
+    //     .stype = ZE_STRUCTURE_TYPE_INIT_DRIVER_TYPE_DESC,
+    //     .pNext = NULL,
+    //     .flags = UINT32_MAX};
+    // uint32_t driverCount = 0;
+    // ze_result_t result = g_ze_ops.zeInitDrivers(&driverCount, NULL, &desc);
+    // if (result != ZE_RESULT_SUCCESS) {
+    //     return ze2umf_result(result);
+    // }
+
+    // assert(zeAllDrivers == NULL);
+    // zeAllDrivers =
+    //     umf_ba_global_alloc(sizeof(ze_driver_handle_t) * driverCount);
+    // result = g_ze_ops.zeInitDrivers(&driverCount, zeAllDrivers, &desc);
+    // if (result != ZE_RESULT_SUCCESS) {
+    //     return ze2umf_result(result);
+    // }
+
+    return ze2umf_result(g_ze_ops.zeInit(ZE_INIT_FLAG_GPU_ONLY));
 }
 
 static void init_ze_global_state(void) {
@@ -256,6 +257,8 @@ static void init_ze_global_state(void) {
         return;
     }
 
+    *(void **)&g_ze_ops.zeInit =
+        utils_get_symbol_addr(lib_handle, "zeInit", lib_name);
     *(void **)&g_ze_ops.zeInitDrivers =
         utils_get_symbol_addr(lib_handle, "zeInitDrivers", lib_name);
     *(void **)&g_ze_ops.zeMemAllocHost =