From 3d639a2bafe081e5418b519f5004cfb1138a6e49 Mon Sep 17 00:00:00 2001
From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
Date: Fri, 15 Aug 2025 10:26:53 +0200
Subject: [PATCH] fix: pretty printing for AutoMooncake(Forward)

---
 src/dense.jl | 14 ++++++++++++++
 test/misc.jl |  4 ++++
 2 files changed, 18 insertions(+)

diff --git a/src/dense.jl b/src/dense.jl
index 4514c2e..38072e6 100644
--- a/src/dense.jl
+++ b/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}
 
diff --git a/test/misc.jl b/test/misc.jl
index d6f693c..95c73bc 100644
--- a/test/misc.jl
+++ b/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(),