Restrict events reset during command buffer enqueue

Author: Xewar313

unified-runtime/source/adapters/level_zero/v2/command_buffer.cpp

@@ -99,6 +99,7 @@ ur_event_handle_t *ur_exp_command_buffer_handle_t_::getWaitListFromSyncPoints(
       UR_LOG(ERR, "Invalid sync point");
       throw UR_RESULT_ERROR_INVALID_VALUE;
     }
+    usedSyncPoints.insert(pSyncPointWaitList[i]);
     syncPointWaitList[i] = syncPoints[pSyncPointWaitList[i]];
   }
   return syncPointWaitList.data();
@@ -132,9 +133,10 @@ ur_result_t ur_exp_command_buffer_handle_t_::finalizeCommandBuffer() {
   if (!isInOrder) {
     ZE2UR_CALL(zeCommandListAppendBarrier,
                (commandListLocked->getZeCommandList(), nullptr, 0, nullptr));
-    for (auto &event : syncPoints) {
+    for (auto &event : usedSyncPoints) {
       ZE2UR_CALL(zeCommandListAppendEventReset,
-                 (commandListLocked->getZeCommandList(), event->getZeEvent()));
+                 (commandListLocked->getZeCommandList(),
+                  syncPoints[event]->getZeEvent()));
     }
     ZE2UR_CALL(zeCommandListAppendBarrier,
                (commandListLocked->getZeCommandList(), nullptr, 0, nullptr));

unified-runtime/source/adapters/level_zero/v2/command_buffer.hpp

@@ -66,6 +66,9 @@ struct ur_exp_command_buffer_handle_t_ : public ur_object {
   // Stores all sync points that are created by the command buffer.
   std::vector<ur_event_handle_t> syncPoints;
 
+  // Stores all sync points that should be reset after execution.
+  std::unordered_set<ur_exp_command_buffer_sync_point_t> usedSyncPoints;
+
   // Temporary storage for sync points that are passed to function that require
   // array of events. This is used to avoid allocating a new memory every time.
   std::vector<ur_event_handle_t> syncPointWaitList;