Sanitizer sync

Author: mateuszpn

unified-runtime/source/loader/layers/sanitizer/sanitizer_common/sanitizer_utils.hpp

@@ -75,8 +75,28 @@ ur_result_t EnqueueUSMSet(ur_queue_handle_t Queue, void *Ptr, T Value,
                           const ur_event_handle_t *EventWaitList = nullptr,
                           ur_event_handle_t *OutEvent = nullptr) {
   assert(Size % sizeof(T) == 0);
-  return getContext()->urDdiTable.Enqueue.pfnUSMFill(
-      Queue, Ptr, sizeof(T), &Value, Size, NumEvents, EventWaitList, OutEvent);
+  ur_event_handle_t LocalEvent = nullptr;
+  ur_event_handle_t *EventStorage = OutEvent ? OutEvent : &LocalEvent;
+
+  auto Result = getContext()->urDdiTable.Enqueue.pfnUSMFill(
+      Queue, Ptr, sizeof(T), &Value, Size, NumEvents, EventWaitList,
+      EventStorage);
+  if (Result != UR_RESULT_SUCCESS) {
+    return Result;
+  }
+
+  if (!OutEvent && LocalEvent) {
+    auto WaitResult = getContext()->urDdiTable.Event.pfnWait(1, &LocalEvent);
+    auto ReleaseResult = getContext()->urDdiTable.Event.pfnRelease(LocalEvent);
+    if (WaitResult != UR_RESULT_SUCCESS) {
+      return WaitResult;
+    }
+    if (ReleaseResult != UR_RESULT_SUCCESS) {
+      return ReleaseResult;
+    }
+  }
+
+  return Result;
 }
 
 void PrintUrBuildLogIfError(ur_result_t Result, ur_program_handle_t Program,