Add back support for ADTypes < 1.5 (#35)

tpapp · Aug 14, 2024 · 013b68f · 013b68f · devmotion · Aug 14, 2024
1 parent b3c5801
commit 013b68f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "LogDensityProblemsAD"
 uuid = "996a588d-648d-4e1f-a8f0-a84b347e47b1"
 authors = ["Tamás K. Papp <[email protected]>"]
-version = "1.9.1"
+version = "1.9.2"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"

diff --git a/ext/LogDensityProblemsADADTypesExt.jl b/ext/LogDensityProblemsADADTypesExt.jl
@@ -37,8 +37,18 @@ function LogDensityProblemsAD.ADgradient(ad::ADTypes.AutoForwardDiff{C}, ℓ) wh
     end
 end
 
-function LogDensityProblemsAD.ADgradient(ad::ADTypes.AutoReverseDiff{T}, ℓ) where {T}
-    return LogDensityProblemsAD.ADgradient(Val(:ReverseDiff), ℓ; compile = Val(T))
+# ADTypes 1.5 introduced a type parameter for `AutoReverseDiff` and deprecated the
+# `compile` field
+# Since Julia < 1.9 uses Requires which does not respect the ADTypes compat entry,
+# we keep the version for ADTypes < 1.5 as well
+@static if ADTypes.AutoReverseDiff isa UnionAll
+    function LogDensityProblemsAD.ADgradient(ad::ADTypes.AutoReverseDiff{T}, ℓ) where {T}
+        return LogDensityProblemsAD.ADgradient(Val(:ReverseDiff), ℓ; compile = Val(T))
+    end
+else
+    function LogDensityProblemsAD.ADgradient(ad::ADTypes.AutoReverseDiff, ℓ)
+        return LogDensityProblemsAD.ADgradient(Val(:ReverseDiff), ℓ; compile = Val(ad.compile))
+    end
 end
 
 function LogDensityProblemsAD.ADgradient(::ADTypes.AutoTracker, ℓ)