Skip to content

Commit c98c023

Browse files
committed
use SparseMatrixColorings instead of SparseDiffTools
1 parent bb29f56 commit c98c023

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

lib/OrdinaryDiffEqDifferentiation/src/OrdinaryDiffEqDifferentiation.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ using FastBroadcast: @..
5454

5555
using ConcreteStructs: @concrete
5656

57-
import SparseMatrixColorings
57+
import SparseMatrixColorings: GreedyColoringAlgorithm, column_colors, ColoringProblem, coloring
5858

5959
@static if isdefined(DiffEqBase, :OrdinaryDiffEqTag)
6060
import DiffEqBase: OrdinaryDiffEqTag

lib/OrdinaryDiffEqDifferentiation/src/derivative_wrappers.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,9 @@ function sparsity_colorvec(f, x)
377377
end
378378
end
379379

380+
col_alg = GreedyColoringAlgorithm()
381+
col_prob = ColoringProblem()
380382
colorvec = DiffEqBase.has_colorvec(f) ? f.colorvec :
381-
(isnothing(sparsity) ? (1:length(x)) : matrix_colors(sparsity))
383+
(isnothing(sparsity) ? (1:length(x)) : column_colors(coloring(sparsity, col_prob, col_alg)))
382384
sparsity, colorvec
383385
end

0 commit comments

Comments
 (0)