Fix GPU conversions for AbstractVectorOfArray

SciML · Dec 27, 2023 · 63421ad · 63421ad
1 parent c5f4cc6
commit 63421ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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.2.7"
+version = "3.2.8"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

diff --git a/src/RecursiveArrayTools.jl b/src/RecursiveArrayTools.jl
@@ -27,7 +27,7 @@ function Base.show(io::IO, x::Union{ArrayPartition, AbstractVectorOfArray})
 end
 
 import GPUArraysCore
-Base.convert(T::Type{<:GPUArraysCore.AbstractGPUArray}, VA::AbstractVectorOfArray) = T(VA)
+Base.convert(T::Type{<:GPUArraysCore.AnyGPUArray}, VA::AbstractVectorOfArray) = stack(VA.u)
 
 import Requires
 @static if !isdefined(Base, :get_extension)