From 6bbed07e96048503fad3dfd9dd3000b37781506c Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 7 Sep 2018 02:05:03 +0100 Subject: [PATCH] enable nested broadcast --- src/tracker/array.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/tracker/array.jl b/src/tracker/array.jl index ffa3a89e..16f91d22 100644 --- a/src/tracker/array.jl +++ b/src/tracker/array.jl @@ -351,9 +351,9 @@ end eltype(y) <: Real || return y eltype(y) == Bool && return y function back(Δ) - Δargs = ntuple(i -> partial.(f, data(Δ), i, args...), Val(N)) - dxs = unbroadcast.(args, Δargs) - return nobacksies(:broadcast, dxs) + Δargs = ntuple(i -> partial.(f, Δ, i, args...), Val(N)) + dxs = map(unbroadcast, args, Δargs) + return dxs end # So we can return non-tracked arrays track(Call(back, tracker.(args)), y)