Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[SYCL] Fix the barrier dependency for OOO profiling tags #16112

Merged
merged 5 commits into from
Nov 19, 2024
Merged

[SYCL] Fix the barrier dependency for OOO profiling tags #16112

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f91373b
[SYCL] Fix the barrier dependency for OOO profiling tags
steffenlarsen Nov 18, 2024
2b5e556
Temporarily use barrier instead of marker
steffenlarsen Nov 19, 2024
d9a4039
Defer barrier optimization path due to timestamp bugs
steffenlarsen Nov 19, 2024
a2f23df
Revert OOO profiling expectation in unittests
steffenlarsen Nov 19, 2024
4bcd383
Adjust unittest expectations
steffenlarsen Nov 19, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 30 additions & 8 deletions sycl/source/detail/scheduler/commands.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3441,25 +3441,47 @@ ur_result_t ExecCGCommand::enqueueImpQueue() {
case CGType::ProfilingTag: {
assert(MQueue && "Profiling tag requires a valid queue");
const auto &Adapter = MQueue->getAdapter();

bool IsInOrderQueue = MQueue->isInOrder();
ur_event_handle_t *TimestampDeps = nullptr;
size_t NumTimestampDeps = 0;

// If the queue is not in-order, the implementation will need to first
// insert a marker event that the timestamp waits for.
ur_event_handle_t PreTimestampMarkerEvent{};
if (!IsInOrderQueue) {
// FIXME: urEnqueueEventsWait on the L0 adapter requires a double-release.
// Use that instead once it has been fixed.
// See https://github.com/oneapi-src/unified-runtime/issues/2347.
Adapter->call<UrApiKind::urEnqueueEventsWaitWithBarrier>(
MQueue->getHandleRef(),
/*num_events_in_wait_list=*/0,
/*event_wait_list=*/nullptr, &PreTimestampMarkerEvent);
TimestampDeps = &PreTimestampMarkerEvent;
NumTimestampDeps = 1;
}

Adapter->call<UrApiKind::urEnqueueTimestampRecordingExp>(
MQueue->getHandleRef(),
/*blocking=*/false, NumTimestampDeps, TimestampDeps, Event);

// If the queue is not in-order, we need to insert a barrier. This barrier
// does not need output events as it will implicitly enforce the following
// enqueue is blocked until it finishes.
if (!MQueue->isInOrder()) {
if (!IsInOrderQueue) {
// We also need to release the timestamp event from the marker.
Adapter->call<UrApiKind::urEventRelease>(PreTimestampMarkerEvent);
// FIXME: Due to a bug in the L0 UR adapter, we will leak events if we do
// not pass an output event to the UR call. Once that is fixed,
// this immediately-deleted event can be removed.
ur_event_handle_t PreTimestampBarrierEvent{};
ur_event_handle_t PostTimestampBarrierEvent{};
Adapter->call<UrApiKind::urEnqueueEventsWaitWithBarrier>(
MQueue->getHandleRef(),
/*num_events_in_wait_list=*/0,
/*event_wait_list=*/nullptr, &PreTimestampBarrierEvent);
Adapter->call<UrApiKind::urEventRelease>(PreTimestampBarrierEvent);
/*event_wait_list=*/nullptr, &PostTimestampBarrierEvent);
Adapter->call<UrApiKind::urEventRelease>(PostTimestampBarrierEvent);
}

Adapter->call<UrApiKind::urEnqueueTimestampRecordingExp>(
MQueue->getHandleRef(),
/*blocking=*/false,
/*num_events_in_wait_list=*/0, /*event_wait_list=*/nullptr, Event);
if (Event)
MEvent->setHandle(*Event);
return UR_RESULT_SUCCESS;
Expand Down
67 changes: 67 additions & 0 deletions sycl/unittests/Extensions/ProfilingTag.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,8 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedDefaultQueue) {
"urEnqueueTimestampRecordingExp", &after_urEnqueueTimestampRecordingExp);
mock::getCallbacks().set_after_callback("urEventGetProfilingInfo",
&after_urEventGetProfilingInfo);
mock::getCallbacks().set_after_callback(
"urEnqueueEventsWaitWithBarrier", &after_urEnqueueEventsWaitWithBarrier);

sycl::context Ctx{sycl::platform()};
sycl::queue Queue{Ctx, sycl::default_selector_v};
Expand All @@ -75,6 +77,39 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedDefaultQueue) {

sycl::event E = sycl::ext::oneapi::experimental::submit_profiling_tag(Queue);
ASSERT_EQ(size_t{1}, counter_urEnqueueTimestampRecordingExp);
// TODO: We expect two barriers for now, while marker events leak. Adjust when
// addressed.
ASSERT_EQ(size_t{2}, counter_urEnqueueEventsWaitWithBarrier);

E.get_profiling_info<sycl::info::event_profiling::command_start>();
ASSERT_TRUE(LatestProfilingQuery.has_value());
ASSERT_EQ(*LatestProfilingQuery, UR_PROFILING_INFO_COMMAND_START);

E.get_profiling_info<sycl::info::event_profiling::command_end>();
ASSERT_TRUE(LatestProfilingQuery.has_value());
ASSERT_EQ(*LatestProfilingQuery, UR_PROFILING_INFO_COMMAND_END);
}

