From d575089400c4879e4bc9c5f58a15c6c40e06da40 Mon Sep 17 00:00:00 2001 From: FredericWantiez Date: Fri, 8 Mar 2024 23:07:59 +0000 Subject: [PATCH] Forward RNG argument --- perf/p0.jl | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/perf/p0.jl b/perf/p0.jl index 81ce7fd2..757d8b3a 100644 --- a/perf/p0.jl +++ b/perf/p0.jl @@ -1,3 +1,4 @@ +using Random using Libtask using Turing, DynamicPPL, AdvancedPS using BenchmarkTools @@ -13,7 +14,8 @@ end # Case 1: Sample from the prior. -m = Turing.Core.TracedModel(gdemo(1.5, 2.), SampleFromPrior(), VarInfo()) +rng = MersenneTwister() +m = Turing.Core.TracedModel(gdemo(1.5, 2.), SampleFromPrior(), VarInfo(), rng) f = m.evaluator[1]; args = m.evaluator[2:end]; @@ -26,7 +28,7 @@ println("Run a tape...") @btime t.tf(args...) # Case 2: SMC sampler -m = Turing.Core.TracedModel(gdemo(1.5, 2.), Sampler(SMC(50)), VarInfo()); +m = Turing.Core.TracedModel(gdemo(1.5, 2.), Sampler(SMC(50)), VarInfo(), rng) f = m.evaluator[1]; args = m.evaluator[2:end];