Fix for ComponentArrays

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