Merge fcd3227 into ee43ced

SciML · Jul 13, 2018 · 15559b5 · 15559b5
2 parents ee43ced + fcd3227
commit 15559b5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/solve.jl b/src/solve.jl
@@ -5,8 +5,9 @@ function init(prob::AbstractDDEProblem{uType,tType,lType,isinplace}, alg::algTyp
               prob.tspan[2]-prob.tspan[1] : tType(7*minimum(prob.constant_lags)),
               dt=zero(tType), saveat=tType[], tstops = tType[],
               save_idxs=nothing, save_everystep=isempty(saveat),
-              save_start=true, save_end = true,
-              dense=save_everystep && !(typeof(alg) <: FunctionMap),
+              save_start = save_everystep || isempty(saveat) || typeof(saveat) <: Number ? true : prob.tspan[1] in saveat,
+              save_end = save_everystep || isempty(saveat) || typeof(saveat) <: Number ? true : prob.tspan[2] in saveat,
+              dense = save_everystep && !(typeof(alg) <: FunctionMap) && isempty(saveat),
               minimal_solution=true, discontinuity_interp_points::Int=10,
               discontinuity_abstol=tType(1//Int64(10)^12), discontinuity_reltol=0,
               initial_order=agrees(prob.h, prob.u0, prob.p, prob.tspan[1]) ? 1 : 0,