Specialize Base.copy for VectorOfArrays

Author: oschulz

src/vector_of_arrays.jl

@@ -345,6 +345,11 @@ function Base.vcat(Vs::(VectorOfArrays{U,N} where U)...) where {N}
 end
 
 
+function Base.copy(V::VectorOfArrays)
+    VectorOfArrays(copy(V.data), copy(V.elem_ptr), copy(V.kernel_size), no_consistency_checks)
+end
+
+
 Base.@propagate_inbounds function Base.unsafe_view(A::VectorOfArrays, idxs::AbstractUnitRange{<:Integer})
     from = first(idxs)
     to = last(idxs)

test/vector_of_arrays.jl

@@ -100,6 +100,16 @@ using ArraysOfArrays: full_consistency_checks, append_elemptr!
     end
 
 
+    @testset "copy" begin
+        A = ref_AoA3(Float32, 3);
+        B = VectorOfArrays(A);
+
+        @test typeof(@inferred copy(A)) == typeof(A)
+        @test copy(A) == A
+        @test copy(A) == B
+    end
+
+
     @testset "examples" begin
         VA = VectorOfArrays{Float64, 2}()