temporary hack to debug what's happening

TuringLang · Nov 30, 2024 · 5c27677 · 5c27677
1 parent 4998d08
commit 5c27677
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/ext/DynamicPPLJETExt.jl b/test/ext/DynamicPPLJETExt.jl
@@ -56,7 +56,17 @@
 
     @testset "demo models" begin
         @testset "$(model.f)" for model in DynamicPPL.TestUtils.DEMO_MODELS
+            using Logging: Logging
+            # Use debug logging below.
             varinfo = DynamicPPL.DynamicPPL.determine_suitable_varinfo(model)
+            if varinfo isa DynamicPPL.UntypedVarInfo
+                # 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
+                )
+                @show JET.report_call(f_eval, argtypes_eval)
+            end
             # They should all result in typed.
             @test varinfo isa DynamicPPL.TypedVarInfo
             # But let's also make sure that they're not lying.