Revert "Update vector_of_array.jl"

This reverts commit 49d6742.
SciML · Feb 27, 2024 · 96a09f3 · 96a09f3
1 parent c668130
commit 96a09f3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
@@ -270,7 +270,13 @@ __parameterless_type(T) = Base.typename(T).wrapper
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},
     ::NotSymbolic, I::Colon...) where {T, N}
     @assert length(I) == ndims(A.u[1]) + 1
-    return Array(A)
+    vecs = if N == 1
+        A.u
+    else
+        vec.(A.u)
+    end
+    return Adapt.adapt(__parameterless_type(T),
+        reshape(reduce(hcat, vecs), size(A.u[1])..., length(A.u)))
 end
 
 Base.@propagate_inbounds function _getindex(A::AbstractVectorOfArray{T, N},