Fix asum for the case where Δx is a matrix

JuliaDiff · Jan 12, 2020 · c1a4451 · c1a4451
1 parent 63b9f7e
commit c1a4451
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/rulesets/LinearAlgebra/blas.jl b/src/rulesets/LinearAlgebra/blas.jl
@@ -68,10 +68,7 @@ end
 #####
 
 function frule(::typeof(BLAS.asum), x, _, Δx)
-    return BLAS.asum(x), sum(zip(x, Δx)) do xs
-        x, Δx = xs
-        return sign(x) * Δx
-    end
+    return BLAS.asum(x), sum(sign.(x) .* Δx)
 end
 
 function rrule(::typeof(BLAS.asum), x)