[SYCL] Return Success in USM Mem Free if L0 driver is already released (#11075)

nrspruit · web-flow · commit 54cd00d559a8 · 2023-09-12T11:23:04.000-07:00
Signed-off-by: Spruit, Neil R &lt;neil.r.spruit@intel.com&gt;
diff --git a/sycl/plugins/unified_runtime/ur/adapters/level_zero/usm.cpp b/sycl/plugins/unified_runtime/ur/adapters/level_zero/usm.cpp
@@ -607,8 +607,13 @@ UR_APIEXPORT ur_result_t UR_APICALL urUSMGetMemAllocInfo(
 }
 
 static ur_result_t USMFreeImpl(ur_context_handle_t Context, void *Ptr) {
-  ZE2UR_CALL(zeMemFree, (Context->ZeContext, Ptr));
-  return UR_RESULT_SUCCESS;
+  auto ZeResult = ZE_CALL_NOCHECK(zeMemFree, (Context->ZeContext, Ptr));
+  // Handle When the driver is already released
+  if (ZeResult == ZE_RESULT_ERROR_UNINITIALIZED) {
+    return UR_RESULT_SUCCESS;
+  } else {
+    return ze2urResult(ZeResult);
+  }
 }
 
 static ur_result_t USMQueryPageSize(ur_context_handle_t Context, void *Ptr,

Original file line number	Diff line number	Diff line change
`@@ -607,8 +607,13 @@ UR_APIEXPORT ur_result_t UR_APICALL urUSMGetMemAllocInfo(`
`607`	`607`	`}`
`608`	`608`
`609`	`609`	`static ur_result_t USMFreeImpl(ur_context_handle_t Context, void *Ptr) {`
`610`		`- ZE2UR_CALL(zeMemFree, (Context->ZeContext, Ptr));`
`611`		`- return UR_RESULT_SUCCESS;`
	`610`	`+ auto ZeResult = ZE_CALL_NOCHECK(zeMemFree, (Context->ZeContext, Ptr));`
	`611`	`+ // Handle When the driver is already released`
	`612`	`+ if (ZeResult == ZE_RESULT_ERROR_UNINITIALIZED) {`
	`613`	`+ return UR_RESULT_SUCCESS;`
	`614`	`+ } else {`
	`615`	`+ return ze2urResult(ZeResult);`
	`616`	`+ }`
`612`	`617`	`}`
`613`	`618`
`614`	`619`	`static ur_result_t USMQueryPageSize(ur_context_handle_t Context, void *Ptr,`