Merge pull request #71 from JuliaGPU/vc/event_async

Run Event(f) on main thread

Author: vchuravy

Project.toml

@@ -1,7 +1,7 @@
 name = "KernelAbstractions"
 uuid = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
 authors = ["Valentin Churavy <[email protected]>"]
-version = "0.1.4"
+version = "0.1.5"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/backends/cpu.jl

@@ -6,11 +6,19 @@ function Event(::CPU)
     return NoneEvent()
 end
 
-function Event(f, args...; dependencies=nothing, progress=nothing)
-    T = Threads.@spawn begin
+"""
+    Event(f, args...; dependencies, progress, sticky=true)
+
+Run function `f` with `args` in a Julia task. If `sticky` is `true` the task
+is run on the thread that launched it.
+"""
+function Event(f, args...; dependencies=nothing, progress=nothing, sticky=true)
+    T = Task() do
         wait(MultiEvent(dependencies), progress)
         f(args...)
     end
+    T.sticky = sticky
+    Base.schedule(T)
     return CPUEvent(T)
 end