Add JACOBIAN_REUSE_THRESHOLD, and default it to 3

SciML · Mar 8, 2019 · 580ea42 · 580ea42
1 parent ea13bf3
commit 580ea42
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/derivative_utils.jl b/src/derivative_utils.jl
@@ -1,3 +1,5 @@
+const JACOBIAN_REUSE_THRESHOLD = Ref(3)
+
 function calc_tderivative!(integrator, cache, dtd1, repeat_step)
   @inbounds begin
     @unpack t,dt,uprev,u,f,p = integrator
@@ -265,7 +267,7 @@ function calc_W!(integrator, cache::OrdinaryDiffEqMutableCache, dtgamma, repeat_
     elseif DiffEqBase.has_jac(f) && f.jac_prototype !== nothing
       # skip calculation of J if step is repeated
       if repeat_step || (alg_can_repeat_jac(alg) &&
-                         (!integrator.last_stepfail && nl_iters == 1 &&
+                         (!integrator.last_stepfail && nl_iters <= JACOBIAN_REUSE_THRESHOLD[] &&
                           ηold < alg.new_jac_conv_bound))
         new_jac = false
       else
@@ -285,7 +287,7 @@ function calc_W!(integrator, cache::OrdinaryDiffEqMutableCache, dtgamma, repeat_
     else # concrete W using jacobian from `calc_J!`
       # skip calculation of J if step is repeated
       if repeat_step || (alg_can_repeat_jac(alg) &&
-                         (!integrator.last_stepfail && nl_iters == 1 &&
+                         (!integrator.last_stepfail && nl_iters < JACOBIAN_REUSE_THRESHOLD[] &&
                           ηold < alg.new_jac_conv_bound))
         new_jac = false
       else