add reduce dims

rayegun · rayegun · commit af215c38e021 · 2022-03-03T00:17:13.000-05:00
diff --git a/src/SuiteSparseGraphBLAS.jl b/src/SuiteSparseGraphBLAS.jl
@@ -84,7 +84,7 @@ include("chainrules/maprules.jl")
 include("chainrules/reducerules.jl")
 include("chainrules/selectrules.jl")
 include("chainrules/constructorrules.jl")
-
+include("chainrules/utilityrules.jl")
 #EXPERIMENTAL
 include("misc.jl")
 include("asjulia.jl")
diff --git a/src/chainrules/utilityrules.jl b/src/chainrules/utilityrules.jl
diff --git a/src/operations/reduce.jl b/src/operations/reduce.jl
@@ -77,4 +77,4 @@ Base.maximum(f::Function, A::GBArray) = reduce(max, map(f, A))
 Base.minimum(A::GBArray) = reduce(min, A)
 Base.minimum(f::Function, A::GBArray) = reduce(min, map(f, A))
 
-Base.sum(A::GBArray) = reduce(+, A)
+Base.sum(A::GBArray; dims=:) = reduce(+, A; dims)
