Change test to use linked

Author: penelopeysm

test/ad.jl

@@ -23,7 +23,8 @@ using DynamicPPL: LogDensityFunction
             varinfos = DynamicPPL.TestUtils.setup_varinfos(m, rand_param_values, vns)
 
             @testset "$(short_varinfo_name(varinfo))" for varinfo in varinfos
-                # TODO: This runs unlinked. Should we test linked as well?
+                # TODO: This runs linked. Should we test unlinked as well?
+                varinfo = DynamicPPL.link(varinfo, m)
                 f = LogDensityFunction(m, varinfo)
                 x = DynamicPPL.getparams(f)
                 # Calculate reference logp + gradient of logp using ForwardDiff
@@ -57,14 +58,15 @@ using DynamicPPL: LogDensityFunction
                             ref_ldf, adtype
                         )
                     else
-                        # TODO: Should we test linked as well?
-                        DynamicPPL.TestUtils.AD.run_ad(
+                        # TODO: Should we test unlinked as well?
+                        @test DynamicPPL.TestUtils.AD.run_ad(
                             m,
                             adtype;
-                            linked=false,
+                            test=true,
+                            benchmark=true,
                             varinfo=varinfo,
                             expected_value_and_grad=(ref_logp, ref_grad),
-                        )
+                        ) isa Any
                     end
                 end
             end