a little shorter

2016-06-03 00:57:52 +01:00 · 2016-06-03 00:57:52 +01:00 · dfc75df407
commit dfc75df407
parent 8b7d575efe
1 changed files with 2 additions and 5 deletions
--- a/src/rt/diff.jl
+++ b/src/rt/diff.jl
@ -24,10 +24,7 @@ function ∇graph(v::IVertex, ∇, out = d())
 end

 macro derive(ex)
-  v = vertex(Flow.Do())
-  for (k, x) in ∇graph(il(graphm(resolve_calls(ex))), @flow(∇))
-    k = Symbol("∇", k)
-    thread!(v, @v(Flow.Assign(k)(x)))
-  end
+  ∇s = ∇graph(il(graphm(resolve_calls(ex))), @flow(∇))
+  v = vertex(Flow.Do(), (@v(Flow.Assign(Symbol("∇", k))(v)) for (k, v) in ∇s)...)
  Expr(:quote, @> v cse syntax prettify)
 end