mid bench

SciML · Jan 27, 2017 · f3b0979 · f3b0979
1 parent 1687f8d
commit f3b0979
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/integrators/integrator_utils.jl b/src/integrators/integrator_utils.jl
@@ -19,7 +19,7 @@ end
   end
 end
 
-function savevalues!(integrator::SDEIntegrator)
+@inline function savevalues!(integrator::SDEIntegrator)
   if integrator.opts.save_timeseries && integrator.iter%integrator.opts.timeseries_steps==0
     push!(integrator.sol.u,copy(integrator.u))
     push!(integrator.sol.t,integrator.t)
@@ -135,6 +135,7 @@ end
         integrator.ΔW = ΔWtmp
         integrator.ΔZ = ΔZtmp
         integrator.dt = dtnew
+        integrator.sqdt = sqrt(integrator.dt)
       else # RSwM3
         if !(uType <: AbstractArray)
           dttmp = 0.0; ΔWtmp = 0.0; ΔZtmp = 0.0
@@ -172,6 +173,7 @@ end
         integrator.dt = dtnew
         integrator.ΔW = ΔWtilde
         integrator.ΔZ = ΔZtilde
+        integrator.sqdt = sqrt(integrator.dt)
       end
     end
   else # Non adaptive
@@ -209,7 +211,7 @@ end
   return nothing
 end
 
-function update_running_noise!(integrator)
+@inline function update_running_noise!(integrator)
   if integrator.opts.save_noise
     if typeof(integrator.u) <: AbstractArray
       for i in eachindex(integrator.u)