Skip to content

Commit d0e3f35

Browse files
committed
Support FastDifferentiation
1 parent 8d50cb9 commit d0e3f35

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

src/ADTypes.jl

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ abstract type AbstractSymbolicDifferentiationMode <: AbstractADType end
1818
abstract type AbstractSparseReverseMode <: AbstractReverseMode end
1919
abstract type AbstractSparseForwardMode <: AbstractForwardMode end
2020
abstract type AbstractSparseFiniteDifferences <: AbstractFiniteDifferencesMode end
21+
abstract type AbstractSparseSymbolicDifferentiationMode <:
22+
AbstractSymbolicDifferentiationMode end
2123

2224
"""
2325
AutoChainRules{RC}
@@ -229,6 +231,20 @@ Chooses [Diffractor.jl](https://github.com/JuliaDiff/Diffractor.jl).
229231
"""
230232
struct AutoDiffractor <: AbstractADType end
231233

234+
"""
235+
AutoFastDifferentiation
236+
237+
Chooses [FastDifferentiation.jl](https://github.com/brianguenter/FastDifferentiation.jl).
238+
"""
239+
struct AutoFastDifferentiation <: AbstractSymbolicDifferentiationMode end
240+
241+
"""
242+
AutoSparseFastDifferentiation
243+
244+
Chooses [FastDifferentiation.jl](https://github.com/brianguenter/FastDifferentiation.jl) while exploiting sparsity.
245+
"""
246+
struct AutoSparseFastDifferentiation <: AbstractSparseSymbolicDifferentiationMode end
247+
232248
export AutoChainRules,
233249
AutoDiffractor,
234250
AutoFiniteDiff,
@@ -244,5 +260,7 @@ export AutoChainRules,
244260
AutoSparseZygote,
245261
AutoSparseReverseDiff,
246262
AutoPolyesterForwardDiff,
247-
AutoSparsePolyesterForwardDiff
263+
AutoSparsePolyesterForwardDiff,
264+
AutoFastDifferentiation,
265+
AutoSparseFastDifferentiation
248266
end

0 commit comments

Comments
 (0)