Skip to content
This repository was archived by the owner on Aug 22, 2025. It is now read-only.

Commit b3d079c

Browse files
author
Avik Pal
committed
Add vecjac operators
1 parent 867ccda commit b3d079c

4 files changed

Lines changed: 198 additions & 100 deletions

File tree

Project.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ Requires = "ae029012-a4dd-5104-9daa-d747884805df"
1616
SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
1717
StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
1818
VertexSafeGraphs = "19fa3120-7c27-5ec5-8db8-b0b0aa330d6f"
19+
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
1920

2021
[compat]
2122
Adapt = "1, 2.0, 3.0"

src/SparseDiffTools.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ using LightGraphs: SimpleGraph
88
using Requires
99
using VertexSafeGraphs
1010
using Adapt
11+
using Zygote
1112

1213
using LinearAlgebra
1314
using SparseArrays, ArrayInterface
@@ -30,6 +31,8 @@ export contract_color,
3031
ForwardColorJacCache,
3132
auto_jacvec,auto_jacvec!,
3233
num_jacvec,num_jacvec!,
34+
auto_vecjac,auto_vecjac!,
35+
num_vecjac,num_vecjac!,
3336
num_hesvec,num_hesvec!,
3437
numauto_hesvec,numauto_hesvec!,
3538
autonum_hesvec,autonum_hesvec!,
@@ -48,6 +51,7 @@ include("coloring/greedy_star2_coloring.jl")
4851
include("coloring/matrix2graph.jl")
4952
include("differentiation/compute_jacobian_ad.jl")
5053
include("differentiation/jaches_products.jl")
54+
include("differentiation/vecjac_products.jl")
5155

5256
Base.@pure __parameterless_type(T) = Base.typename(T).wrapper
5357
parameterless_type(x) = parameterless_type(typeof(x))

0 commit comments

Comments
 (0)