@@ -34,7 +34,6 @@ Construct a cache for the Jacobian of `f` w.r.t. `u`.
34
34
fu
35
35
u
36
36
p
37
- alg
38
37
stats:: NLStats
39
38
autodiff
40
39
di_extras
@@ -106,18 +105,18 @@ function JacobianCache(prob, alg, f::F, fu_, u, p; stats, autodiff = nothing,
106
105
end
107
106
108
107
return JacobianCache {iip} (
109
- J, f, fu, u, p, alg, stats, autodiff, di_extras, sdifft_extras)
108
+ J, f, fu, u, p, stats, autodiff, di_extras, sdifft_extras)
110
109
end
111
110
112
111
function JacobianCache (prob, alg, f:: F , :: Number , u:: Number , p; stats,
113
112
autodiff = nothing , kwargs... ) where {F}
114
113
fu = f (u, p)
115
114
if SciMLBase. has_jac (f) || SciMLBase. has_vjp (f) || SciMLBase. has_jvp (f)
116
- return JacobianCache {false} (u, f, fu, u, p, alg, stats, autodiff, nothing )
115
+ return JacobianCache {false} (u, f, fu, u, p, stats, autodiff, nothing )
117
116
end
118
117
autodiff = get_concrete_forward_ad (autodiff, prob; check_forward_mode = false )
119
118
di_extras = DI. prepare_derivative (f, autodiff, u, Constant (prob. p))
120
- return JacobianCache {false} (u, f, fu, u, p, alg, stats, autodiff, di_extras, nothing )
119
+ return JacobianCache {false} (u, f, fu, u, p, stats, autodiff, di_extras, nothing )
121
120
end
122
121
123
122
(cache:: JacobianCache )(u = cache. u) = cache (cache. J, u, cache. p)
0 commit comments