Skip to content

Commit 9c12ac6

Browse files
check for NaN in first function call
1 parent babd22c commit 9c12ac6

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/initdt.jl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,24 @@ function sde_determine_initdt{tType,uType}(u0::uType,t::tType,tdir,dtmax,abstol,
44
d₀ = internalnorm(u0./(abstol.+abs.(u0).*reltol))
55
if !isinplace(prob)
66
f₀ = f(t,u0)
7+
if isnan(f₀)
8+
error("First function call for f produced NaNs. Exiting.")
9+
end
710
g₀ = 3g(t,u0)
11+
if isnan(f₀)
12+
error("First function call for g produced NaNs. Exiting.")
13+
end
814
else
915
f₀ = zeros(u0)
1016
g₀ = zeros(u0)
1117
f(t,u0,f₀)
18+
if any((isnan(x) for x in f₀))
19+
error("First function call for f produced NaNs. Exiting.")
20+
end
1221
g(t,u0,g₀); g₀.*=3
22+
if any((isnan(x) for x in g₀))
23+
error("First function call for g produced NaNs. Exiting.")
24+
end
1325
end
1426

1527
d₁ = internalnorm(max.(abs.(f₀.+g₀),abs.(f₀.-g₀))./(abstol.+abs.(u0).*reltol))

0 commit comments

Comments
 (0)