don't wait in consume if task is done

TuringLang · Mar 6, 2019 · 6b5a0ac · 6b5a0ac
1 parent 31e624f
commit 6b5a0ac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/taskcopy.jl b/src/taskcopy.jl
@@ -174,6 +174,9 @@ consume(p::Task, values...) = begin
         isa(p.storage, IdDict) && haskey(p.storage, :_libtask_state) &&
             (p.state = p.storage[:_libtask_state])
 
+        if p.state == :done
+            return p.result
+        end
         if p.exception != nothing
             throw(p.exception)
         end