From 60c3090981379c0a9c8924dfd4c640b3c3c5afba Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 23 Aug 2017 17:21:02 +0100 Subject: [PATCH] broadcasting fix --- src/tracker/lib.jl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/tracker/lib.jl b/src/tracker/lib.jl index 4332f910..28929d73 100644 --- a/src/tracker/lib.jl +++ b/src/tracker/lib.jl @@ -78,6 +78,8 @@ end Base.Broadcast._containertype(::Type{<:TrackedArray}) = TrackedArray Base.Broadcast.promote_containertype(::Type{Array}, ::Type{TrackedArray}) = TrackedArray Base.Broadcast.promote_containertype(::Type{TrackedArray}, ::Type{Array}) = TrackedArray +Base.Broadcast.promote_containertype(::Type{TrackedArray}, ct) = TrackedArray +Base.Broadcast.promote_containertype(ct, ::Type{TrackedArray}) = TrackedArray Base.Broadcast.broadcast_indices(::Type{TrackedArray}, A::Ref) = () Base.Broadcast.broadcast_indices(::Type{TrackedArray}, A) = indices(A)