Skip to content

Commit d3f81d2

Browse files
unify scope trace among batched and in-order queues
1 parent 5f22881 commit d3f81d2

File tree

2 files changed

+18
-6
lines changed

2 files changed

+18
-6
lines changed

unified-runtime/source/adapters/level_zero/v2/queue_batched.cpp

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,15 +172,19 @@ ur_result_t ur_queue_batched_t::queueFinishBatchAndPoolsUnlocked(
172172
enqueueCurrentBatchUnlocked(immediateList, regularList);
173173

174174
{
175+
// TRACK_SCOPE_LATENCY(
176+
// "ur_queue_batched_t::queueFinishBatchAndPoolsUnlocked_hostSynchronize");
175177
TRACK_SCOPE_LATENCY(
176-
"ur_queue_batched_t::queueFinishBatchAndPoolsUnlocked_hostSynchronize");
178+
"ur_queue_batched_t::hostSynchronize");
177179
// finish queue
178180
ZE2UR_CALL(zeCommandListHostSynchronize, (immediateList, UINT64_MAX));
179181
}
180182

181183
{
184+
// TRACK_SCOPE_LATENCY(
185+
// "ur_queue_batched_t::queueFinishBatchAndPoolsUnlocked_asyncPools");
182186
TRACK_SCOPE_LATENCY(
183-
"ur_queue_batched_t::queueFinishBatchAndPoolsUnlocked_asyncPools");
187+
"ur_queue_batched_t::asyncPools");
184188
hContext->getAsyncPool()->cleanupPoolsForQueue(this);
185189
hContext->forEachUsmPool([this](ur_usm_pool_handle_t hPool) {
186190
hPool->cleanupPoolsForQueue(this);
@@ -201,8 +205,10 @@ ur_queue_batched_t::queueFinishUnlocked(locked<batch_manager> &batchLocked) {
201205
batchLocked->activeBatch.getZeCommandList()));
202206

203207
{
208+
// TRACK_SCOPE_LATENCY(
209+
// "ur_queue_batched_t::queueFinishUnlocked_releaseSubmittedKernels");
204210
TRACK_SCOPE_LATENCY(
205-
"ur_queue_batched_t::queueFinishUnlocked_releaseSubmittedKernels");
211+
"ur_queue_batched_t::releaseSubmittedKernels");
206212
UR_CALL(batchLocked->immediateList.releaseSubmittedKernels());
207213
}
208214

unified-runtime/source/adapters/level_zero/v2/queue_immediate_in_order.cpp

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,15 +106,19 @@ ur_result_t ur_queue_immediate_in_order_t::queueFinish() {
106106
auto lockedCommandListManager = commandListManager.lock();
107107

108108
{
109+
// TRACK_SCOPE_LATENCY(
110+
// "ur_queue_immediate_in_order_t::queueFinish_hostSynchronize");
109111
TRACK_SCOPE_LATENCY(
110-
"ur_queue_immediate_in_order_t::queueFinish_hostSynchronize");
112+
"ur_queue_immediate_in_order_t::hostSynchronize");
111113
ZE2UR_CALL(zeCommandListHostSynchronize,
112114
(lockedCommandListManager->getZeCommandList(), UINT64_MAX));
113115
}
114116

115117
{
118+
// TRACK_SCOPE_LATENCY(
119+
// "ur_queue_immediate_in_order_t::queueFinish_asyncPools");
116120
TRACK_SCOPE_LATENCY(
117-
"ur_queue_immediate_in_order_t::queueFinish_asyncPools");
121+
"ur_queue_immediate_in_order_t::asyncPools");
118122
hContext->getAsyncPool()->cleanupPoolsForQueue(this);
119123
hContext->forEachUsmPool([this](ur_usm_pool_handle_t hPool) {
120124
hPool->cleanupPoolsForQueue(this);
@@ -123,8 +127,10 @@ ur_result_t ur_queue_immediate_in_order_t::queueFinish() {
123127
}
124128

125129
{
130+
// TRACK_SCOPE_LATENCY(
131+
// "ur_queue_immediate_in_order_t::queueFinish_releaseSubmittedKernels");
126132
TRACK_SCOPE_LATENCY(
127-
"ur_queue_immediate_in_order_t::queueFinish_releaseSubmittedKernels");
133+
"ur_queue_immediate_in_order_t::releaseSubmittedKernels");
128134
UR_CALL(lockedCommandListManager->releaseSubmittedKernels());
129135
}
130136

0 commit comments

Comments
 (0)