TracedModel - CTask (#1770)

* Add taped-ctask * Add prefix for libtask APIs. Co-authored-by: Hong Ge <[email protected]>
TuringLang · Feb 3, 2022 · 715dd98 · 715dd98 · yebai · Feb 3, 2022
1 parent fca2fb0
commit 715dd98
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/essential/container.jl b/src/essential/container.jl
@@ -70,3 +70,11 @@ function AdvancedPS.reset_logprob!(f::TracedModel)
     DynamicPPL.resetlogp!!(f.varinfo)
     return
 end
+
+function Libtask.CTask(model::TracedModel)
+    return Libtask.CTask(model.evaluator[1], model.evaluator[2:end]...)
+end
+
+function Libtask.CTask(model::TracedModel, ::Random.AbstractRNG)
+    return Libtask.CTask(model)
+end
diff --git a/src/inference/AdvancedSMC.jl b/src/inference/AdvancedSMC.jl
@@ -363,7 +363,7 @@ function DynamicPPL.assume(
 end
 
 function DynamicPPL.observe(spl::Sampler{<:Union{PG,SMC}}, dist::Distribution, value, vi)
-    produce(logpdf(dist, value))
+    Libtask.produce(logpdf(dist, value))
     return 0, vi
 end