Skip to content

Commit aea598e

Browse files
devmotiondevmotion
authored
Load only LogDensityProblemsAD and weak dependency in extension (#7)
* Load only LogDensityProblemsAD and weak dependency in extension * Bump version
1 parent 6b36a58 commit aea598e

File tree

8 files changed

+19
-31
lines changed

8 files changed

+19
-31
lines changed

‎Project.toml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name = "LogDensityProblemsAD"
22
uuid = "996a588d-648d-4e1f-a8f0-a84b347e47b1"
33
authors = ["Tamás K. Papp <[email protected]>"]
4-
version = "1.2.0"
4+
version = "1.2.1"
55

66
[deps]
77
DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"

‎ext/EnzymeExt.jl‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,10 @@ Gradient AD implementation using Enzyme.
33
"""
44
module EnzymeExt
55

6-
using LogDensityProblems: logdensity
7-
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED
8-
using UnPack: @unpack
6+
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED, logdensity
7+
using LogDensityProblemsAD.UnPack: @unpack
98

10-
import LogDensityProblems: logdensity_and_gradient
11-
import LogDensityProblemsAD: ADgradient
9+
import LogDensityProblemsAD: ADgradient, logdensity_and_gradient
1210
if EXTENSIONS_SUPPORTED
1311
import Enzyme
1412
else

‎ext/ForwardDiffBenchmarkToolsExt.jl‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,7 @@ Loaded when both ForwardDiff and BenchmarkTools are loaded.
55
"""
66
module ForwardDiffBenchmarkToolsExt
77

8-
using DocStringExtensions: SIGNATURES
9-
using LogDensityProblems: dimension, logdensity_and_gradient
10-
using LogDensityProblemsAD: ADgradient, EXTENSIONS_SUPPORTED
8+
using LogDensityProblemsAD: ADgradient, EXTENSIONS_SUPPORTED, SIGNATURES, dimension, logdensity_and_gradient
119

1210
if EXTENSIONS_SUPPORTED
1311
using BenchmarkTools: @belapsed

‎ext/ForwardDiffExt.jl‎

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,10 @@ Gradient AD implementation using ForwardDiff.
33
"""
44
module ForwardDiffExt
55

6-
using DocStringExtensions: SIGNATURES
7-
using LogDensityProblems: dimension, logdensity
8-
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED
9-
using UnPack: @unpack
6+
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED, SIGNATURES, dimension, logdensity
7+
using LogDensityProblemsAD.UnPack: @unpack
108

11-
import LogDensityProblems: logdensity_and_gradient
12-
import LogDensityProblemsAD: ADgradient
9+
import LogDensityProblemsAD: ADgradient, logdensity_and_gradient
1310
if EXTENSIONS_SUPPORTED
1411
import ForwardDiff
1512
import ForwardDiff: DiffResults

‎ext/ReverseDiffExt.jl‎

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,10 @@ Gradient AD implementation using ReverseDiff.
33
"""
44
module ReverseDiffExt
55

6-
using DocStringExtensions: SIGNATURES
7-
using LogDensityProblems: dimension, logdensity
8-
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED
9-
using UnPack: @unpack
6+
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED, SIGNATURES, dimension, logdensity
7+
using LogDensityProblemsAD.UnPack: @unpack
108

11-
import LogDensityProblems: logdensity_and_gradient
12-
import LogDensityProblemsAD: ADgradient
9+
import LogDensityProblemsAD: ADgradient, logdensity_and_gradient
1310
if EXTENSIONS_SUPPORTED
1411
import ReverseDiff
1512
import ReverseDiff: DiffResults

‎ext/TrackerExt.jl‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,10 @@ Gradient AD implementation using Tracker.
33
"""
44
module TrackerExt
55

6-
using LogDensityProblems: logdensity
7-
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED
8-
using UnPack: @unpack
6+
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED, logdensity
7+
using LogDensityProblemsAD.UnPack: @unpack
98

10-
import LogDensityProblems: logdensity_and_gradient
11-
import LogDensityProblemsAD: ADgradient
9+
import LogDensityProblemsAD: ADgradient, logdensity_and_gradient
1210
if EXTENSIONS_SUPPORTED
1311
import Tracker
1412
else

‎ext/ZygoteExt.jl‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,10 @@ Gradient AD implementation using Zygote.
33
"""
44
module ZygoteExt
55

6-
using LogDensityProblems: logdensity
7-
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED
8-
using UnPack: @unpack
6+
using LogDensityProblemsAD: ADGradientWrapper, EXTENSIONS_SUPPORTED, logdensity
7+
using LogDensityProblemsAD.UnPack: @unpack
98

10-
import LogDensityProblems: logdensity_and_gradient
11-
import LogDensityProblemsAD: ADgradient
9+
import LogDensityProblemsAD: ADgradient, logdensity_and_gradient
1210
if EXTENSIONS_SUPPORTED
1311
import Zygote
1412
else

‎src/LogDensityProblemsAD.jl‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ using DocStringExtensions: SIGNATURES
99
import LogDensityProblems: logdensity, logdensity_and_gradient, capabilities, dimension
1010
using LogDensityProblems: LogDensityOrder
1111

12+
import UnPack
13+
1214

1315
#####
1416
##### AD wrappers --- interface and generic code

0 commit comments

Comments
 (0)