Remove GPU sum() rule

FluxML · Sep 1, 2023 · 73ebf4b · 73ebf4b
1 parent 69c2616
commit 73ebf4b
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/lib/broadcast.jl b/src/lib/broadcast.jl
@@ -364,11 +364,6 @@ using GPUArraysCore  # replaces @require CUDA block, weird indenting to preserve
   @adjoint (::Type{T})(xs::Array) where {T <: AbstractGPUArray} =
     T(xs), Δ -> (convert(Array, Δ), )
 
-  @adjoint function sum(xs::AbstractGPUArray; dims = :)
-    placeholder = similar(xs)
-    sum(xs, dims = dims), Δ -> (placeholder .= Δ,)
-  end
-
   # Make sure sum(f, ::CuArray) uses broadcast through forward-mode defined above
   # Not the ChainRules.rrule which will use the Zygote.Context and thus not be GPU compatible
   function _pullback(cx::AContext, ::typeof(sum), f, xs::AbstractGPUArray)