some logs upon free and add closing library

Author: lslusarczyk

src/provider/provider_level_zero.c

@@ -425,19 +425,22 @@ static umf_result_t ze_memory_provider_free_helper(void *provider, void *ptr,
                                                    size_t bytes,
                                                    int update_stats) {
     if (ptr == NULL) {
+        LOG_DEBUG("Free skipped as ptr is NULL");
         return UMF_RESULT_SUCCESS;
     }
 
-    ze_memory_provider_t *ze_provider = (ze_memory_provider_t *)provider;
+    ze_memory_provider_t *ze_provider = provider;
     umf_result_t ret;
     if (ze_provider->freePolicyFlags == 0) {
+        LOG_DEBUG("No free policy set, caling zeMemFree with %p", ptr);
         ret = ze2umf_result(g_ze_ops.zeMemFree(ze_provider->context, ptr));
     } else {
         ze_memory_free_ext_desc_t desc = {
             .stype = ZE_STRUCTURE_TYPE_MEMORY_FREE_EXT_DESC,
             .pNext = NULL,
             .freePolicy = ze_provider->freePolicyFlags};
 
+        LOG_DEBUG("Free policy was set, caling zeMemFreeExt with %p", ptr);
         ret = ze2umf_result(
             g_ze_ops.zeMemFreeExt(ze_provider->context, &desc, ptr));
     }
@@ -559,8 +562,6 @@ static umf_result_t query_min_page_size(ze_memory_provider_t *ze_provider,
                                         size_t *min_page_size) {
     assert(min_page_size);
 
-    LOG_DEBUG("Querying minimum page size");
-
     void *ptr;
     umf_result_t result =
         ze_memory_provider_alloc_helper(ze_provider, 1, 0, 0, &ptr);
@@ -574,6 +575,7 @@ static umf_result_t query_min_page_size(ze_memory_provider_t *ze_provider,
         ze_provider->context, ptr, &properties, NULL);
 
     *min_page_size = properties.pageSize;
+    LOG_DEBUG("Querying minimum page size, got:%lu", properties.pageSize);
 
     ze_memory_provider_free_helper(ze_provider, ptr, 1, 0);
 

test/common/level_zero_mocks.cpp

@@ -101,7 +101,7 @@ void MockedLevelZeroTestEnvironment::SetUp() {
     setenv("UMF_ZE_LOADER_LIB_NAME", lib_name, 1);
 #endif
 
-    void *lib_handle = utils_open_library(lib_name, 0);
+    lib_handle = utils_open_library(lib_name, 0);
     ASSERT_NE(lib_handle, nullptr);
 
     void *l0interface_sym =
@@ -113,6 +113,8 @@ void MockedLevelZeroTestEnvironment::SetUp() {
 
     ASSERT_EQ(*l0interface, nullptr);
 }
-void MockedLevelZeroTestEnvironment::TearDown() {}
+void MockedLevelZeroTestEnvironment::TearDown() {
+    utils_close_library(lib_handle);
+}
 
 LevelZero **MockedLevelZeroTestEnvironment::l0interface;

test/common/level_zero_mocks.h

@@ -79,6 +79,9 @@ class LevelZeroMock : public LevelZero {
 
 // important, makes UMF load ze_loopback instead of regular l0
 class MockedLevelZeroTestEnvironment : public ::testing::Environment {
+
+    void *lib_handle;
+
   public:
     static LevelZero **l0interface;