diff --git a/Project.toml b/Project.toml index 1340f61..2c17262 100644 --- a/Project.toml +++ b/Project.toml @@ -22,7 +22,7 @@ FiniteDiff = "2.3" ForwardDiff = "0.10" OrdinaryDiffEq = "5.33, 6" RecursiveArrayTools = "1,2" -SparseDiffTools = "1.6" +SparseDiffTools = "1.19" StochasticDiffEq = "6.13" TreeViews = "0.3" julia = "1.6" diff --git a/src/diffeq.jl b/src/diffeq.jl index 5727f28..135f350 100644 --- a/src/diffeq.jl +++ b/src/diffeq.jl @@ -103,25 +103,29 @@ end function add_node_jac_config!(cache,config::SparseDiffTools.ForwardColorJacCache,i,x,node...) @assert cache.jac_config.colorvec isa UnitRange - cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize) + T = SparseDiffTools.gettag(eltype(config.t)) + cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize,tag = T) nothing end function add_node_jac_config!(cache,config::SparseDiffTools.ForwardColorJacCache,i,x) @assert cache.jac_config.colorvec isa UnitRange - cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize) + T = SparseDiffTools.gettag(eltype(config.t)) + cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize,tag = T) nothing end function remove_node_jac_config!(cache,config::SparseDiffTools.ForwardColorJacCache,i,x,node...) @assert cache.jac_config.colorvec isa UnitRange - cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize) + T = SparseDiffTools.gettag(eltype(config.t)) + cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize,tag = T) nothing end function remove_node_jac_config!(cache,config::SparseDiffTools.ForwardColorJacCache,i,x) @assert cache.jac_config.colorvec isa UnitRange - cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize) + T = SparseDiffTools.gettag(eltype(config.t)) + cache.jac_config = SparseDiffTools.ForwardColorJacCache(cache.uf,cache.uprev,config.chunksize,tag = T) nothing end