From 3115694fc3fcdd9462f3a6f0707a1edffc3d10be Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Tue, 21 Dec 2021 21:40:19 -0500 Subject: [PATCH 1/2] Fix diffeq tag --- src/diffeq.jl | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) 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 From e6bc9dbde0c3a4205861bfc92c9c4c6a9f40606d Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Tue, 21 Dec 2021 21:40:42 -0500 Subject: [PATCH 2/2] Update Project.toml --- Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"