device agnostic convert

SciML · May 14, 2024 · 9bf6c85 · 9bf6c85
1 parent 3957332
commit 9bf6c85
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "RecursiveArrayTools"
 uuid = "731186ca-8d62-57ce-b412-fbd966d074cd"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "3.16.0"
+version = "3.16.1"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
@@ -135,6 +135,12 @@ function Base.Array{U}(VA::AbstractVectorOfArray) where {U}
     vecs = vec.(VA.u)
     Array(reshape(reduce(hcat, vecs), size(VA.u[1])..., length(VA.u)))
 end
+
+function Base.convert(::Type{AbstractArray}, VA::AbstractVectorOfArray)
+    vecs = vec.(VA.u)
+    return reshape(stack(vecs), size(VA.u[1])..., length(VA.u))
+end
+
 function Adapt.adapt_structure(to, VA::AbstractVectorOfArray)
     Adapt.adapt(to, Array(VA))
 end