Skip to content

Commit 2db58b0

Browse files
committed
make sure krylov verbosity works
1 parent 0396d23 commit 2db58b0

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

ext/LinearSolveKrylovKitExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ function SciMLBase.solve!(cache::LinearCache, alg::KrylovKitJL; kwargs...)
4242
iters = iters)
4343
end
4444

45-
function verbosity_to_KrylovKit(verb::SciML.Verbosity.Type)
45+
function verbosity_to_KrylovKit(verb::SciMLBase.Verbosity.Type)
4646
SciML.@match verb begin
4747
Verbosity.None() => 0
4848
Verbosity.Warn() => 1

src/iterative_wrappers.jl

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -284,8 +284,14 @@ function SciMLBase.solve!(cache::LinearCache, alg::KrylovJL; kwargs...)
284284
cache.cacheval
285285
end
286286

287+
krylovJL_verbose = SciMLBase.@match verbose.numerical.KrylovJL_verbosity begin
288+
SciMLBase.Verbosity.None() => 0
289+
::SciMLBase.Verbosity.Type => 1
290+
_ => error("Invalid verbosity.")
291+
end
292+
287293
args = (cacheval, cache.A, cache.b)
288-
kwargs = (atol = atol, rtol, itmax, verbose,
294+
kwargs = (atol = atol, rtol, itmax, verbose = krylovJL_verbose,
289295
ldiv = true, history = true, alg.kwargs...)
290296

291297
if cache.cacheval isa Krylov.CgWorkspace
@@ -333,3 +339,5 @@ function SciMLBase.solve!(cache::LinearCache, alg::KrylovJL; kwargs...)
333339
return SciMLBase.build_linear_solution(alg, cache.u, Ref(resid), cache;
334340
iters = stats.niter, retcode, stats)
335341
end
342+
343+

0 commit comments

Comments
 (0)