You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The only backends concerned by this are AutoEnzyme(Enzyme.ReverseModeSplit) and AutoTapir.
For mutating functions, it is dangerous to interrupt reverse mode autodiff after the forward sweep. The variables are in an intermediate state which will only be reverted after the reverse sweep completes.
In particular, Tapir requires running another forward sweep between reverse sweep, so a pullback closure would be useless.
I'm not sure how split reverse Enzyme works, will have to investigate.
The only backends concerned by this are
AutoEnzyme(Enzyme.ReverseModeSplit)
andAutoTapir
.For mutating functions, it is dangerous to interrupt reverse mode autodiff after the forward sweep. The variables are in an intermediate state which will only be reverted after the reverse sweep completes.
In particular, Tapir requires running another forward sweep between reverse sweep, so a pullback closure would be useless.
I'm not sure how split reverse Enzyme works, will have to investigate.
Related:
The text was updated successfully, but these errors were encountered: