Add AutoMooncake type

Project.toml

@@ -1,7 +1,7 @@
 name = "ADTypes"
 uuid = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 authors = ["Vaibhav Dixit <vaibhavyashdixit@gmail.com>, Guillaume Dalle and contributors"]
-version = "1.8.1"
+version = "1.9.0"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

docs/src/index.md

@@ -39,9 +39,10 @@ AutoGTPSA
 ### Reverse mode
 
 ```@docs
+AutoMooncake
 AutoReverseDiff
-AutoTapir
 AutoTracker
+AutoTapir
 AutoZygote
 ```
 

src/ADTypes.jl

@@ -42,6 +42,7 @@ export AutoChainRules,
  AutoForwardDiff,
  AutoGTPSA,
  AutoModelingToolkit,
+ AutoMooncake,
  AutoPolyesterForwardDiff,
  AutoReverseDiff,
  AutoSymbolics,

src/dense.jl

@@ -227,6 +227,30 @@ function Base.show(io::IO, backend::AutoGTPSA{D}) where {D}
  print(io, ")")
 end
 
+"""
+ AutoMooncake
+
+Struct used to select the [Mooncake.jl](https://github.com/compintell/Mooncake.jl) backend
+for automatic differentiation.
+
+Defined by [ADTypes.jl](https://github.com/SciML/ADTypes.jl).
+
+# Constructors
+
+ AutoMooncake(; config)
+
+# Fields
+
+ - `config`: either `nothing` or an instance of `Mooncake.Config` -- see the docstring for
+ `Mooncake.Config` for more information. `AutoMooncake(; config=nothing)` is equivalent
+ to `AutoMooncake(; config=Mooncake.Config())`, i.e. the default configuration.
+"""
+Base.@kwdef struct AutoMooncake{Tconfig} <: AbstractADType
+ config::Tconfig
+end
+
+mode(::AutoMooncake) = ReverseMode()
+
 """
  AutoPolyesterForwardDiff{chunksize,T}
 
@@ -323,26 +347,10 @@ mode(::AutoSymbolics) = SymbolicMode()
 """
  AutoTapir
 
-Struct used to select the [Tapir.jl](https://github.com/withbayes/Tapir.jl) backend for automatic differentiation.
-
-Defined by [ADTypes.jl](https://github.com/SciML/ADTypes.jl).
-
-# Constructors
-
- AutoTapir(; safe_mode=true)
-
-# Fields
-
- - `safe_mode::Bool`: whether to run additional checks to catch errors early. While this is
- on by default to ensure that users are aware of this option, you should generally turn
- it off for actual use, as it has substantial performance implications.
- If you encounter a problem with using Tapir (it fails to differentiate a function, or
- something truly nasty like a segfault occurs), then you should try switching `safe_mode`
- on and look at what happens. Often errors are caught earlier and the error messages are
- more useful.
+This ADType is deprecated. `AutoMooncake` should be used instead.
 """
-Base.@kwdef struct AutoTapir <: AbstractADType
- safe_mode::Bool = true
+struct AutoTapir <: AbstractADType
+ safe_mode::Bool
 end
 
 mode(::AutoTapir) = ReverseMode()

src/legacy.jl

@@ -36,3 +36,10 @@ function AutoModelingToolkit(; obj_sparse::Bool = false, cons_sparse::Bool = fal
  :AutoModelingToolkit; force = false)
  return mtk_to_symbolics(obj_sparse, cons_sparse)
 end
+
+function AutoTapir(; safe_mode=true)
+ Base.depwarn(
+ "AutoTapir is deprecated in favour of AutoMooncake.", :AutoTapir; force=false
+ )
+ return AutoTapir(safe_mode)
+end

test/dense.jl

@@ -119,6 +119,14 @@ end
  @test ad.descriptor == Val(:descriptor)
 end
 
+@testset "AutoMooncake" begin
+ ad = AutoMooncake(; config=nothing)
+ @test ad isa AbstractADType
+ @test ad isa AutoMooncake
+ @test mode(ad) isa ReverseMode
+ @test ad.config === nothing
+end
+
 @testset "AutoPolyesterForwardDiff" begin
  ad = AutoPolyesterForwardDiff()
  @test ad isa AbstractADType

test/legacy.jl

@@ -68,3 +68,7 @@ end
  ad = @test_deprecated AutoReverseDiff(true)
  @test ad.compile
 end
+
+@testset "AutoTapir" begin
+ @test_deprecated AutoTapir()
+end