TEST_F(ProfilingTagTest, ProfilingTagSupportedInOrderQueue) {
mock::getCallbacks().set_after_callback("urDeviceGetInfo",
&after_urDeviceGetInfo<true>);
mock::getCallbacks().set_after_callback(
"urEnqueueTimestampRecordingExp", &after_urEnqueueTimestampRecordingExp);
mock::getCallbacks().set_after_callback("urEventGetProfilingInfo",
&after_urEventGetProfilingInfo);
mock::getCallbacks().set_after_callback(
"urEnqueueEventsWaitWithBarrier", &after_urEnqueueEventsWaitWithBarrier);

sycl::context Ctx{sycl::platform()};
sycl::queue Queue{
Ctx, sycl::default_selector_v, {sycl::property::queue::in_order()}};
sycl::device Dev = Queue.get_device();

ASSERT_TRUE(Dev.has(sycl::aspect::ext_oneapi_queue_profiling_tag));

sycl::event E = sycl::ext::oneapi::experimental::submit_profiling_tag(Queue);
ASSERT_EQ(size_t{1}, counter_urEnqueueTimestampRecordingExp);
ASSERT_EQ(size_t{0}, counter_urEnqueueEventsWaitWithBarrier);

E.get_profiling_info<sycl::info::event_profiling::command_start>();
ASSERT_TRUE(LatestProfilingQuery.has_value());
Expand Down Expand Up @@ -113,6 +148,38 @@ TEST_F(ProfilingTagTest, ProfilingTagSupportedProfilingQueue) {
ASSERT_EQ(*LatestProfilingQuery, UR_PROFILING_INFO_COMMAND_END);
}

TEST_F(ProfilingTagTest, ProfilingTagSupportedProfilingInOrderQueue) {
mock::getCallbacks().set_after_callback("urDeviceGetInfo",
&after_urDeviceGetInfo<true>);
mock::getCallbacks().set_after_callback(
"urEnqueueTimestampRecordingExp", &after_urEnqueueTimestampRecordingExp);
mock::getCallbacks().set_after_callback("urEventGetProfilingInfo",
&after_urEventGetProfilingInfo);
mock::getCallbacks().set_after_callback(
"urEnqueueEventsWaitWithBarrier", &after_urEnqueueEventsWaitWithBarrier);

sycl::context Ctx{sycl::platform()};
sycl::queue Queue{Ctx,
sycl::default_selector_v,
{sycl::property::queue::enable_profiling(),
sycl::property::queue::in_order()}};
sycl::device Dev = Queue.get_device();

ASSERT_TRUE(Dev.has(sycl::aspect::ext_oneapi_queue_profiling_tag));

sycl::event E = sycl::ext::oneapi::experimental::submit_profiling_tag(Queue);
ASSERT_EQ(size_t{1}, counter_urEnqueueTimestampRecordingExp);
ASSERT_EQ(size_t{0}, counter_urEnqueueEventsWaitWithBarrier);

E.get_profiling_info<sycl::info::event_profiling::command_start>();
ASSERT_TRUE(LatestProfilingQuery.has_value());
ASSERT_EQ(*LatestProfilingQuery, UR_PROFILING_INFO_COMMAND_START);

E.get_profiling_info<sycl::info::event_profiling::command_end>();
ASSERT_TRUE(LatestProfilingQuery.has_value());
ASSERT_EQ(*LatestProfilingQuery, UR_PROFILING_INFO_COMMAND_END);
}

TEST_F(ProfilingTagTest, ProfilingTagFallbackDefaultQueue) {
mock::getCallbacks().set_after_callback("urDeviceGetInfo",
&after_urDeviceGetInfo<false>);
Expand Down
Loading