Fix for default AutoForwardDiff (#581)

* fix for default `AutoForwardDiff`

* bump patch version

* fixed bug in `dot_tilde_assume`

* fixed another `dot_tilde_assume` argument ordering

Author: torfjelde

Project.toml

@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.24.7"
+version = "0.24.8"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

ext/DynamicPPLForwardDiffExt.jl

@@ -26,7 +26,7 @@ function LogDensityProblemsAD.ADgradient(
         ForwardDiff.Tag(f, eltype(θ))
     end
     chunk_size = getchunksize(ad)
-    chunk = if chunk_size == 0
+    chunk = if chunk_size == 0 || chunk_size === nothing
         ForwardDiff.Chunk(θ)
     else
         ForwardDiff.Chunk(length(θ), chunk_size)

src/context_implementations.jl

@@ -289,7 +289,7 @@ function dot_tilde_assume(context::AbstractContext, args...)
     return dot_tilde_assume(NodeTrait(dot_tilde_assume, context), context, args...)
 end
 function dot_tilde_assume(rng, context::AbstractContext, args...)
-    return dot_tilde_assume(rng, NodeTrait(dot_tilde_assume, context), context, args...)
+    return dot_tilde_assume(NodeTrait(dot_tilde_assume, context), rng, context, args...)
 end
 
 function dot_tilde_assume(::IsLeaf, ::AbstractContext, right, left, vns, vi)
@@ -302,7 +302,7 @@ end
 function dot_tilde_assume(::IsParent, context::AbstractContext, args...)
     return dot_tilde_assume(childcontext(context), args...)
 end
-function dot_tilde_assume(rng, ::IsParent, context::AbstractContext, args...)
+function dot_tilde_assume(::IsParent, rng, context::AbstractContext, args...)
     return dot_tilde_assume(rng, childcontext(context), args...)
 end
 

test/ext/DynamicPPLForwardDiffExt.jl

@@ -1,5 +1,5 @@
 @testset "tag" begin
-    for chunksize in (0, 1, 10)
+    for chunksize in (nothing, 0, 1, 10)
         ad = ADTypes.AutoForwardDiff(; chunksize=chunksize)
         standardtag = if !isdefined(Base, :get_extension)
             DynamicPPL.DynamicPPLForwardDiffExt.standardtag