Skip to content

Commit 8ffb279

Browse files
remove the reverse
1 parent b0f6676 commit 8ffb279

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ext/RecursiveArrayToolsZygoteExt.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ end
1313
# Define a new species of projection operator for this type:
1414
ChainRulesCore.ProjectTo(x::VectorOfArray) = ChainRulesCore.ProjectTo{VectorOfArray}()
1515

16+
Zygote._tryreverse(m::typeof(map), x::AbstractVectorOfArray) = x
17+
1618
function ChainRulesCore.rrule(T::Type{<:RecursiveArrayTools.GPUArraysCore.AbstractGPUArray},
1719
xs::AbstractVectorOfArray)
1820
T(xs), ȳ -> (ChainRulesCore.NoTangent(), ȳ)

0 commit comments

Comments
 (0)