Fix random init

Author: wsmoses

src/Overlay.jl

@@ -103,7 +103,7 @@ for randfun in (:rand, :randn, :randexp)
             return TracedRandom.$(overload_randfun!)(rng, A)
         end
         @reactant_overlay @noinline function Random.$(randfun!)(A::AnyTracedRArray)
-            return TracedRandom.$(overload_randfun!)(TracedRandom.default_rng(), A)
+            return TracedRandom.$(overload_randfun!)(call_with_reactant(TracedRandom.default_rng), A)
         end
     end
 end

src/stdlibs/Random.jl

@@ -39,7 +39,9 @@ Base.copy(rng::ReactantRNG) = ReactantRNG(copy(rng.seed), rng.algorithm)
 end
 @noinline ReactantRNG(seed::AbstractVector) = ReactantRNG(seed, "DEFAULT")
 
-@noinline default_rng() = ReactantRNG()
+@noinline function default_rng()
+    ReactantRNG()
+end
 
 @noinline rng_algorithm(rng::ReactantRNG) = rng.algorithm
 @noinline rng_algorithm(::AbstractRNG) = "DEFAULT"