Update vector_of_array.jl

Author: ChrisRackauckas

src/vector_of_array.jl

@@ -143,7 +143,11 @@ end
 Base.convert(::Type{AbstractArray}, VA::AbstractVectorOfArray) = stack(VA.u)
 
 function Adapt.adapt_structure(to, VA::AbstractVectorOfArray)
-    Adapt.adapt.(to, VA.u)
+    VectorOfArray(Adapt.adapt.(to, VA.u))
+end
+
+function Adapt.adapt_structure(to, VA::AbstractDiffEqArray)
+    DiffEqArray(Adapt.adapt.(to, VA.u), Adapt.adapt(to, VA.t))
 end
 
 function VectorOfArray(vec::AbstractVector{T}, ::NTuple{N}) where {T, N}