Merge pull request #122 from SciML/gd/mooncake_display

fix: pretty printing for AutoMooncake(Forward)

Author: ChrisRackauckas

src/dense.jl

@@ -300,6 +300,13 @@ end
 
 mode(::AutoMooncake) = ReverseMode()
 
+function Base.show(io::IO, backend::AutoMooncake)
+    print(io, AutoMooncake, "(")
+    backend.config !== nothing &&
+        print(io, "config=", repr(backend.config; context = io))
+    print(io, ")")
+end
+
 """
     AutoMooncakeForward
 
@@ -327,6 +334,13 @@ end
 
 mode(::AutoMooncakeForward) = ForwardMode()
 
+function Base.show(io::IO, backend::AutoMooncakeForward)
+    print(io, AutoMooncakeForward, "(")
+    backend.config !== nothing &&
+        print(io, "config=", repr(backend.config; context = io))
+    print(io, ")")
+end
+
 """
     AutoPolyesterForwardDiff{chunksize,T}
 

test/misc.jl

@@ -42,6 +42,10 @@ for backend in [
     ADTypes.AutoForwardDiff(chunksize = 3, tag = :tag),
     ADTypes.AutoGTPSA(),
     ADTypes.AutoGTPSA(; descriptor = Val(:descriptor)),
+    ADTypes.AutoMooncake(),
+    ADTypes.AutoMooncake(; config = :config),
+    ADTypes.AutoMooncakeForward(),
+    ADTypes.AutoMooncakeForward(; config = :config),
     ADTypes.AutoPolyesterForwardDiff(),
     ADTypes.AutoPolyesterForwardDiff(chunksize = 3, tag = :tag),
     ADTypes.AutoReverseDiff(),