Add model dimension to benchmark table

Author: mhauru

benchmarks/benchmarks.jl

@@ -2,7 +2,7 @@ using Pkg
 # To ensure we benchmark the local version of DynamicPPL, dev the folder above.
 Pkg.develop(; path=joinpath(@__DIR__, ".."))
 
-using DynamicPPLBenchmarks: Models, make_suite
+using DynamicPPLBenchmarks: Models, make_suite, model_dimension
 using BenchmarkTools: @benchmark, median, run
 using PrettyTables: PrettyTables, ft_printf
 using Random: seed!
@@ -63,7 +63,7 @@ reference_time = begin
     median(@benchmark Models.simple_assume_observe_non_model(obs)).time
 end
 
-results_table = Tuple{String,String,String,Bool,Float64,Float64}[]
+results_table = Tuple{String,Int,String,String,Bool,Float64,Float64}[]
 
 for (model_name, model, varinfo_choice, adbackend, islinked) in chosen_combinations
     suite = make_suite(model, varinfo_choice, adbackend, islinked)
@@ -76,6 +76,7 @@ for (model_name, model, varinfo_choice, adbackend, islinked) in chosen_combinati
         results_table,
         (
             model_name,
+            model_dimension(model, islinked),
             string(adbackend),
             string(varinfo_choice),
             islinked,
@@ -88,6 +89,7 @@ end
 table_matrix = hcat(Iterators.map(collect, zip(results_table...))...)
 header = [
     "Model",
+    "Dimension",
     "AD Backend",
     "VarInfo Type",
     "Linked",

benchmarks/src/DynamicPPLBenchmarks.jl

@@ -13,7 +13,21 @@ using ReverseDiff: ReverseDiff
 include("./Models.jl")
 using .Models: Models
 
-export Models, make_suite
+export Models, make_suite, model_dimension
+
+"""
+    model_dimension(model, islinked)
+
+Return the dimension of `model`, accounting for linking, if any.
+"""
+function model_dimension(model, islinked)
+    vi = VarInfo()
+    model(vi)
+    if islinked
+        vi = DynamicPPL.link(vi, model)
+    end
+    return length(vi[:])
+end
 
 # Utility functions for representing AD backends using symbols.
 # Copied from TuringBenchmarking.jl.