Avoid allocations in mapreduce with multiple arrays

Should fix JuliaLang#53417, seeing if it passes tests.
raminammour · Feb 21, 2024 · ad7b4ad · ad7b4ad
1 parent 65f24da
commit ad7b4ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/base/reducedim.jl b/base/reducedim.jl
@@ -327,7 +327,7 @@ julia> mapreduce(isodd, |, a, dims=1)
 mapreduce(f, op, A::AbstractArrayOrBroadcasted; dims=:, init=_InitialValue()) =
     _mapreduce_dim(f, op, init, A, dims)
 mapreduce(f, op, A::AbstractArrayOrBroadcasted...; kw...) =
-    reduce(op, map(f, A...); kw...)
+    reduce(op, Iterators.map(f, A...); kw...)
 
 _mapreduce_dim(f, op, nt, A::AbstractArrayOrBroadcasted, ::Colon) =
     mapfoldl_impl(f, op, nt, A)