From e9aa66e84eee3bfee1a04f83ee950cf6bbfe5a36 Mon Sep 17 00:00:00 2001 From: Pietro Vertechi Date: Thu, 16 Jul 2020 16:47:16 +0100 Subject: [PATCH] simplify sum rrule (#239) * simplify sum rrule * bump version --- Project.toml | 2 +- src/rulesets/Base/mapreduce.jl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 7ee353236..07895d043 100644 --- a/Project.toml +++ b/Project.toml @@ -1,6 +1,6 @@ name = "ChainRules" uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2" -version = "0.7.9" +version = "0.7.10" [deps] ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" diff --git a/src/rulesets/Base/mapreduce.jl b/src/rulesets/Base/mapreduce.jl index f4a71fb3e..33eae96d1 100644 --- a/src/rulesets/Base/mapreduce.jl +++ b/src/rulesets/Base/mapreduce.jl @@ -7,7 +7,7 @@ function frule((_, ẋ), ::typeof(sum), x; dims=:) end function rrule(::typeof(sum), x::AbstractArray{T}; dims=:) where {T<:Number} - y = sum(sum, x; dims=dims) + y = sum(x; dims=dims) function sum_pullback(ȳ) # broadcasting the two works out the size no-matter `dims` x̄ = broadcast(x, ȳ) do xi, ȳi