Move the content of ad.jl from Turing.jl to here

ext/DynamicPPLReverseDiffExt.jl

@@ -12,7 +12,7 @@ function LogDensityProblemsAD.ADgradient(
     ad::ADTypes.AutoReverseDiff, ℓ::DynamicPPL.LogDensityFunction
 )
     return LogDensityProblemsAD.ADgradient(
-        Val(:ReverseDiff), ℓ; compile=Val(ad.compile), x=DynamicPPL.getparams(ℓ)
+        Val(:ReverseDiff), ℓ; compile=Val(ad.compile), x=identity.(DynamicPPL.getparams(ℓ))
     )
 end
 

test/ad.jl

@@ -4,14 +4,14 @@
     vns = DynamicPPL.TestUtils.varnames(m)
     varinfos = DynamicPPL.TestUtils.setup_varinfos(m, rand_param_values, vns)
 
-    @testset "$varinfo" for varinfo in varinfos
+    @testset "$(short_varinfo_name(varinfo))" for varinfo in varinfos
         f = DynamicPPL.LogDensityFunction(m, varinfo)
 
         # use ForwardDiff result as reference
         ad_forwarddiff_f = LogDensityProblemsAD.ADgradient(
             ADTypes.AutoForwardDiff(; chunksize=0), f
         )
-        θ = varinfo[:]
+        θ = identity.(varinfo[:])
         logp, ref_grad = LogDensityProblems.logdensity_and_gradient(ad_forwarddiff_f, θ)
 
         @testset "with ADType $adtype" for adtype in (
@@ -22,17 +22,13 @@
             @test grad ≈ ref_grad
         end
 
-        if m.f ∉ (
-            DynamicPPL.TestUtils.demo_dot_assume_dot_observe,
-            DynamicPPL.TestUtils.demo_assume_index_observe,
-            DynamicPPL.TestUtils.demo_dot_assume_observe_index,
-            DynamicPPL.TestUtils.demo_dot_assume_observe_index_literal,
-            DynamicPPL.TestUtils.demo_assume_submodel_observe_index_literal,
-            DynamicPPL.TestUtils.demo_dot_assume_observe_submodel,
-            DynamicPPL.TestUtils.demo_dot_assume_dot_observe_matrix,
-            DynamicPPL.TestUtils.demo_dot_assume_matrix_dot_observe_matrix,
-            DynamicPPL.TestUtils.demo_assume_matrix_dot_observe_matrix,
-        )
+        if m.f ∈ (
+            DynamicPPL.TestUtils.demo_assume_multivariate_observe,
+            DynamicPPL.TestUtils.demo_assume_dot_observe,
+            DynamicPPL.TestUtils.demo_assume_observe_literal,
+            DynamicPPL.TestUtils.demo_assume_literal_dot_observe,
+        ) &&
+            varinfo isa Union{DynamicPPL.TypedVarInfo,DynamicPPL.SimpleVarInfo{<:NamedTuple}}
             adtype = ADTypes.AutoZygote()
             ad_f = LogDensityProblemsAD.ADgradient(adtype, f)
             _, grad = LogDensityProblems.logdensity_and_gradient(ad_f, θ)