vtsymbal hotkernels sample (#45)

* adding host synchronize event handling to hot functions sample * adding host synchronize event handling to hot functions sample
intel · Dec 15, 2023 · 9ee0e46 · 9ee0e46
1 parent c60f92b
commit 9ee0e46
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.49.20
+0.49.21
diff --git a/samples/ze_hot_kernels/ze_kernel_collector.h b/samples/ze_hot_kernels/ze_kernel_collector.h
@@ -174,6 +174,8 @@ class ZeKernelCollector {
 
     prologue_callbacks.Event.pfnHostResetCb = OnEnterEventHostReset;
 
+    epilogue_callbacks.Event.pfnHostSynchronizeCb = OnExitEventHostSynchronize;
+
     prologue_callbacks.EventPool.pfnCreateCb = OnEnterEventPoolCreate;
     epilogue_callbacks.EventPool.pfnCreateCb = OnExitEventPoolCreate;
 
@@ -475,6 +477,19 @@ class ZeKernelCollector {
     }
   }
 
+  static void OnExitEventHostSynchronize(ze_event_host_synchronize_params_t *params,
+                                    ze_result_t result,
+                                    void *global_data,
+                                    void **instance_data) {
+    if (*(params->phEvent) != nullptr) {
+      ZeKernelCollector* collector =
+        reinterpret_cast<ZeKernelCollector*>(global_data);
+      PTI_ASSERT(collector != nullptr);
+    collector->ProcessCall(*(params->phEvent));
+    collector->ProcessCalls();
+    }
+  }
+
   static void CreateEvent(ze_context_handle_t context,
                           ze_event_pool_handle_t& event_pool,
                           ze_event_handle_t& event) {