Fix more ambiguity

FluxML · Nov 13, 2023 · 81a210a · 81a210a
1 parent 1e7fca1
commit 81a210a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "Tracker"
 uuid = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
-version = "0.2.28"
+version = "0.2.29"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

diff --git a/src/lib/array.jl b/src/lib/array.jl
@@ -184,6 +184,11 @@ for (T, S) in [(:TrackedReal, :Real), (:Real, :TrackedReal), (:TrackedReal, :Tra
 end
 Base.vcat(A::TrackedVecOrMat{T1, <:AbstractArray}, B::TrackedVecOrMat{T2, <:AbstractArray}) where {T1, T2} = track(vcat, A, B)
 Base.hcat(A::TrackedVecOrMat{T1, <:AbstractArray}, B::TrackedVecOrMat{T2, <:AbstractArray}) where {T1, T2} = track(hcat, A, B)
+Base.vcat(A::TrackedVector{<:Number, <:AbstractVector{<:Number}}, B::TrackedVector{<:Number, <:AbstractVector{<:Number}}) = track(vcat, A, B)
+Base.vcat(A::TrackedMatrix{<:Number, <:AbstractMatrix{<:Number}}, B::AbstractMatrix{<:Number}) = track(vcat, A, B)
+Base.vcat(A::TrackedMatrix{<:Number, <:AbstractMatrix{<:Number}}, B::TrackedMatrix{<:Number, <:AbstractMatrix{<:Number}}) = track(vcat, A, B)
+Base.vcat(A::TrackedVector{<:Number}, B::AbstractVector{<:Number}) = track(vcat, A, B)
+Base.vcat(A::AbstractVector{<:Number}, B::TrackedVector{<:Number}) = track(vcat, A, B)
 
 Base.vcat(A::TrackedArray) = track(vcat, A)
 Base.hcat(A::TrackedArray) = track(hcat, A)