test

Signed-off-by: Jaime Arteaga <[email protected]>

Author: Jaime Arteaga

source/adapters/level_zero/usm.cpp

@@ -430,6 +430,9 @@ UR_APIEXPORT ur_result_t UR_APICALL urUSMDeviceAlloc(
     return umf2urResult(umfRet);
   }
 
+  printf("%s %d *RetMem 0x%lx\n", __FILE__, __LINE__,
+         (unsigned long int)*RetMem);
+
   if (IndirectAccessTrackingEnabled) {
     // Keep track of all memory allocations in the context
     Context->MemAllocs.emplace(std::piecewise_construct,
@@ -555,6 +558,8 @@ UR_APIEXPORT ur_result_t UR_APICALL urUSMFree(
   std::scoped_lock<ur_shared_mutex> Lock(
       IndirectAccessTrackingEnabled ? Plt->ContextsMutex : Context->Mutex);
 
+  printf("%s %d\n", __FILE__, __LINE__);
+
   return USMFreeHelper(Context, Mem);
 }
 
@@ -985,6 +990,7 @@ const bool UseUSMAllocator = ShouldUseUSMAllocator();
 // mutex.
 ur_result_t USMFreeHelper(ur_context_handle_t Context, void *Ptr,
                           bool OwnZeMemHandle) {
+  printf("%s %d\n", __FILE__, __LINE__);
   if (!OwnZeMemHandle) {
     // Memory should not be freed
     return UR_RESULT_SUCCESS;
@@ -1005,16 +1011,20 @@ ur_result_t USMFreeHelper(ur_context_handle_t Context, void *Ptr,
     Context->MemAllocs.erase(It);
   }
 
+  printf("%s %d Ptr 0x%lx\n", __FILE__, __LINE__, (unsigned long int)Ptr);
+
   auto hPool = umfPoolByPtr(Ptr);
   if (!hPool) {
     if (IndirectAccessTrackingEnabled)
       UR_CALL(ContextReleaseHelper(Context));
+    printf("%s %d\n", __FILE__, __LINE__);
     return UR_RESULT_ERROR_INVALID_MEM_OBJECT;
   }
 
   auto umfRet = umfPoolFree(hPool, Ptr);
   if (IndirectAccessTrackingEnabled)
     UR_CALL(ContextReleaseHelper(Context));
+  printf("%s %d\n", __FILE__, __LINE__);
   return umf2urResult(umfRet);
 }
 

test/conformance/usm/urUSMDeviceAlloc.cpp

@@ -75,8 +75,9 @@ TEST_P(urUSMDeviceAllocTest, SuccessWithDescriptors) {
     ASSERT_SUCCESS(urEnqueueUSMFill(queue, ptr, sizeof(pattern), &pattern,
                                     allocation_size, 0, nullptr, &event));
     ASSERT_SUCCESS(urEventWait(1, &event));
-
+    printf("%s %d\n", __FILE__, __LINE__);
     ASSERT_SUCCESS(urUSMFree(context, ptr));
+    printf("%s %d\n", __FILE__, __LINE__);
     EXPECT_SUCCESS(urEventRelease(event));
 }