Allocate output vector with empty tracers

adrhill · Jan 24, 2025 · bd7d0bf · bd7d0bf
1 parent 365d7f9
commit bd7d0bf
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/trace_functions.jl b/src/trace_functions.jl
@@ -51,8 +51,7 @@ end
 
 function trace_function(::Type{T}, f!, y, x) where {T<:Union{AbstractTracer,Dual}}
     xt = trace_input(T, x)
-    yt = similar(y, T)
-    fill!(yt, T(0))
+    yt = fill(myempty(T), size(y))
     f!(yt, xt)
     return xt, yt
 end