Skip to content

Commit 1f8cfa3

Browse files
committed
fixup! implement stream priority
1 parent 9406496 commit 1f8cfa3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/CUDAKernels.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,6 @@ function KernelAbstractions.priority!(::CUDABackend, prio::Symbol)
259259
if !(prio in (:high, :normal, :low))
260260
error("priority must be one of :high, :normal, :low")
261261
end
262-
return nothing
263262

264263
range = CUDA.priority_range()
265264
# 0:-1:-5
@@ -280,9 +279,10 @@ function KernelAbstractions.priority!(::CUDABackend, prio::Symbol)
280279
CUDA.record(event, old_stream)
281280

282281
@debug "Switching default stream" flags priority
283-
new_stream = CuStream(; flags, priority))
282+
new_stream = CuStream(; flags, priority)
284283
CUDA.wait(event, new_stream)
285284
stream!(new_stream)
285+
return nothing
286286
end
287287

288288
end

0 commit comments

Comments
 (0)