use IdDict in grad_mut fallback for IdDict

Fixes hashing issues with CuArrays
FluxML · Jan 20, 2020 · d47a712 · d47a712
1 parent 257a26e
commit d47a712
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/lib/base.jl b/src/lib/base.jl
@@ -24,6 +24,7 @@ end
 # Dictionaries
 
 grad_mut(d::AbstractDict) = Dict()
+grad_mut(d::IdDict) = IdDict()
 
 # TODO perhaps look up mutable gradients in `pullback`
 function accum(a::AbstractDict, b::AbstractDict)