Skip to content

Commit b756361

Browse files
committed
Fix unittest failure
Signed-off-by: Larsen, Steffen <[email protected]>
1 parent 7bb3edb commit b756361

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

sycl/unittests/Extensions/ProfilingTag.cpp

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)