Skip to content

Commit db180fd

Browse files
committed
fix stats of stabilized RK methods
1 parent 185d91f commit db180fd

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

lib/OrdinaryDiffEqStabilizedRK/src/rkc_perform_step.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ end
3535
μ, κ = recf[cache.start + (i - 2) * 2 + 1], recf[cache.start + (i - 2) * 2 + 2]
3636
ν = -1 - κ
3737
u = f(uᵢ₋₁, p, tᵢ₋₁)
38+
OrdinaryDiffEqCore.increment_nf!(integrator.stats, 1)
3839
tᵢ₋₁ = dt * μ - ν * tᵢ₋₂ - κ * tᵢ₋₃
3940
u = (dt * μ) * u - ν * uᵢ₋₁ - κ * uᵢ₋₂
4041
i < cache.mdeg && (uᵢ₋₂ = uᵢ₋₁;
@@ -110,6 +111,7 @@ end
110111
μ, κ = recf[ccache.start + (i - 2) * 2 + 1], recf[ccache.start + (i - 2) * 2 + 2]
111112
ν = -1 - κ
112113
f(k, uᵢ₋₁, p, tᵢ₋₁)
114+
OrdinaryDiffEqCore.increment_nf!(integrator.stats, 1)
113115
tᵢ₋₁ = dt * μ - ν * tᵢ₋₂ - κ * tᵢ₋₃
114116
@.. broadcast=false u=(dt * μ) * k - ν * uᵢ₋₁ - κ * uᵢ₋₂
115117
if i < ccache.mdeg
@@ -192,6 +194,7 @@ end
192194
μ, κ = recf[cache.start + (i - 2) * 2 + 1], recf[cache.start + (i - 2) * 2 + 2]
193195
ν = -1 - κ
194196
u = f(uᵢ₋₁, p, tᵢ₋₁)
197+
OrdinaryDiffEqCore.increment_nf!(integrator.stats, 1)
195198
tᵢ₋₁ = dt * μ - ν * tᵢ₋₂ - κ * tᵢ₋₃
196199
u = (dt * μ) * u - ν * uᵢ₋₁ - κ * uᵢ₋₂
197200
i < cache.mdeg && (uᵢ₋₂ = uᵢ₋₁;
@@ -314,6 +317,7 @@ end
314317
μ, κ = recf[ccache.start + (i - 2) * 2 + 1], recf[ccache.start + (i - 2) * 2 + 2]
315318
ν = -1 - κ
316319
f(k, uᵢ₋₁, p, tᵢ₋₁)
320+
OrdinaryDiffEqCore.increment_nf!(integrator.stats, 1)
317321
tᵢ₋₁ = (dt * μ) - ν * tᵢ₋₂ - κ * tᵢ₋₃
318322
@.. broadcast=false u=(dt * μ) * k - ν * uᵢ₋₁ - κ * uᵢ₋₂
319323
if i < ccache.mdeg

0 commit comments

Comments
 (0)