DifferentiationInterface.jl/DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/DifferentiationInterfaceSparseMatrixColoringsExt.jl at 38488624cc271aa256477e02724188af21dd07b1 · JuliaDiff/DifferentiationInterface.jl · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
module DifferentiationInterfaceSparseMatrixColoringsExt

using ADTypes: ADTypes, AutoSparse, coloring_algorithm, dense_ad, sparsity_detector
import DifferentiationInterface as DI
using SparseMatrixColorings:
    AbstractColoringResult,
    ColoringProblem,
    coloring,
    column_colors,
    row_colors,
    ncolors,
    column_groups,
    row_groups,
    sparsity_pattern,
    decompress!
import SparseMatrixColorings as SMC

## SMC overloads

abstract type SMCSparseJacobianPrep{SIG} <: DI.SparseJacobianPrep{SIG} end

SMC.sparsity_pattern(prep::DI.SparseJacobianPrep) = prep.sparsity
SMC.column_colors(prep::DI.SparseJacobianPrep) = column_colors(prep.coloring_result)
SMC.column_groups(prep::DI.SparseJacobianPrep) = column_groups(prep.coloring_result)
SMC.row_colors(prep::DI.SparseJacobianPrep) = row_colors(prep.coloring_result)
SMC.row_groups(prep::DI.SparseJacobianPrep) = row_groups(prep.coloring_result)
SMC.ncolors(prep::DI.SparseJacobianPrep) = ncolors(prep.coloring_result)

SMC.sparsity_pattern(prep::DI.SparseHessianPrep) = prep.sparsity
SMC.column_colors(prep::DI.SparseHessianPrep) = column_colors(prep.coloring_result)
SMC.column_groups(prep::DI.SparseHessianPrep) = column_groups(prep.coloring_result)
SMC.ncolors(prep::DI.SparseHessianPrep) = ncolors(prep.coloring_result)

include("jacobian.jl")
include("jacobian_mixed.jl")
include("hessian.jl")

end