-
Notifications
You must be signed in to change notification settings - Fork 219
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Make setadbackend
more consistent + use ZygoteRules
#1401
Conversation
Codecov Report
@@ Coverage Diff @@
## master #1401 +/- ##
==========================================
+ Coverage 66.95% 67.01% +0.06%
==========================================
Files 25 25
Lines 1619 1616 -3
==========================================
- Hits 1084 1083 -1
+ Misses 535 533 -2
Continue to review full report at Codecov.
|
This PR moves the definition of the adjoint for `updategid!` to DynamicPPL and replaces ZygoteRules with ChainRulesCore, as mentioned in a comment in TuringLang/Turing.jl#1401.
This PR moves the definition of the adjoint for `updategid!` to DynamicPPL and replaces ZygoteRules with ChainRulesCore, as mentioned in a comment in TuringLang/Turing.jl#1401.
PkgServer lags behind (again)... The new DynamicPPL release is not available in CI yet. Just have to retry until it works. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thanks!
This PR fixes an inconsistency observed in #1400 which leads to different AD settings depending on whether
setadbackend(::Symbol)
orsetadbackend(::Val)
is called. Moreover, it separates the deprecations into a separate file so that (hopefully) we don't forget to remove them in the next breaking release. Additionally, ZygoteRules is used instead of Zygote inside of a@require
block, hopefully eliminating potential precompilation and loading time issues for Zygote similar to those observed for ReverseDiff in #1400.