Fixes

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/utils.jl

@@ -26,7 +26,7 @@ nanify(::NoRData) = NoRData()
 
 function nanify_fdata_and_rdata!!(contexts::Vararg{CoDual, C}) where {C}
     primal_contexts = map(primal, contexts)
-    fdata_contexts = map(fdata, contexts)
+    fdata_contexts = map(tangent, contexts)
     zero_rdata_contexts = map(zero_rdata, primal_contexts)
     foreach(fdata_contexts) do fc
         increment!!(fc, nanify(fc))

DifferentiationInterface/src/misc/differentiate_with.jl

@@ -83,14 +83,16 @@ function (dw::DifferentiateWith{C})(x, args::Vararg{Any, C}) where {C}
 end
 
 function Base.show(io::IO, dw::DifferentiateWith)
-    (; f, backend) = dw
+    (; f, backend, context_wrappers) = dw
     return print(
         io,
         DifferentiateWith,
         "(",
         repr(f; context = io),
         ", ",
         repr(backend; context = io),
+        ", ",
+        repr(context_wrappers; context = io),
         ")",
     )
 end