Merge pull request #350 from yebai/patch-1

ChrisRackauckas · web-flow · commit 410d4b783f6b · 2026-04-11T02:08:06.000Z
Add friendly_tangent_cache function to Mooncake
diff --git a/ext/ComponentArraysMooncakeExt.jl b/ext/ComponentArraysMooncakeExt.jl
@@ -11,4 +11,8 @@ function Mooncake.increment_and_get_rdata!(
     return Mooncake.increment_and_get_rdata!(f.data[:data], r, t)
 end
 
+function Mooncake.friendly_tangent_cache(x::ComponentArray)
+    Mooncake.FriendlyTangentCache{Mooncake.AsPrimal}(copy(x))
+end
+
 end
