From 1f8cfa31f09894deab874c483d9f34f4ec53b6ee Mon Sep 17 00:00:00 2001
From: Valentin Churavy <v.churavy@gmail.com>
Date: Mon, 13 Mar 2023 20:49:38 -0400
Subject: [PATCH] fixup! implement stream priority

---
 src/CUDAKernels.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/CUDAKernels.jl b/src/CUDAKernels.jl
index 5054dda6f9..1eba4ab301 100644
--- a/src/CUDAKernels.jl
+++ b/src/CUDAKernels.jl
@@ -259,7 +259,6 @@ function KernelAbstractions.priority!(::CUDABackend, prio::Symbol)
     if !(prio in (:high, :normal, :low))
         error("priority must be one of :high, :normal, :low")
     end
-    return nothing
 
     range = CUDA.priority_range()
     # 0:-1:-5
@@ -280,9 +279,10 @@ function KernelAbstractions.priority!(::CUDABackend, prio::Symbol)
     CUDA.record(event, old_stream)
 
     @debug "Switching default stream" flags priority
-    new_stream = CuStream(; flags, priority))
+    new_stream = CuStream(; flags, priority)
     CUDA.wait(event, new_stream)
     stream!(new_stream)
+    return nothing
 end
 
 end