Skip to content

Commit 72d705a

Browse files
Merge pull request #437 from SciML/ChrisRackauckas-patch-1
Make VectorOfArray adapt not broadcast the `to`
2 parents 242f370 + a2a0e41 commit 72d705a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/vector_of_array.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,11 @@ end
143143
Base.convert(::Type{AbstractArray}, VA::AbstractVectorOfArray) = stack(VA.u)
144144

145145
function Adapt.adapt_structure(to, VA::AbstractVectorOfArray)
146-
VectorOfArray(Adapt.adapt.(to, VA.u))
146+
VectorOfArray(Adapt.adapt.((to,), VA.u))
147147
end
148148

149149
function Adapt.adapt_structure(to, VA::AbstractDiffEqArray)
150-
DiffEqArray(Adapt.adapt.(to, VA.u), Adapt.adapt(to, VA.t))
150+
DiffEqArray(Adapt.adapt.((to,), VA.u), Adapt.adapt(to, VA.t))
151151
end
152152

153153
function VectorOfArray(vec::AbstractVector{T}, ::NTuple{N}) where {T, N}

0 commit comments

Comments
 (0)