Add Symbol -> AbstractADType mapping

Project.toml

@@ -3,7 +3,7 @@ uuid = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 authors = [
     "Vaibhav Dixit <[email protected]>, Guillaume Dalle and contributors",
 ]
-version = "1.3.0"
+version = "1.3.1"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

docs/src/index.md

@@ -92,3 +92,9 @@ ADTypes.ForwardOrReverseMode
 ADTypes.ReverseMode
 ADTypes.SymbolicMode
 ```
+
+## Miscellaneous
+
+```@docs
+ADTypes.Auto
+```

src/ADTypes.jl

@@ -20,6 +20,7 @@ include("mode.jl")
 include("dense.jl")
 include("sparse.jl")
 include("legacy.jl")
+include("symbols.jl")
 
 if !isdefined(Base, :get_extension)
     include("../ext/ADTypesChainRulesCoreExt.jl")

src/symbols.jl

@@ -0,0 +1,42 @@
+"""
+    ADTypes.Auto(package::Symbol)
+
+A shortcut that converts an AD package name into an instance of [`AbstractADType`](@ref), with all parameters set to their default values.
+
+!!! warning
+
+    This function is type-unstable by design and might lead to suboptimal performance.
+    In most cases, you should never need it: use the individual backend types directly.
+
+# Example
+
+```jldoctest
+import ADTypes
+backend = ADTypes.Auto(:Zygote)
+
+# output
+
+ADTypes.AutoZygote()
+```
+"""
+Auto(package::Symbol) = Auto(Val(package))
+
+Auto(::Val{:Diffractor}) = AutoDiffractor()
+Auto(::Val{:Enzyme}) = AutoEnzyme()
+Auto(::Val{:FastDifferentiation}) = AutoFastDifferentiation()
+Auto(::Val{:FiniteDiff}) = AutoFiniteDiff()
+Auto(::Val{:ForwardDiff}) = AutoForwardDiff()
+Auto(::Val{:PolyesterForwardDiff}) = AutoPolyesterForwardDiff()
+Auto(::Val{:ReverseDiff}) = AutoReverseDiff()
+Auto(::Val{:Symbolics}) = AutoSymbolics()
+Auto(::Val{:Tapir}) = AutoTapir()
+Auto(::Val{:Tracker}) = AutoTracker()
+Auto(::Val{:Zygote}) = AutoZygote()
+
+function Auto(::Val{:ChainRules})
+    throw(ArgumentError("ChainRules backend has mandatory arguments"))
+end
+
+function Auto(::Val{:FiniteDifferences})
+    throw(ArgumentError("FiniteDifferences backend has mandatory arguments"))
+end

test/runtests.jl

@@ -66,6 +66,9 @@ end
     @testset "Sparse" begin
         include("sparse.jl")
     end
+    @testset "Symbols" begin
+        include("symbols.jl")
+    end
     @testset "Legacy" begin
         include("legacy.jl")
     end

test/symbols.jl

@@ -0,0 +1,17 @@
+using ADTypes
+using Test
+
+@test ADTypes.Auto(:Diffractor) isa AutoDiffractor
+@test ADTypes.Auto(:Enzyme) isa AutoEnzyme
+@test ADTypes.Auto(:FastDifferentiation) isa AutoFastDifferentiation
+@test ADTypes.Auto(:FiniteDiff) isa AutoFiniteDiff
+@test ADTypes.Auto(:ForwardDiff) isa AutoForwardDiff
+@test ADTypes.Auto(:PolyesterForwardDiff) isa AutoPolyesterForwardDiff
+@test ADTypes.Auto(:ReverseDiff) isa AutoReverseDiff
+@test ADTypes.Auto(:Symbolics) isa AutoSymbolics
+@test ADTypes.Auto(:Tapir) isa AutoTapir
+@test ADTypes.Auto(:Tracker) isa AutoTracker
+@test ADTypes.Auto(:Zygote) isa AutoZygote
+
+@test_throws ArgumentError ADTypes.Auto(:ChainRules)
+@test_throws ArgumentError ADTypes.Auto(:FiniteDifferences)