Revert "Don't unthunk explicitly in unbroadcast"

This reverts commit 34865ea.
FluxML · Mar 11, 2022 · ea2dafa · ea2dafa
1 parent e6aa6d0
commit ea2dafa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lib/broadcast.jl b/src/lib/broadcast.jl
@@ -45,7 +45,8 @@ function Base.reducedim_init(::typeof(identity), ::typeof(accum), A::AbstractArr
   Base.reducedim_initarray(A, region, nothing, Union{Nothing,eltype(A)})
 end
 
-function unbroadcast(x::AbstractArray, x̄)
+function unbroadcast(x::AbstractArray, maybethunked_x̄)
+  x̄ = unthunk_tangent(maybethunked_x̄)
   N = ndims(x̄)
   if length(x) == length(x̄)
     _project(x, x̄)  # ProjectTo handles reshape, offsets, structured matrices, row vectors