more debugging

torfjelde · torfjelde · commit 99d4df7cb80d · 2024-11-30T08:59:45.000+01:00
diff --git a/test/ext/DynamicPPLJETExt.jl b/test/ext/DynamicPPLJETExt.jl
@@ -63,8 +63,9 @@
                 # HACK: Let's debug this.
                 varinfo_typed = DynamicPPL.typed_varinfo(model)
                 f_eval, argtypes_eval = DynamicPPL.DebugUtils.gen_evaluator_call_with_types(
-                    model, varinfo_typed
+                    model, varinfo_typed, SamplingContext()
                 )
+                @show JET.report_call(f_eval, argtypes_eval; target_modules=(JET.AnyFrameModule(DynamicPPL),))
                 @show JET.report_call(f_eval, argtypes_eval)
             end
             # They should all result in typed.
