From 5b1ea0ac5e535597812e72719d8827b2e2542a6b Mon Sep 17 00:00:00 2001 From: Chris Rackauckas Date: Thu, 24 Mar 2022 06:40:45 -0400 Subject: [PATCH] Better sparseAD inference --- src/differentiation/compute_jacobian_ad.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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