Skip to content

Commit 7e3610e

Browse files
committed
use full form
1 parent d63e00b commit 7e3610e

File tree

2 files changed

+18
-16
lines changed

2 files changed

+18
-16
lines changed

src/common_defaults.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ end
4242
end
4343
@inline ODE_DEFAULT_NORM(u, t) = norm(u)
4444

45-
@inline NLSOLVE_DEFAULT_NORM(u) = ODE_DEFAULT_NORM(u, nothing)
45+
@inline NONLINEARSOLVE_DEFAULT_NORM(u) = ODE_DEFAULT_NORM(u, nothing)
4646

4747
@inline ODE_DEFAULT_ISOUTOFDOMAIN(u, p, t) = false
4848
@inline function ODE_DEFAULT_PROG_MESSAGE(dt, u::Array, p, t)

src/termination_conditions.jl

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -109,11 +109,11 @@ end
109109
function (cache::NonlinearTerminationModeCache)(mode::AbstractSafeNonlinearTerminationMode,
110110
du, u, uprev)
111111
if mode isa AbsSafeTerminationMode || mode isa AbsSafeBestTerminationMode
112-
objective = NLSOLVE_DEFAULT_NORM(du)
112+
objective = NONLINEARSOLVE_DEFAULT_NORM(du)
113113
criteria = cache.abstol
114114
else
115-
objective = NLSOLVE_DEFAULT_NORM(du) /
116-
(NLSOLVE_DEFAULT_NORM(du .+ u) + eps(cache.abstol))
115+
objective = NONLINEARSOLVE_DEFAULT_NORM(du) /
116+
(NONLINEARSOLVE_DEFAULT_NORM(du .+ u) + eps(cache.abstol))
117117
criteria = cache.reltol
118118
end
119119

@@ -138,7 +138,7 @@ function (cache::NonlinearTerminationModeCache)(mode::AbstractSafeNonlinearTermi
138138
if objective cache.mode.patience_objective_multiplier * criteria
139139
if cache.nsteps cache.mode.patience_steps
140140
if cache.nsteps < length(cache.objectives_trace)
141-
min_obj, max_obj = extrema(@view(cache.objectives_trace[1:cache.nsteps]))
141+
min_obj, max_obj = extrema(@view(cache.objectives_trace[1:(cache.nsteps)]))
142142
else
143143
min_obj, max_obj = extrema(cache.objectives_trace)
144144
end
@@ -169,22 +169,23 @@ function check_convergence(::SimpleNonlinearSolveTerminationMode, duₙ, uₙ, u
169169
isapprox(uₙ, uₙ₋₁; atol = abstol, rtol = reltol)
170170
end
171171
function check_convergence(::NormTerminationMode, duₙ, uₙ, uₙ₋₁, abstol, reltol)
172-
du_norm = NLSOLVE_DEFAULT_NORM(duₙ)
173-
return du_norm abstol || du_norm reltol * NLSOLVE_DEFAULT_NORM(duₙ .+ uₙ)
172+
du_norm = NONLINEARSOLVE_DEFAULT_NORM(duₙ)
173+
return du_norm abstol || du_norm reltol * NONLINEARSOLVE_DEFAULT_NORM(duₙ .+ uₙ)
174174
end
175175
function check_convergence(::RelNormTerminationMode, duₙ, uₙ, uₙ₋₁, abstol, reltol)
176176
return all(abs.(duₙ) .≤ reltol .* abs.(uₙ))
177177
end
178178
function check_convergence(::Union{RelNormTerminationMode, RelSafeTerminationMode,
179179
RelSafeBestTerminationMode}, duₙ, uₙ, uₙ₋₁, abstol, reltol)
180-
return NLSOLVE_DEFAULT_NORM(duₙ) reltol * NLSOLVE_DEFAULT_NORM(duₙ .+ uₙ)
180+
return NONLINEARSOLVE_DEFAULT_NORM(duₙ)
181+
reltol * NONLINEARSOLVE_DEFAULT_NORM(duₙ .+ uₙ)
181182
end
182183
function check_convergence(::AbsTerminationMode, duₙ, uₙ, uₙ₋₁, abstol, reltol)
183184
return all(abs.(duₙ) .≤ abstol)
184185
end
185186
function check_convergence(::Union{AbsNormTerminationMode, AbsSafeTerminationMode,
186187
AbsSafeBestTerminationMode}, duₙ, uₙ, uₙ₋₁, abstol, reltol)
187-
return NLSOLVE_DEFAULT_NORM(duₙ) abstol
188+
return NONLINEARSOLVE_DEFAULT_NORM(duₙ) abstol
188189
end
189190

190191
# NOTE: Deprecate the following API eventually. This API leads to quite a bit of type
@@ -363,11 +364,11 @@ function (cond::NLSolveTerminationCondition)(storage::Union{
363364
end
364365

365366
if mode SAFE_BEST_TERMINATION_MODES
366-
objective = NLSOLVE_DEFAULT_NORM(du)
367+
objective = NONLINEARSOLVE_DEFAULT_NORM(du)
367368
criteria = abstol
368369
else
369-
objective = NLSOLVE_DEFAULT_NORM(du) /
370-
(NLSOLVE_DEFAULT_NORM(du .+ u) + eps(aType))
370+
objective = NONLINEARSOLVE_DEFAULT_NORM(du) /
371+
(NONLINEARSOLVE_DEFAULT_NORM(du .+ u) + eps(aType))
371372
criteria = reltol
372373
end
373374

@@ -426,18 +427,19 @@ end
426427

427428
@inline @inbounds function _has_converged(du, u, uprev, mode, abstol, reltol)
428429
if mode == NLSolveTerminationMode.Norm
429-
du_norm = NLSOLVE_DEFAULT_NORM(du)
430-
return du_norm abstol || du_norm reltol * NLSOLVE_DEFAULT_NORM(du + u)
430+
du_norm = NONLINEARSOLVE_DEFAULT_NORM(du)
431+
return du_norm abstol || du_norm reltol * NONLINEARSOLVE_DEFAULT_NORM(du + u)
431432
elseif mode == NLSolveTerminationMode.Rel
432433
return all(abs.(du) .≤ reltol .* abs.(u))
433434
elseif mode (NLSolveTerminationMode.RelNorm, NLSolveTerminationMode.RelSafe,
434435
NLSolveTerminationMode.RelSafeBest)
435-
return NLSOLVE_DEFAULT_NORM(du) reltol * NLSOLVE_DEFAULT_NORM(du .+ u)
436+
return NONLINEARSOLVE_DEFAULT_NORM(du)
437+
reltol * NONLINEARSOLVE_DEFAULT_NORM(du .+ u)
436438
elseif mode == NLSolveTerminationMode.Abs
437439
return all(abs.(du) .≤ abstol)
438440
elseif mode (NLSolveTerminationMode.AbsNorm, NLSolveTerminationMode.AbsSafe,
439441
NLSolveTerminationMode.AbsSafeBest)
440-
return NLSOLVE_DEFAULT_NORM(du) abstol
442+
return NONLINEARSOLVE_DEFAULT_NORM(du) abstol
441443
elseif mode == NLSolveTerminationMode.SteadyStateDefault
442444
return all((abs.(du) .≤ abstol) .| (abs.(du) .≤ reltol .* abs.(u)))
443445
elseif mode == NLSolveTerminationMode.NLSolveDefault

0 commit comments

Comments
 (0)