avoid vector allocation when PartitionIterator is cut short (JuliaLan…

…g#41950)
LilithHafner · Mar 8, 2022 · 726af79 · 726af79
1 parent 2bea4df
commit 726af79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/base/iterators.jl b/base/iterators.jl
@@ -1191,11 +1191,11 @@ end
 struct IterationCutShort; end
 
 function iterate(itr::PartitionIterator, state...)
-    v = Vector{eltype(itr.c)}(undef, itr.n)
     # This is necessary to remember whether we cut the
     # last element short. In such cases, we do return that
     # element, but not the next one
     state === (IterationCutShort(),) && return nothing
+    v = Vector{eltype(itr.c)}(undef, itr.n)
     i = 0
     y = iterate(itr.c, state...)
     while y !== nothing