@opt_out of sum(array, array)

Author: Miha Zgubic

Project.toml

@@ -1,6 +1,6 @@
 name = "ChainRules"
 uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2"
-version = "1.11.3"
+version = "1.11.4"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/rulesets/Base/mapreduce.jl

@@ -88,6 +88,9 @@ function rrule(
     return y, sum_pullback
 end
 
+# https://github.com/JuliaDiff/ChainRules.jl/issues/522
+@opt_out ChainRulesCore.rrule(config::RuleConfig{>:HasReverseMode}, ::typeof(sum), x::AbstractArray, y::AbstractArray; dims=:)
+
 function frule(
     (_, _, Δx),
     ::typeof(sum),

test/rulesets/Base/mapreduce.jl

@@ -90,6 +90,16 @@
         @test_skip test_rrule(sum, iszero, randn(5))  # DimensionMismatch("second dimension of A, 1, does not match length of x, 0")
     end
 
+    # https://github.com/JuliaDiff/ChainRules.jl/issues/522
+    @begin "sum(xs, weights) (#522)" begin
+        xs = rand(5)
+        weights = rand(5)
+        Base.sum(xs::AbstractArray, weights::AbstractArray) = dot(xs, weights)
+        struct MyRuleConfig <: RuleConfig{Union{HasReverseMode}} end
+
+        @test rrule(MyRuleConfig(), Base.sum, xs, weights) isa Nothing
+    end
+
     @testset "prod" begin
         @testset "Array{$T}" for T in [Float64, ComplexF64]
             @testset "size = $sz, dims = $dims" for (sz, dims) in [