Exit early with levelzero in unit tests

Signed-off-by: Tuomas Katila <[email protected]>

Author: tkatila

.github/workflows/lib-validate.yaml

@@ -66,6 +66,8 @@ jobs:
       - run: make go-mod-tidy
       - run: make BUILDTAGS=kerneldrv
       - run: make test BUILDTAGS=kerneldrv
+        env:
+          UNITTEST: 1
       - run: make check-github-actions
       #- name: Codecov report
       #  run: bash <(curl -s https://codecov.io/bash)

cmd/gpu_levelzero/ze.c

@@ -14,6 +14,7 @@
 
 #include <stdio.h>
 #include <string.h>
+#include <stdlib.h>
 
 #include <ze_api.h>
 
@@ -92,6 +93,10 @@ static ze_driver_handle_t initialize_ze(void)
 
 bool ze_try_initialize(void)
 {
+    if (getenv("UNITTEST") != NULL) {
+        return false;
+    }
+
     return zeInit(0) == ZE_RESULT_SUCCESS;
 }
 
@@ -101,6 +106,10 @@ bool ze_try_initialize(void)
 /// @return Number of indices stored
 int ze_intel_device_indices(uint32_t* indices, uint32_t indices_size, uint32_t *error)
 {
+    if (getenv("UNITTEST") != NULL) {
+        return 0;
+    }
+
     if (indices == NULL || 0 == indices_size) {
         *error = ZE_RESULT_ERROR_INVALID_NULL_POINTER;
 

cmd/gpu_levelzero/zes.c

@@ -35,6 +35,10 @@ static bool device_enumerated = false;
 
 bool zes_try_initialize(void)
 {
+    if (getenv("UNITTEST") != NULL) {
+        return false;
+    }
+
     return zesInit(0) == ZE_RESULT_SUCCESS;
 }
 
@@ -161,6 +165,10 @@ bool is_integrated(zes_device_handle_t handle)
 /// @return memory amount for the device
 uint64_t zes_device_memory_amount(char* bdf_address, uint32_t* error)
 {
+    if (getenv("UNITTEST") != NULL) {
+        return 0;
+    }
+
     ze_result_t res = ZE_RESULT_SUCCESS;
 
     if (!device_enumerated) {
@@ -208,6 +216,10 @@ uint64_t zes_device_memory_amount(char* bdf_address, uint32_t* error)
 /// @return true for good, false for bad
 bool zes_device_memory_is_healthy(char* bdf_address, uint32_t* error)
 {
+    if (getenv("UNITTEST") != NULL) {
+        return false;
+    }
+
     if (!device_enumerated) {
         ze_result_t res = enumerate_zes_devices();
         if (res != ZE_RESULT_SUCCESS) {
@@ -254,6 +266,10 @@ bool zes_device_memory_is_healthy(char* bdf_address, uint32_t* error)
 /// @return true for good, false for bad
 bool zes_device_bus_is_healthy(char* bdf_address, uint32_t* error)
 {
+    if (getenv("UNITTEST") != NULL) {
+        return false;
+    }
+
     if (!device_enumerated) {
         ze_result_t res = enumerate_zes_devices();
         if (res != ZE_RESULT_SUCCESS) {