@@ -9,52 +9,53 @@ using SimpleNonlinearSolve: SimpleNonlinearSolve
99function SimpleNonlinearSolve. __internal_solve_up (
1010 prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg,
1111 u0:: TrackedArray , u0_changed, p:: TrackedArray , p_changed, alg, args... ; kwargs... )
12- return ReverseDiff. track (SimpleNonlinearSolve. __internal_solve_up, prob, sensealg, u0,
13- u0_changed, p, p_changed, alg, args... ; kwargs... )
12+ return ReverseDiff. track (SimpleNonlinearSolve. __internal_solve_up, prob, sensealg,
13+ u0, u0_changed, p, p_changed, alg, args... ; kwargs... )
1414end
1515
1616function SimpleNonlinearSolve. __internal_solve_up (
17- prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg, u0, u0_changed,
18- p:: TrackedArray , p_changed, alg, args... ; kwargs... )
19- return ReverseDiff. track (SimpleNonlinearSolve. __internal_solve_up, prob, sensealg, u0,
20- u0_changed, p, p_changed, alg, args... ; kwargs... )
17+ prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg,
18+ u0, u0_changed, p:: TrackedArray , p_changed, alg, args... ; kwargs... )
19+ return ReverseDiff. track (SimpleNonlinearSolve. __internal_solve_up, prob, sensealg,
20+ u0, u0_changed, p, p_changed, alg, args... ; kwargs... )
2121end
2222
2323function SimpleNonlinearSolve. __internal_solve_up (
2424 prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg,
2525 u0:: TrackedArray , u0_changed, p, p_changed, alg, args... ; kwargs... )
26- return ReverseDiff. track (SimpleNonlinearSolve. __internal_solve_up, prob, sensealg, u0,
27- u0_changed, p, p_changed, alg, args... ; kwargs... )
26+ return ReverseDiff. track (SimpleNonlinearSolve. __internal_solve_up, prob, sensealg,
27+ u0, u0_changed, p, p_changed, alg, args... ; kwargs... )
2828end
2929
3030function SimpleNonlinearSolve. __internal_solve_up (
31- prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg,
32- u0:: AbstractArray{<:TrackedReal} , u0_changed, p :: AbstractArray{<:TrackedReal} ,
33- p_changed, alg, args... ; kwargs... )
31+ prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} ,
32+ sensealg, u0:: AbstractArray{<:TrackedReal} , u0_changed,
33+ p :: AbstractArray{<:TrackedReal} , p_changed, alg, args... ; kwargs... )
3434 return SimpleNonlinearSolve. __internal_solve_up (
3535 prob, sensealg, ArrayInterface. aos_to_soa (u0), true ,
3636 ArrayInterface. aos_to_soa (p), true , alg, args... ; kwargs... )
3737end
3838
3939function SimpleNonlinearSolve. __internal_solve_up (
40- prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg, u0, u0_changed,
41- p:: AbstractArray{<:TrackedReal} , p_changed, alg, args... ; kwargs... )
40+ prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg, u0,
41+ u0_changed, p:: AbstractArray{<:TrackedReal} , p_changed, alg, args... ; kwargs... )
4242 return SimpleNonlinearSolve. __internal_solve_up (
43- prob, sensealg, u0, true , ArrayInterface. aos_to_soa (p), true , alg, args ... ;
44- kwargs... )
43+ prob, sensealg, u0, true , ArrayInterface. aos_to_soa (p),
44+ true , alg, args ... ; kwargs... )
4545end
4646
4747function SimpleNonlinearSolve. __internal_solve_up (
48- prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg,
49- u0:: AbstractArray{<:TrackedReal} , u0_changed, p, p_changed, alg, args... ; kwargs... )
48+ prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} ,
49+ sensealg, u0:: AbstractArray{<:TrackedReal} ,
50+ u0_changed, p, p_changed, alg, args... ; kwargs... )
5051 return SimpleNonlinearSolve. __internal_solve_up (
51- prob, sensealg, u0, true , ArrayInterface. aos_to_soa (p), true , alg, args ... ;
52- kwargs... )
52+ prob, sensealg, u0, true , ArrayInterface. aos_to_soa (p),
53+ true , alg, args ... ; kwargs... )
5354end
5455
5556ReverseDiff. @grad function SimpleNonlinearSolve. __internal_solve_up (
56- prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} , sensealg, u0,
57- u0_changed, p, p_changed, alg, args... ; kwargs... )
57+ prob:: Union{NonlinearProblem, NonlinearLeastSquaresProblem} ,
58+ sensealg, u0, u0_changed, p, p_changed, alg, args... ; kwargs... )
5859 out, ∇internal = DiffEqBase. _solve_adjoint (
5960 prob, sensealg, ReverseDiff. value (u0), ReverseDiff. value (p),
6061 ReverseDiffOriginator (), alg, args... ; kwargs... )
0 commit comments