remove call from value_and_pullback!!

compintell · AstitvaAggarwal · Mar 21, 2025 · Mar 21, 2025 · Mar 21, 2025 · Mar 22, 2025
commit 0a3fd0ea8db1af9b0a2dd20adbd17135652a3f80
diff --git a/src/interface.jl b/src/interface.jl
@@ -16,7 +16,7 @@ function __value_and_pullback!!(
     out, pb!! = rule(fx_fwds...)
     @assert _typeof(tangent(out)) == fdata_type(T)
     increment!!(tangent(out), fdata(ȳ))
-    v = y_cache === nothing ? _copy_temp(primal(out)) : _copy!!(y_cache, primal(out))
+    v = y_cache === nothing ? copy(primal(out)) : _copy!!(y_cache, primal(out))
     return v, tuple_map((f, r) -> tangent(fdata(tangent(f)), r), fx, pb!!(rdata(ȳ)))
 end