Add AutoGTPSA taylor mode

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,42 @@ 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=\n", 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/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=:descriptor),
  AutoPolyesterForwardDiff(),
  AutoPolyesterForwardDiff(chunksize = 3, tag = :tag),
  AutoReverseDiff(),