Skip to content

Commit cd3e6ec

Browse files
committed
[L0] Fix usm prefect and advise to always set signal event.
Signed-off-by: Neil R. Spruit <[email protected]>
1 parent bad94a5 commit cd3e6ec

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

source/adapters/level_zero/memory.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1240,8 +1240,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueUSMPrefetch(
12401240
ur_event_handle_t *Event = OutEvent ? OutEvent : &InternalEvent;
12411241
UR_CALL(createEventAndAssociateQueue(Queue, Event, UR_COMMAND_USM_PREFETCH,
12421242
CommandList, IsInternal, false));
1243-
UR_CALL(setSignalEvent(Queue, UseCopyEngine, &ZeEvent, Event,
1244-
NumEventsInWaitList, EventWaitList));
1243+
ZeEvent = (*Event)->ZeEvent;
12451244
(*Event)->WaitList = TmpWaitList;
12461245

12471246
const auto &WaitList = (*Event)->WaitList;
@@ -1297,7 +1296,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueUSMAdvise(
12971296
ur_event_handle_t *Event = OutEvent ? OutEvent : &InternalEvent;
12981297
UR_CALL(createEventAndAssociateQueue(Queue, Event, UR_COMMAND_USM_ADVISE,
12991298
CommandList, IsInternal, false));
1300-
UR_CALL(setSignalEvent(Queue, UseCopyEngine, &ZeEvent, Event, 0, nullptr));
1299+
ZeEvent = (*Event)->ZeEvent;
13011300
(*Event)->WaitList = TmpWaitList;
13021301

13031302
const auto &ZeCommandList = CommandList->first;

0 commit comments

Comments
 (0)