From c1a4451032856f5262cf8d17e91f0e75d97698cd Mon Sep 17 00:00:00 2001 From: Yingbo Ma Date: Sun, 12 Jan 2020 16:59:12 -0500 Subject: [PATCH] =?UTF-8?q?Fix=20asum=20for=20the=20case=20where=20=CE=94x?= =?UTF-8?q?=20is=20a=20matrix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/rulesets/LinearAlgebra/blas.jl | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/rulesets/LinearAlgebra/blas.jl b/src/rulesets/LinearAlgebra/blas.jl index bb0734e52..77940fbed 100644 --- 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)