Merge 8fbe739 into 4136420

JuliaDiff · Oct 28, 2019 · 136d3c8 · 136d3c8
2 parents 4136420 + 8fbe739
commit 136d3c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "ChainRules"
 uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2"
-version = "0.2.2-DEV"
+version = "0.2.3"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

diff --git a/src/rulesets/Base/mapreduce.jl b/src/rulesets/Base/mapreduce.jl
@@ -57,7 +57,7 @@ function frule(::typeof(sum), x)
     return sum(x), sum_pushforward
 end
 
-function rrule(::typeof(sum), x)
+function rrule(::typeof(sum), x::AbstractArray{<:Real})
     function sum_pullback(ȳ)
         return (NO_FIELDS, @thunk(fill(ȳ, size(x))))
     end
@@ -67,15 +67,15 @@ end
 function rrule(::typeof(sum), f, x::AbstractArray{<:Real}; dims=:)
     y, mr_pullback = rrule(mapreduce, f, Base.add_sum, x; dims=dims)
     function sum_pullback(ȳ)
-        NO_FIELDS, DNE(), last(mr_pullback(ȳ))
+        return NO_FIELDS, DNE(), last(mr_pullback(ȳ))
     end
     return y, sum_pullback
 end
 
 function rrule(::typeof(sum), x::AbstractArray{<:Real}; dims=:)
     y,  inner_pullback = rrule(sum, identity, x; dims=dims)
     function sum_pullback(ȳ)
-        NO_FIELDS, last(inner_pullback(ȳ))
+        return NO_FIELDS, last(inner_pullback(ȳ))
     end
     return y, sum_pullback
 end