Merge pull request #31 from FluxML/bc/unthunk-hotfix

Hotfix: remove extra `unthunk_tangent` methods

Author: ToucheSir

Project.toml

@@ -1,6 +1,6 @@
 name = "ZygoteRules"
 uuid = "700de1a5-db45-46bc-99cf-38207098b444"
-version = "0.2.6"
+version = "0.2.7"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/adjoint.jl

@@ -37,12 +37,6 @@ function unthunk_tangent end
 @inline unthunk_tangent(x) = x
 @inline unthunk_tangent(x::Tuple) = map(unthunk_tangent, x)
 @inline unthunk_tangent(x::NamedTuple) = map(unthunk_tangent, x)
-@inline unthunk_tangent(x::AbstractThunk) = wrap_chainrules_output(unthunk(x))
-@inline unthunk_tangent(x::NTuple{N,<:Number}) where N = x
-@inline unthunk_tangent(x::AbstractArray{<:Number,N}) where N = x
-@inline unthunk_tangent(x::AbstractArray) = map(unthunk_tangent, x)
-unthunk_tangent(d::IdDict) = IdDict([unthunk_tangent(k) => unthunk_tangent(v) for (k, v) in d])
-@non_differentiable unthunk_tangent(::IdDict)
 
 
 function gradm(ex, mut = false, keepthunks = false)