diff --git a/src/differentiation/compute_jacobian_ad.jl b/src/differentiation/compute_jacobian_ad.jl index 9e715892..f1ca7fde 100644 --- a/src/differentiation/compute_jacobian_ad.jl +++ b/src/differentiation/compute_jacobian_ad.jl @@ -36,7 +36,7 @@ function ForwardColorJacCache(f::F,x,_chunksize = nothing; if x isa Array p = generate_chunked_partials(x,colorvec,chunksize) - t = similar(x,Dual{T,eltype(x),length(first(first(p)))}) + t = Array{Dual{T,eltype(x),length(first(first(p)))}}(undef,size(x)) for i in eachindex(t) t[i] = Dual{T,eltype(x),length(first(first(p)))}(x[i],ForwardDiff.Partials(first(p)[i])) end