[UR][L0] Event cleanup in urEnqueueKernelLaunch

When using internal events we do not clean up after execution. In the
case of repeated call to urEnqueueKernelLaunch we eventually return
UR_RESULT_ERROR_OUT_OF_RESOURCES.

Signed-off-by: Zhang, Winston <[email protected]>

Author: winstonzhang-intel

unified-runtime/source/adapters/level_zero/kernel.cpp

@@ -221,6 +221,20 @@ ur_result_t urEnqueueKernelLaunch(
   UR_CALL(Queue->executeCommandList(CommandList, false /*IsBlocking*/,
                                     true /*OKToBatchCommand*/));
 
+  // For internal events, trigger cleanup to prevent event pool exhaustion
+  // by ensuring completed internal events are cleaned up periodically
+  if (IsInternal) {
+    if (Queue->UsingImmCmdLists) {
+      UR_CALL(CleanupEventsInImmCmdLists(Queue, false /*QueueLocked*/,
+                                         false /*QueueSynced*/,
+                                         nullptr /*CompletedEvent*/));
+    } else {
+      // For regular command lists, call resetCommandLists to clean up
+      // completed command lists and their associated events
+      UR_CALL(resetCommandLists(Queue));
+    }
+  }
+
   return UR_RESULT_SUCCESS;
 }
 

unified-runtime/source/adapters/level_zero/kernel.hpp

@@ -14,6 +14,7 @@
 #include "common.hpp"
 #include "common/ur_ref_count.hpp"
 #include "memory.hpp"
+#include "queue.hpp"
 
 struct ur_kernel_handle_t_ : ur_object {
   ur_kernel_handle_t_(bool OwnZeHandle, ur_program_handle_t Program)