Take adjoint inside FlatteningRF as well

JuliaLang · Dec 4, 2019 · ce49c7e · ce49c7e
1 parent af0b06c
commit ce49c7e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/base/reduce.jl b/base/reduce.jl
@@ -111,7 +111,10 @@ struct FlatteningRF{T}
     rf::T
 end
 
-@inline (op::FlatteningRF)(acc, x) = _foldl_impl(op.rf, acc, x)
+@inline function (op::FlatteningRF)(acc, x)
+    op′, itr′ = _xfadjoint(BottomRF(op.rf), x)
+    return _foldl_impl(op′, acc, itr′)
+end
 
 """
     _xfadjoint(op, itr) -> op′, itr′