Add success test guard around LoaderDriverCheck() (oneapi-src#134)

Signed-off-by: davidoli <[email protected]>
Jemale · Jun 17, 2024 · 766975c · 766975c
1 parent 4339454
commit 766975c
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/source/lib/ze_lib.cpp b/source/lib/ze_lib.cpp
@@ -44,7 +44,7 @@ namespace ze_lib
 
         if( NULL == loader )
             return ZE_RESULT_ERROR_UNINITIALIZED;
-          
+
         typedef ze_result_t (ZE_APICALL *loaderInit_t)();
         auto loaderInit = reinterpret_cast<loaderInit_t>(
                 GET_FUNCTION_PTR(loader, "zeLoaderInit") );
@@ -60,7 +60,7 @@ namespace ze_lib
         if( ZE_RESULT_SUCCESS == result ) {
             tracing_lib = zeLoaderGetTracingHandle();
         }
-        
+
 #endif
 
         if( ZE_RESULT_SUCCESS == result )
@@ -76,14 +76,17 @@ namespace ze_lib
         if( ZE_RESULT_SUCCESS == result )
         {
             result = zesInit();
-        } 
+        }
 
         if( ZE_RESULT_SUCCESS == result )
         {
             result = zelTracingInit();
         }
 
-        result = zelLoaderDriverCheck(flags);
+        if( ZE_RESULT_SUCCESS == result )
+        {
+            result = zelLoaderDriverCheck(flags);
+        }
 
         return result;
     }
@@ -95,7 +98,7 @@ extern "C" {
 
 ze_result_t ZE_APICALL
 zelLoaderGetVersions(
-   size_t *num_elems,                     //Pointer to num versions to get.  
+   size_t *num_elems,                     //Pointer to num versions to get.
    zel_component_version_t *versions)     //Pointer to array of versions. If set to NULL, num_elems is returned
 {
 #ifdef DYNAMIC_LOAD_LOADER
@@ -114,8 +117,8 @@ zelLoaderGetVersions(
 ze_result_t ZE_APICALL
 zelLoaderTranslateHandle(
    zel_handle_type_t handleType,
-   void *handleIn,                     
-   void **handleOut)    
+   void *handleIn,
+   void **handleOut)
 
 {
     return zelLoaderTranslateHandleInternal(handleType, handleIn, handleOut);