File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change 31
31
x = map (x-> Float64 (x), vi[SampleFromPrior ()])
32
32
33
33
trackerℓ = LogDensityProblemsAD. ADgradient (TrackerAD (), ℓ)
34
- @test trackerℓ isa LogDensityProblemsAD. LogDensityProblemsADTrackerExt. TrackerGradientLogDensity
34
+ if isdefined (Base, :get_extension )
35
+ @test trackerℓ isa Base. get_extension (LogDensityProblemsAD, :LogDensityProblemsADTrackerExt ). TrackerGradientLogDensity
36
+ else
37
+ @test trackerℓ isa LogDensityProblemsAD. LogDensityProblemsADTrackerExt. TrackerGradientLogDensity
38
+ end
35
39
@test trackerℓ. ℓ === ℓ
36
40
∇E1 = LogDensityProblems. logdensity_and_gradient (trackerℓ, x)[2 ]
37
41
@test sort (∇E1) ≈ grad_FWAD atol= 1e-9
38
42
39
43
zygoteℓ = LogDensityProblemsAD. ADgradient (ZygoteAD (), ℓ)
40
- @test zygoteℓ isa LogDensityProblemsAD. LogDensityProblemsADZygoteExt. ZygoteGradientLogDensity
44
+ if isdefined (Base, :get_extension )
45
+ @test zygoteℓ isa Base. get_extension (LogDensityProblemsAD, :LogDensityProblemsADZygoteExt ). ZygoteGradientLogDensity
46
+ else
47
+ @test zygoteℓ isa LogDensityProblemsAD. LogDensityProblemsADZygoteExt. ZygoteGradientLogDensity
48
+ end
41
49
@test zygoteℓ. ℓ === ℓ
42
50
∇E2 = LogDensityProblems. logdensity_and_gradient (zygoteℓ, x)[2 ]
43
51
@test sort (∇E2) ≈ grad_FWAD atol= 1e-9
You can’t perform that action at this time.
0 commit comments