Skip to content

Commit 43f83a4

Browse files
committed
[UR][CUDA] Fix stream guard in async alloc
1 parent bd47804 commit 43f83a4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

unified-runtime/source/adapters/cuda/async_alloc.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueUSMDeviceAllocExp(
2525

2626
ScopedContext Active(hQueue->getDevice());
2727
uint32_t StreamToken;
28-
ur_stream_guard_ Guard;
28+
ur_stream_guard Guard;
2929
CUstream CuStream = hQueue->getNextComputeStream(
3030
numEventsInWaitList, phEventWaitList, Guard, &StreamToken);
3131

@@ -83,7 +83,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueUSMFreeExp(
8383

8484
ScopedContext Active(hQueue->getDevice());
8585
uint32_t StreamToken;
86-
ur_stream_guard_ Guard;
86+
ur_stream_guard Guard;
8787
CUstream CuStream = hQueue->getNextComputeStream(
8888
numEventsInWaitList, phEventWaitList, Guard, &StreamToken);
8989

0 commit comments

Comments
 (0)