From 227e41c37bfb469f582c8f936de6a6348f74aa4e Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Mon, 21 Aug 2017 16:35:39 +0100
Subject: [PATCH] gpu-friendly dualify

---
 src/Tracker/lib.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Tracker/lib.jl b/src/Tracker/lib.jl
index 3cfdd382..41b7e676 100644
--- a/src/Tracker/lib.jl
+++ b/src/Tracker/lib.jl
@@ -34,7 +34,7 @@ end
 (b::Broadcasted)(xs...) = map(x -> x.value, b.data)
 
 dualify(xs, n) = xs
-dualify(xs::TrackedArray, ps) = Dual.(data(xs), Ref(ps))
+dualify(xs::TrackedArray, ps) = map(x -> Dual(x, ps), data(xs))
 
 function tracked_broadcast(f, args::Vararg{Any,N}) where N
   dargs = map((x,i) -> dualify(x, ntuple(j -> i==j, Val{N})), args, ntuple(identity, Val{N}))