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)