Convert to the right type in similar

Signed-off-by: ErikQQY <[email protected]>
SciML · Sep 11, 2024 · c87bc1c · c87bc1c
1 parent fdd3e66
commit c87bc1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/vector_of_array.jl b/src/vector_of_array.jl
@@ -732,7 +732,7 @@ function Base.similar(vec::VectorOfArray{
 end
 
 @inline function Base.similar(VA::VectorOfArray{T1, N, AT}, ::Type{T2}) where {T1, T2, N, AT <: AbstractArray{<:AbstractArray{T1}}}
-    eltype(VA.u) <: Vector ? VectorOfArray(similar(VA.u, Vector{T1})) : VectorOfArray(similar(VA.u, Matrix{T1}))
+    eltype(VA.u) <: Vector ? VectorOfArray(similar(VA.u, Vector{T2})) : VectorOfArray(similar(VA.u, Matrix{T2}))
 end
 
 @inline function Base.similar(VA::VectorOfArray, dims::N) where {N}