Skip to content

Commit f6f4689

Browse files
feat: implement initialization for no-init cache
1 parent b0a4ba1 commit f6f4689

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

lib/NonlinearSolveBase/src/solve.jl

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,11 @@ end
246246
alg
247247
args
248248
kwargs::Any
249+
initializealg
250+
end
251+
252+
function update_parameter_object!(cache::NonlinearSolveNoInitCache, p)
253+
SciMLBase.reinit!(cache, cache.prob.u0, p)
249254
end
250255

251256
get_u(cache::NonlinearSolveNoInitCache) = SII.state_values(cache.prob)
@@ -264,9 +269,12 @@ end
264269

265270
function SciMLBase.__init(
266271
prob::AbstractNonlinearProblem, alg::AbstractNonlinearSolveAlgorithm, args...;
272+
initializealg = NonlinearSolveDefaultInit(),
267273
kwargs...
268274
)
269-
return NonlinearSolveNoInitCache(prob, alg, args, kwargs)
275+
cache = NonlinearSolveNoInitCache(prob, alg, args, kwargs, initializealg)
276+
initialize_cache!(cache)
277+
return cache
270278
end
271279

272280
function CommonSolve.solve!(cache::NonlinearSolveNoInitCache)

0 commit comments

Comments
 (0)