From e5658573f3cf2e9a30d4c692bbb01118f46a2a93 Mon Sep 17 00:00:00 2001
From: "Kassen, Andrew" <andrew.kassen@intel.com>
Date: Tue, 2 Jul 2024 10:44:38 -0700
Subject: [PATCH] xpu: ocl: enable profiling for copy

---
 src/xpu/ocl/stream_impl.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/xpu/ocl/stream_impl.cpp b/src/xpu/ocl/stream_impl.cpp
index 90660dda334..f56e6fdd95c 100644
--- a/src/xpu/ocl/stream_impl.cpp
+++ b/src/xpu/ocl/stream_impl.cpp
@@ -165,6 +165,12 @@ status_t stream_impl_t::copy(impl::stream_t *stream,
         return status::success;
     }
 
+    if (is_profiling_enabled()) {
+        auto ocl_event = utils::make_unique<gpu::intel::ocl::ocl_event_t>(
+                std::vector<xpu::ocl::wrapper_t<cl_event>> {out_event});
+        stream_profiler->register_event(std::move(ocl_event));
+    }
+
     if (flags() & stream_flags::out_of_order)
         gpu::intel::ocl::ocl_event_t::from(out_dep).events
                 = {std::move(out_event)};