Skip to content

Commit 617ff43

Browse files
committed
Avoid reshaping of data view in getindex for VectorOfVectors
1 parent 4eaced0 commit 617ff43

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/vector_of_arrays.jl

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,11 +183,17 @@ Base.size(A::VectorOfArrays) = size(A.kernel_size)
183183
Base.IndexStyle(A::VectorOfArrays) = IndexLinear()
184184

185185

186+
Base.@propagate_inbounds _reshape_dataview(dataview::AbstractArray, s::NTuple{1,Integer}) = dataview
187+
188+
Base.@propagate_inbounds _reshape_dataview(dataview::AbstractArray, s::NTuple{N,Integer}) where {N} =
189+
Base.__reshape((dataview, IndexStyle(dataview)), s)
190+
191+
186192
Base.@propagate_inbounds function Base.getindex(A::VectorOfArrays, i::Integer)
187193
@boundscheck checkbounds(A, i)
188194
r, s = _elem_range_size(A, i)
189195
dataview = view(A.data, r)
190-
Base.__reshape((dataview, IndexStyle(dataview)), s)
196+
_reshape_dataview(dataview, s)
191197
end
192198

193199

0 commit comments

Comments
 (0)