fix SROCK2 for NoiseWrapper noise

fixes SciML#505
axsk · Sep 30, 2022 · 9a556d7 · 9a556d7
1 parent 5c13cc7
commit 9a556d7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/perform_step/SROCK_perform_step.jl b/src/perform_step/SROCK_perform_step.jl
@@ -332,8 +332,12 @@ end
 
   sqrt_dt   = sqrt(abs(dt))
   if gen_prob
-    vec_χ .= 1//2 .+ oftype(W.dW, rand(W.rng, length(W.dW)))
-    @.. vec_χ = 2*floor(vec_χ) - 1
+    if hasfield(W, :rng)
+        rand!(W.rng, vec_χ)
+    else
+        rand!(vec_χ)
+    end
+    @.. vec_χ = 2*floor(vec_χ + 1//2) - 1
   end
 
   μ = recf[start]  # here κ = 0