Skip to content

Commit

Permalink
update reversediff grad
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael Abbott committed Aug 16, 2020
1 parent dcfe9bf commit 9d54a24
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/grad/reverse.jl
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,6 @@ ReverseDiff.@grad function (ev::Eval)(args...)
Z = ev.fwd(ReverseDiff.value.(args)...)
Z, Δ -> begin
isnothing(ev.rev) && error("no gradient definition here!")
ev.rev(Δ, ReverseDiff.value.(args)...)
ev.rev(ReverseDiff.value(Δ), Z, ReverseDiff.value.(args)...)
end
end

0 comments on commit 9d54a24

Please sign in to comment.