@@ -67,7 +67,8 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedDefaultQueue) {
6767 mock::getCallbacks ().set_after_callback (" urEventGetProfilingInfo" ,
6868 &after_urEventGetProfilingInfo);
6969 mock::getCallbacks ().set_after_callback (
70- " urEnqueueEventsWaitWithBarrier" , &after_urEnqueueEventsWaitWithBarrier);
70+ " urEnqueueEventsWaitWithBarrier" ,
71+ &after_urEnqueueEventsWaitWithBarrierExt);
7172
7273 sycl::context Ctx{sycl::platform ()};
7374 sycl::queue Queue{Ctx, sycl::default_selector_v};
@@ -79,7 +80,7 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedDefaultQueue) {
7980 ASSERT_EQ (size_t {1 }, counter_urEnqueueTimestampRecordingExp);
8081 // TODO: We expect two barriers for now, while marker events leak. Adjust when
8182 // addressed.
82- ASSERT_EQ (size_t {2 }, counter_urEnqueueEventsWaitWithBarrier );
83+ ASSERT_EQ (size_t {2 }, counter_urEnqueueEventsWaitWithBarrierExt );
8384
8485 E.get_profiling_info <sycl::info::event_profiling::command_start>();
8586 ASSERT_TRUE (LatestProfilingQuery.has_value ());
@@ -98,7 +99,8 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedInOrderQueue) {
9899 mock::getCallbacks ().set_after_callback (" urEventGetProfilingInfo" ,
99100 &after_urEventGetProfilingInfo);
100101 mock::getCallbacks ().set_after_callback (
101- " urEnqueueEventsWaitWithBarrier" , &after_urEnqueueEventsWaitWithBarrier);
102+ " urEnqueueEventsWaitWithBarrier" ,
103+ &after_urEnqueueEventsWaitWithBarrierExt);
102104
103105 sycl::context Ctx{sycl::platform ()};
104106 sycl::queue Queue{
@@ -109,7 +111,7 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedInOrderQueue) {
109111
110112 sycl::event E = sycl::ext::oneapi::experimental::submit_profiling_tag (Queue);
111113 ASSERT_EQ (size_t {1 }, counter_urEnqueueTimestampRecordingExp);
112- ASSERT_EQ (size_t {0 }, counter_urEnqueueEventsWaitWithBarrier );
114+ ASSERT_EQ (size_t {0 }, counter_urEnqueueEventsWaitWithBarrierExt );
113115
114116 E.get_profiling_info <sycl::info::event_profiling::command_start>();
115117 ASSERT_TRUE (LatestProfilingQuery.has_value ());
@@ -156,7 +158,8 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedProfilingInOrderQueue) {
156158 mock::getCallbacks ().set_after_callback (" urEventGetProfilingInfo" ,
157159 &after_urEventGetProfilingInfo);
158160 mock::getCallbacks ().set_after_callback (
159- " urEnqueueEventsWaitWithBarrier" , &after_urEnqueueEventsWaitWithBarrier);
161+ " urEnqueueEventsWaitWithBarrier" ,
162+ &after_urEnqueueEventsWaitWithBarrierExt);
160163
161164 sycl::context Ctx{sycl::platform ()};
162165 sycl::queue Queue{Ctx,
@@ -169,7 +172,7 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedProfilingInOrderQueue) {
169172
170173 sycl::event E = sycl::ext::oneapi::experimental::submit_profiling_tag (Queue);
171174 ASSERT_EQ (size_t {1 }, counter_urEnqueueTimestampRecordingExp);
172- ASSERT_EQ (size_t {0 }, counter_urEnqueueEventsWaitWithBarrier );
175+ ASSERT_EQ (size_t {0 }, counter_urEnqueueEventsWaitWithBarrierExt );
173176
174177 E.get_profiling_info <sycl::info::event_profiling::command_start>();
175178 ASSERT_TRUE (LatestProfilingQuery.has_value ());
0 commit comments