diff --git a/src/dense.jl b/src/dense.jl index c69d476..3d73448 100644 --- a/src/dense.jl +++ b/src/dense.jl @@ -185,7 +185,6 @@ function Base.show(io::IO, backend::AutoForwardDiff{chunksize}) where {chunksize print(io, ")") end - """ AutoGTPSA{D} @@ -201,11 +200,10 @@ Defined by [ADTypes.jl](https://github.com/SciML/ADTypes.jl). - `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 + + 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 @@ -215,12 +213,10 @@ 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)) + D != Nothing && print(io, "descriptor=", repr(backend.descriptor; context = io)) print(io, ")") end - - """ AutoPolyesterForwardDiff{chunksize,T} diff --git a/test/misc.jl b/test/misc.jl index b3e501f..e22c380 100644 --- a/test/misc.jl +++ b/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(), diff --git a/test/runtests.jl b/test/runtests.jl index 6b82466..9f6db28 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -66,7 +66,7 @@ function every_ad_with_options() AutoForwardDiff(), AutoForwardDiff(chunksize = 3, tag = :tag), AutoGTPSA(), - AutoGTPSA(descriptor=:descriptor), + AutoGTPSA(descriptor = Val(:descriptor)), AutoPolyesterForwardDiff(), AutoPolyesterForwardDiff(chunksize = 3, tag = :tag), AutoReverseDiff(),