avoid vector allocation when PartitionIterator is cut short (#41950)

JuliaLang · Aug 24, 2021 · 5451c7d · 5451c7d
1 parent cdd2f30
commit 5451c7d
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