Improve count performance (JuliaLang#40564)

koekjesdoos · Nov 18, 2021 · ebcce3f · ebcce3f
1 parent 85f4db2
commit ebcce3f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/base/reduce.jl b/base/reduce.jl
@@ -1221,10 +1221,12 @@ count(itr; init=0) = count(identity, itr; init)
 
 count(f, itr; init=0) = _simple_count(f, itr, init)
 
-function _simple_count(pred, itr, init::T) where {T}
+_simple_count(pred, itr, init) = _simple_count_helper(Generator(pred, itr), init)
+
+function _simple_count_helper(g, init::T) where {T}
     n::T = init
-    for x in itr
-        n += pred(x)::Bool
+    for x in g
+        n += x::Bool
     end
     return n
 end