Iterator norm in isapprox for Arrays

Author: jishnub

src/generic.jl

@@ -2004,7 +2004,7 @@ function isapprox(x::AbstractArray, y::AbstractArray;
     atol::Real=0,
     rtol::Real=Base.rtoldefault(promote_leaf_eltypes(x),promote_leaf_eltypes(y),atol),
     nans::Bool=false, norm::Function=norm)
-    d = norm(x - y)
+    d = norm_x_minus_y(x, y)
     if isfinite(d)
         return iszero(rtol) ? d <= atol : d <= max(atol, rtol*max(norm(x), norm(y)))
     else
@@ -2014,6 +2014,12 @@ function isapprox(x::AbstractArray, y::AbstractArray;
     end
 end
 
+norm_x_minus_y(x, y) = norm(x - y)
+function norm_x_minus_y(x::Array, y::Array)
+    Base.promote_shape(size(x), size(y)) # ensure same size
+    norm(Iterators.map(splat(-), zip(x,y)))
+end
+
 """
     normalize!(a::AbstractArray, p::Real=2)
 

test/generic.jl

@@ -938,4 +938,11 @@ end
     @test B == A2
 end
 
+@testset "isapprox alloc" begin
+    A = rand(3,3)
+    isapprox(A, A)
+    n = @allocated isapprox(A, A)
+    @test n == 0
+end
+
 end # module TestGeneric