Skip to content

Commit 1bbd40a

Browse files
add iters count to linear solvers
1 parent dacd618 commit 1bbd40a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/iterative_wrappers.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,13 +208,15 @@ function SciMLBase.solve(cache::LinearCache, alg::IterativeSolversJL; kwargs...)
208208
purge_history!(cache.cacheval, cache.u, cache.b)
209209

210210
cache.verbose && println("Using IterativeSolvers.$(alg.generate_iterator)")
211+
i = 0
211212
for iter in enumerate(cache.cacheval)
213+
i += 1
212214
cache.verbose && println("Iter: $(iter[1]), residual: $(iter[2])")
213215
# TODO inject callbacks KSP into solve cb!(cache.cacheval)
214216
end
215217
cache.verbose && println()
216218

217-
return SciMLBase.build_linear_solution(alg,cache.u,nothing,cache)
219+
return SciMLBase.build_linear_solution(alg,cache.u,nothing,cache; iters = i)
218220
end
219221

220222
purge_history!(iter, x, b) = nothing

0 commit comments

Comments
 (0)