Add AutoGTPSA backend

docs/src/index.md

@@ -30,6 +30,12 @@ AutoFiniteDiff
 AutoFiniteDifferences
 ```
 
+Taylor mode:
+
+```@docs
+AutoGTPSA
+```
+
 ### Reverse mode
 
 ```@docs

src/ADTypes.jl

@@ -39,6 +39,7 @@ export AutoChainRules,
        AutoFiniteDiff,
        AutoFiniteDifferences,
        AutoForwardDiff,
+       AutoGTPSA,
        AutoModelingToolkit,
        AutoPolyesterForwardDiff,
        AutoReverseDiff,

src/dense.jl

@@ -185,6 +185,38 @@ function Base.show(io::IO, backend::AutoForwardDiff{chunksize}) where {chunksize
     print(io, ")")
 end
 
+"""
+    AutoGTPSA{D}
+
+Struct used to select the [GTPSA.jl](https://github.com/bmad-sim/GTPSA.jl) backend for automatic differentiation.
+
+Defined by [ADTypes.jl](https://github.com/SciML/ADTypes.jl).
+
+# Constructors
+
+    AutoGTPSA(; descriptor=nothing)
+
+# Fields
+
+  - `descriptor::D`: can be either
+
+      + a GTPSA `Descriptor` specifying the number of variables/parameters, parameter
+        order, individual variable/parameter truncation orders, and maximum order. See
+        the [GTPSA.jl documentation](https://bmad-sim.github.io/GTPSA.jl/stable/man/c_descriptor/) for more details.
+      + `nothing` to automatically use a `Descriptor` given the context.
+"""
+Base.@kwdef struct AutoGTPSA{D} <: AbstractADType
+    descriptor::D = nothing
+end
+
+mode(::AutoGTPSA) = ForwardMode()
+
+function Base.show(io::IO, backend::AutoGTPSA{D}) where {D}
+    print(io, AutoGTPSA, "(")
+    D != Nothing && print(io, "descriptor=", repr(backend.descriptor; context = io))
+    print(io, ")")
+end
+
 """
     AutoPolyesterForwardDiff{chunksize,T}
 

test/dense.jl

@@ -104,6 +104,20 @@ end
     @test ad.tag == CustomTag()
 end
 
+@testset "AutoGTPSA" begin
+    ad = AutoGTPSA(; descriptor = nothing)
+    @test ad isa AbstractADType
+    @test ad isa AutoGTPSA{Nothing}
+    @test mode(ad) isa ForwardMode
+    @test ad.descriptor === nothing
+
+    ad = AutoGTPSA(; descriptor = Val(:descriptor))
+    @test ad isa AbstractADType
+    @test ad isa AutoGTPSA{Val{:descriptor}}
+    @test mode(ad) isa ForwardMode
+    @test ad.descriptor == Val(:descriptor)
+end
+
 @testset "AutoPolyesterForwardDiff" begin
     ad = AutoPolyesterForwardDiff()
     @test ad isa AbstractADType

test/misc.jl

@@ -33,6 +33,8 @@ for backend in [
     ADTypes.AutoFiniteDifferences(; fdm = :fdm),
     ADTypes.AutoForwardDiff(),
     ADTypes.AutoForwardDiff(chunksize = 3, tag = :tag),
+    ADTypes.AutoGTPSA(),
+    ADTypes.AutoGTPSA(; descriptor = Val(:descriptor)),
     ADTypes.AutoPolyesterForwardDiff(),
     ADTypes.AutoPolyesterForwardDiff(chunksize = 3, tag = :tag),
     ADTypes.AutoReverseDiff(),

test/runtests.jl

@@ -43,6 +43,7 @@ function every_ad()
         AutoFiniteDiff(),
         AutoFiniteDifferences(; fdm = :fdm),
         AutoForwardDiff(),
+        AutoGTPSA(),
         AutoPolyesterForwardDiff(),
         AutoReverseDiff(),
         AutoSymbolics(),
@@ -64,6 +65,8 @@ function every_ad_with_options()
         AutoFiniteDifferences(; fdm = :fdm),
         AutoForwardDiff(),
         AutoForwardDiff(chunksize = 3, tag = :tag),
+        AutoGTPSA(),
+        AutoGTPSA(descriptor = Val(:descriptor)),
         AutoPolyesterForwardDiff(),
         AutoPolyesterForwardDiff(chunksize = 3, tag = :tag),
         AutoReverseDiff(),