fix VectorizedRNG to only the no seed case

SciML · May 18, 2020 · e0771be · e0771be
1 parent 03f8294
commit e0771be
Showing 1 changed file with 7 additions and 13 deletions.
diff --git a/src/solve.jl b/src/solve.jl
@@ -289,19 +289,13 @@ function DiffEqBase.__init(
   if typeof(prob) <: DiffEqBase.AbstractRODEProblem && prob.noise === nothing
     rswm = isadaptive(alg) ? RSWM(adaptivealg=:RSwM3) : RSWM(adaptivealg=:RSwM1)
     if isinplace(prob)
-        _rng = typeof(prob.u0) <: Array{Float64} ? VectorizedRNG.local_rng(_seed) :
-                                        Xorshifts.Xoroshiro128Plus(_seed)
-      #if isadaptive(alg) || callback !== nothing
-        if alg_needs_extra_process(alg)
-          W = WienerProcess!(t,rand_prototype,rand_prototype,
-                             save_everystep=save_noise,
-                             rng = _rng)
-        else
-          W = WienerProcess!(t,rand_prototype,
-                             save_everystep=save_noise,
-                             rng = _rng)
-        end
-      #=
+      _rng = typeof(prob.u0) <: Array{Float64} && seed !== nothing ?
+                                            VectorizedRNG.local_rng() :
+                                            Xorshifts.Xoroshiro128Plus(_seed)
+      if alg_needs_extra_process(alg)
+        W = constructor(t,rand_prototype,rand_prototype,
+                           save_everystep=save_noise,
+                           rng = _rng)
       else
         if alg_needs_extra_process(alg)
           W = SimpleWienerProcess!(t,rand_prototype,rand_prototype,