Skip to content
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.

Sign up for GitHub

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

Jump to bottom

update scimlbase for function unwrapping #715

Merged
merged 12 commits into from
Aug 25, 2022
Merged

update scimlbase for function unwrapping #715

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
333404b
update scimlbase for function unwrapping
ChrisRackauckas Aug 24, 2022
20fafc7
unwrap f before Enzyme
ChrisRackauckas Aug 24, 2022
063ff07
more blind unwrapping
ChrisRackauckas Aug 24, 2022
90beb93
more direct unwrapping
ChrisRackauckas Aug 24, 2022
fe46839
unwrap in quadrature adjoint
ChrisRackauckas Aug 24, 2022
21e53c8
Update derivative_wrappers.jl
ChrisRackauckas Aug 24, 2022
cb5f58f
fix derivative wrapper dispatch
ChrisRackauckas Aug 24, 2022
6b945e1
more fixes
ChrisRackauckas Aug 24, 2022
462402d
handle stochastic cases
ChrisRackauckas Aug 25, 2022
00c254b
handle shadowing cases
ChrisRackauckas Aug 25, 2022
2822a58
more shadowing cases
ChrisRackauckas Aug 25, 2022
fd451c2
fix shadowing
ChrisRackauckas Aug 25, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
more blind unwrapping
  • Loading branch information
@ChrisRackauckas
ChrisRackauckas committed Aug 24, 2022
commit 063ff078b06e6b7121942c6cb7143d74622b64a4
4 changes: 4 additions & 0 deletions src/adjoint_common.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,10 @@ function adjointdiffcache(g::G, sensealg, discrete, sol, dgdu::DG1, dgdp::DG2, f
else
@unpack u0, p, tspan = prob
end

# Remove any function wrappers: it breaks autodiff
f = unwrapped_f(f)

numparams = p === nothing || p === DiffEqBase.NullParameters() ? 0 : length(p)
numindvar = length(u0)
isautojacvec = get_jacvec(sensealg)
Expand Down