Merge #474

474: use IdDict in grad_mut fallback for IdDict r=CarloLucibello a=ChrisRackauckas Fixes hashing issues with CuArrays, i.e. FluxML/Flux.jl#989 Co-authored-by: Chris Rackauckas <[email protected]>
FluxML · Feb 27, 2020 · 200ae53 · 200ae53
2 parents af498fa + d47a712
commit 200ae53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/lib/base.jl b/src/lib/base.jl
@@ -22,6 +22,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)