Tapir -> Mooncake

Author: penelopeysm

src/Turing.jl

@@ -107,7 +107,7 @@ export @model,                 # modelling
     AutoReverseDiff,
     AutoZygote,
     AutoTracker,
-    AutoTapir,
+    AutoMooncake,
     setprogress!,           # debugging
     Flat,
     FlatPos,

src/essential/Essential.jl

@@ -11,7 +11,7 @@ using Bijectors: PDMatDistribution
 using AdvancedVI
 using StatsFuns: logsumexp, softmax
 @reexport using DynamicPPL
-using ADTypes: ADTypes, AutoForwardDiff, AutoTracker, AutoReverseDiff, AutoZygote, AutoTapir
+using ADTypes: ADTypes, AutoForwardDiff, AutoTracker, AutoReverseDiff, AutoZygote, AutoMooncake
 
 using AdvancedPS: AdvancedPS
 
@@ -23,7 +23,7 @@ export @model,
     AutoTracker,
     AutoZygote,
     AutoReverseDiff,
-    AutoTapir,
+    AutoMooncake,
     @logprob_str,
     @prob_str
 

test/Project.toml

@@ -16,6 +16,7 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LogDensityProblems = "6fdf6af0-433a-55f7-b3ed-c6c6e0b8df7c"
 LogDensityProblemsAD = "996a588d-648d-4e1f-a8f0-a84b347e47b1"
 MCMCChains = "c7f686f2-ff18-58e9-bc7b-31028e88f75d"
+Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 NamedArrays = "86f7a689-2022-50b4-a561-43c23ac3c673"
 Optim = "429524aa-4258-5aef-a3af-852621145aeb"
 Optimization = "7f7a1694-90dd-40f0-9382-eb1efda571ba"
@@ -30,7 +31,6 @@ SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 StatsFuns = "4c63d2b9-4356-54db-8cca-17b64c39e42c"
-Tapir = "07d77754-e150-4737-8c94-cd238a1fb45b"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
@@ -68,7 +68,6 @@ SpecialFunctions = "0.10.3, 1, 2"
 StableRNGs = "1"
 StatsBase = "0.33, 0.34"
 StatsFuns = "0.9.5, 1"
-Tapir = "0.2.45"
 TimerOutputs = "0.5"
 Tracker = "0.2.11"
 Zygote = "0.5.4, 0.6"

test/mcmc/Inference.jl

@@ -12,7 +12,7 @@ using LinearAlgebra: I
 import MCMCChains
 import Random
 import ReverseDiff
-import Tapir
+import Mooncake
 using Test: @test, @test_throws, @testset
 using Turing
 

test/mcmc/abstractmcmc.jl

@@ -12,7 +12,7 @@ using LogDensityProblemsAD: LogDensityProblemsAD
 using Random: Random
 using ReverseDiff: ReverseDiff
 using StableRNGs: StableRNG
-import Tapir
+import Mooncake
 using Test: @test, @test_throws, @testset
 using Turing
 using Turing.Inference: AdvancedHMC
@@ -116,7 +116,7 @@ end
 
 @testset "External samplers" begin
     @testset "AdvancedHMC.jl" begin
-        # TODO(mhauru) The below tests fail with Tapir, see
+        # TODO(mhauru) The below tests fail with Mooncake, see
         # https://github.com/TuringLang/Turing.jl/pull/2289.
         # Once that is fixed, this should say `for adtype in ADUtils.adbackends`.
         @testset "adtype=$adtype" for adtype in [AutoForwardDiff(), AutoReverseDiff()]

test/mcmc/gibbs.jl

@@ -8,7 +8,7 @@ using Distributions: sample
 using ForwardDiff: ForwardDiff
 using Random: Random
 using ReverseDiff: ReverseDiff
-import Tapir
+import Mooncake
 using Test: @test, @testset
 using Turing
 using Turing: Inference

test/mcmc/gibbs_conditional.jl

@@ -12,7 +12,7 @@ using ReverseDiff: ReverseDiff
 using StableRNGs: StableRNG
 using StatsBase: counts
 using StatsFuns: StatsFuns
-import Tapir
+import Mooncake
 using Test: @test, @testset
 using Turing
 

test/mcmc/hmc.jl

@@ -14,7 +14,7 @@ using LinearAlgebra: I, dot, vec
 import Random
 using StableRNGs: StableRNG
 using StatsFuns: logistic
-import Tapir
+import Mooncake
 using Test: @test, @test_logs, @testset
 using Turing
 

test/mcmc/sghmc.jl

@@ -8,7 +8,7 @@ import ForwardDiff
 using LinearAlgebra: dot
 import ReverseDiff
 using StableRNGs: StableRNG
-import Tapir
+import Mooncake
 using Test: @test, @testset
 using Turing
 

test/test_utils/ad_utils.jl

@@ -4,7 +4,7 @@ using ForwardDiff: ForwardDiff
 using Pkg: Pkg
 using Random: Random
 using ReverseDiff: ReverseDiff
-using Tapir: Tapir
+using Mooncake: Mooncake
 using Test: Test
 using Tracker: Tracker
 using Turing: Turing
@@ -43,7 +43,7 @@ const eltypes_by_adtype = Dict(
         Tracker.TrackedVecOrMat,
         Tracker.TrackedVector,
     ),
-    Turing.AutoTapir => (Tapir.CoDual,),
+    Turing.AutoMooncake => (Mooncake.CoDual,),
 )
 
 """
@@ -285,7 +285,7 @@ All the ADTypes on which we want to run the tests.
 adbackends = [
     Turing.AutoForwardDiff(; chunksize=0),
     Turing.AutoReverseDiff(; compile=false),
-    Turing.AutoTapir(false),
+    Turing.AutoMooncake(; config=Mooncake.Config()),
 ]
 
 end