Skip to content

Commit 6b30aad

Browse files
authored
fix subsolver naming in TrunkSolverNLS #304 (#305)
* updated the name krylov_subsolver
1 parent ad9970a commit 6b30aad

2 files changed

Lines changed: 14 additions & 14 deletions

File tree

src/trunk.jl

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ mutable struct TrunkSolver{
125125
gt::V
126126
gn::V
127127
Hs::V
128-
subsolver::Sub
128+
krylov_subsolver::Sub
129129
H::Op
130130
tr::TrustRegion{T, V}
131131
params::TRUNKParameterSet
@@ -146,12 +146,12 @@ function TrunkSolver(
146146
gt = V(undef, nvar)
147147
gn = isa(nlp, QuasiNewtonModel) ? V(undef, nvar) : V(undef, 0)
148148
Hs = V(undef, nvar)
149-
workspace = krylov_workspace(Val(subsolver), nvar, nvar, V)
150-
Sub = typeof(workspace)
149+
krylov_subsolver = krylov_workspace(Val(subsolver), nvar, nvar, V)
150+
Sub = typeof(krylov_subsolver)
151151
H = hess_op!(nlp, x, Hs)
152152
Op = typeof(H)
153153
tr = TrustRegion(gt, one(T))
154-
return TrunkSolver{T, V, Sub, Op}(x, xt, gx, gt, gn, Hs, workspace, H, tr, params)
154+
return TrunkSolver{T, V, Sub, Op}(x, xt, gx, gt, gn, Hs, krylov_subsolver, H, tr, params)
155155
end
156156

157157
function SolverCore.reset!(solver::TrunkSolver)
@@ -219,7 +219,7 @@ function SolverCore.solve!(
219219
∇f = solver.gx
220220
∇fn = solver.gn
221221
Hs = solver.Hs
222-
subsolver = solver.subsolver
222+
krylov_subsolver = solver.krylov_subsolver
223223
H = solver.H
224224
tr = solver.tr
225225

@@ -285,7 +285,7 @@ function SolverCore.solve!(
285285
cgtol = max(rtol, min(T(0.1), ∇fNormM, T(0.9) * cgtol))
286286
∇f .*= -1
287287
krylov_solve!(
288-
subsolver,
288+
krylov_subsolver,
289289
H,
290290
∇f,
291291
atol = atol,
@@ -296,7 +296,7 @@ function SolverCore.solve!(
296296
verbose = subsolver_verbose,
297297
M = M,
298298
)
299-
s, cg_stats = subsolver.x, subsolver.stats
299+
s, cg_stats = krylov_subsolver.x, krylov_subsolver.stats
300300

301301
# Compute actual vs. predicted reduction.
302302
sNorm = nrm2(n, s)

src/trunkls.jl

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ mutable struct TrunkSolverNLS{
139139
Av::V
140140
Atv::V
141141
A::Op
142-
subsolver::Sub
142+
krylov_subsolver::Sub
143143
params::TRUNKLSParameterSet
144144
end
145145

@@ -170,8 +170,8 @@ function TrunkSolverNLS(
170170
A = jac_op_residual!(nlp, x, Av, Atv)
171171
Op = typeof(A)
172172

173-
workspace = krylov_workspace(Val(subsolver), nequ, nvar, V)
174-
Sub = typeof(workspace)
173+
krylov_subsolver = krylov_workspace(Val(subsolver), nequ, nvar, V)
174+
Sub = typeof(krylov_subsolver)
175175

176176
return TrunkSolverNLS{T, V, Sub, Op}(
177177
x,
@@ -185,7 +185,7 @@ function TrunkSolverNLS(
185185
Av,
186186
Atv,
187187
A,
188-
workspace,
188+
krylov_subsolver,
189189
params,
190190
)
191191
end
@@ -248,7 +248,7 @@ function SolverCore.solve!(
248248
solver.x .= x
249249
x = solver.x
250250
∇f = solver.gx
251-
subsolver = solver.subsolver
251+
krylov_subsolver = solver.krylov_subsolver
252252

253253
n = nlp.nls_meta.nvar
254254
m = nlp.nls_meta.nequ
@@ -321,7 +321,7 @@ function SolverCore.solve!(
321321
cgtol = max(rtol, min(T(0.1), 9 * cgtol / 10, sqrt(∇fNorm2)))
322322
temp .= .-r
323323
krylov_solve!(
324-
subsolver,
324+
krylov_subsolver,
325325
A,
326326
temp,
327327
atol = atol,
@@ -331,7 +331,7 @@ function SolverCore.solve!(
331331
timemax = max_time - stats.elapsed_time,
332332
verbose = subsolver_verbose,
333333
)
334-
s, cg_stats = subsolver.x, subsolver.stats
334+
s, cg_stats = krylov_subsolver.x, krylov_subsolver.stats
335335

336336
# Compute actual vs. predicted reduction.
337337
sNorm = nrm2(n, s)

0 commit comments

Comments
 (0)