File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
unified-runtime/source/adapters/cuda Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -33,9 +33,9 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueUSMDeviceAllocExp(
33
33
phEventWaitList));
34
34
35
35
if (phEvent) {
36
- RetImplEvent =
37
- std::unique_ptr< ur_event_handle_t_>( ur_event_handle_t_ ::makeNative (
38
- UR_COMMAND_KERNEL_LAUNCH, hQueue, CuStream, StreamToken));
36
+ RetImplEvent = std::unique_ptr<ur_event_handle_t_>(
37
+ ur_event_handle_t_::makeNative (UR_COMMAND_ENQUEUE_USM_DEVICE_ALLOC_EXP,
38
+ hQueue, CuStream, StreamToken));
39
39
UR_CHECK_ERROR (RetImplEvent->start ());
40
40
}
41
41
@@ -93,7 +93,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueUSMFreeExp(
93
93
if (phEvent) {
94
94
RetImplEvent =
95
95
std::unique_ptr<ur_event_handle_t_>(ur_event_handle_t_::makeNative (
96
- UR_COMMAND_KERNEL_LAUNCH , hQueue, CuStream, StreamToken));
96
+ UR_COMMAND_ENQUEUE_USM_FREE_EXP , hQueue, CuStream, StreamToken));
97
97
UR_CHECK_ERROR (RetImplEvent->start ());
98
98
}
99
99
You can’t perform that action at this time.
0 commit comments