depwarn fixes for max

SciML · May 3, 2017 · 45c4a6c · 45c4a6c
1 parent 0b16531
commit 45c4a6c
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/src/initdt.jl b/src/initdt.jl
@@ -12,14 +12,14 @@ function sde_determine_initdt(u0,t,tdir,dtmax,abstol,reltol,internalnorm,prob,or
     g(t,u0,g₀); g₀.*=3
   end
 
-  d₁ = internalnorm(max(abs.(f₀.+g₀),abs.(f₀-g₀))./(abstol+abs.(u0)*reltol))
+  d₁ = internalnorm(max.(abs.(f₀.+g₀),abs.(f₀.-g₀))./(abstol.+abs.(u0).*reltol))
   if d₀ < 1e-5 || d₁ < 1e-5
     dt₀ = 1e-6
   else
     dt₀ = 0.01*(d₀/d₁)
   end
   dt₀ = min(dt₀,tdir*dtmax)
-  u₁ = u0 + tdir*dt₀*f₀
+  u₁ = u0 .+ tdir.*dt₀.*f₀
   if typeof(u0) <: Number
     f₁ = f(t+tdir*dt₀,u₁)
     g₁ = 3g(t+tdir*dt₀,u₁)
@@ -29,8 +29,8 @@ function sde_determine_initdt(u0,t,tdir,dtmax,abstol,reltol,internalnorm,prob,or
     f(t,u0,f₁)
     g(t,u0,g₁); g₁.*=3
   end
-  ΔgMax = max(abs.(g₀-g₁),abs.(g₀+g₁))
-  d₂ = internalnorm(max(abs.(f₁.-f₀.+ΔgMax),abs.(f₁.-f₀.-ΔgMax))./(abstol+abs.(u0)*reltol))/dt₀
+  ΔgMax = max.(abs.(g₀.-g₁),abs.(g₀.+g₁))
+  d₂ = internalnorm(max.(abs.(f₁.-f₀.+ΔgMax),abs.(f₁.-f₀.-ΔgMax))./(abstol.+abs.(u0).*reltol))./dt₀
   if max(d₁,d₂)<=1e-15
     dt₁ = max(1e-6,dt₀*1e-3)
   else

diff --git a/src/integrators/low_order.jl b/src/integrators/low_order.jl
@@ -101,7 +101,7 @@ end
   end
   u = K+L*W.dW+mil_correction
   if integrator.opts.adaptive
-    integrator.EEst = integrator.opts.internalnorm(mil_correction/(@muladd(integrator.opts.abstol + max(abs(uprev),abs(u))*integrator.opts.reltol)))
+    integrator.EEst = integrator.opts.internalnorm(mil_correction/(@muladd(integrator.opts.abstol + max.(abs.(uprev),abs.(u))*integrator.opts.reltol)))
   end
   @pack integrator = t,dt,u
 end

diff --git a/src/integrators/sra.jl b/src/integrators/sra.jl
@@ -9,7 +9,7 @@
 
   if integrator.opts.adaptive
     u = uprev + k₁/3 + 2k₂/3 + E₂ + W.dW*gpdt
-    integrator.EEst = abs(@muladd(integrator.opts.delta*E₁+E₂)./@muladd(integrator.opts.abstol + max(abs(uprev),abs(u))*integrator.opts.reltol))
+    integrator.EEst = abs(@muladd(integrator.opts.delta*E₁+E₂)./@muladd(integrator.opts.abstol + max.(abs.(uprev),abs.(u))*integrator.opts.reltol))
   else
     u = uprev + k₁/3 + 2k₂/3 + E₂ + W.dW*gpdt
   end
@@ -136,7 +136,7 @@ end
 
   if integrator.opts.adaptive
     E₁ = dt*E₁temp
-    integrator.EEst = integrator.opts.internalnorm(@muladd(integrator.opts.delta*E₁+E₂)./@muladd(integrator.opts.abstol + max(abs(uprev),abs(u))*integrator.opts.reltol))
+    integrator.EEst = integrator.opts.internalnorm(@muladd(integrator.opts.delta*E₁+E₂)./@muladd(integrator.opts.abstol + max.(abs.(uprev),abs.(u))*integrator.opts.reltol))
   end
   @pack integrator = t,dt,u
 end
diff --git a/src/integrators/sri.jl b/src/integrators/sri.jl
@@ -152,7 +152,7 @@ end
 
   u = uprev + (fH01 + 2fH02)/3 + W.dW.*(mg₁ + Fg₂o3 + Tg₃o3) + chi1.*(mg₁ + Fg₂o3 - g₃o3) + E₂
   if integrator.opts.adaptive
-    integrator.EEst = integrator.opts.internalnorm(@muladd(integrator.opts.delta*E₁+E₂)/(@muladd(integrator.opts.abstol + max(abs(uprev),abs(u))*integrator.opts.reltol)))
+    integrator.EEst = integrator.opts.internalnorm(@muladd(integrator.opts.delta*E₁+E₂)/(@muladd(integrator.opts.abstol + max.(abs.(uprev),abs.(u))*integrator.opts.reltol)))
   end
   @pack integrator = t,dt,u
 end
@@ -199,7 +199,7 @@ end
   u = @muladd(uprev + dt*atemp) + btemp + E₂
 
   if integrator.opts.adaptive
-    integrator.EEst = integrator.opts.internalnorm(@muladd(integrator.opts.delta*E₁+E₂)./@muladd(integrator.opts.abstol + max(abs(uprev),abs(u))*integrator.opts.reltol))
+    integrator.EEst = integrator.opts.internalnorm(@muladd(integrator.opts.delta*E₁+E₂)./@muladd(integrator.opts.abstol + max.(abs.(uprev),abs.(u))*integrator.opts.reltol))
   end
   @pack integrator = t,dt,u
 end