Skip to content

Commit f648177

Browse files
author
KDr2
committed
fix perf code
1 parent 0e9a4bb commit f648177

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

perf/p0.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ args = m.evaluator[2:end];
3030
t = @btime Libtask.CTask(f, args...)
3131
# schedule(t.task) # work fine!
3232
# @show Libtask.result(t.tf)
33-
@show "Step in a tape..."
34-
@btime Libtask.step_in(t.tf, args)
33+
@show "Run a tape..."
34+
@btime t.tf(args...)
3535

3636
# Case 2: SMC sampler
3737

@@ -43,5 +43,5 @@ m = Turing.Core.TracedModel(gdemo(1.5, 2.), Sampler(SMC(50)), VarInfo());
4343
t = @btime Libtask.CTask(m.evaluator[1], m.evaluator[2:end]...);
4444
# schedule(t.task)
4545
# @show Libtask.result(t.tf.tape)
46-
@show "Step in a tape..."
47-
@btime Libtask.step_in(t.tf, m.evaluator[2:end])
46+
@show "Run a tape..."
47+
@btime t.tf(m.evaluator[2:end]...)

0 commit comments

Comments
 (0)