Merge pull request #34 from SciML/gd/new_types

Support FastDifferentiation.jl
SciML · Apr 3, 2024 · ec248df · ec248df
2 parents 93b8c6c + 13857fb
commit ec248df
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/src/ADTypes.jl b/src/ADTypes.jl
@@ -18,6 +18,8 @@ abstract type AbstractSymbolicDifferentiationMode <: AbstractADType end
 abstract type AbstractSparseReverseMode <: AbstractReverseMode end
 abstract type AbstractSparseForwardMode <: AbstractForwardMode end
 abstract type AbstractSparseFiniteDifferences <: AbstractFiniteDifferencesMode end
+abstract type AbstractSparseSymbolicDifferentiationMode <:
+ AbstractSymbolicDifferentiationMode end
 
 """
  AutoChainRules{RC}
@@ -233,6 +235,20 @@ Chooses [Diffractor.jl](https://github.com/JuliaDiff/Diffractor.jl).
 """
 struct AutoDiffractor <: AbstractADType end
 
+"""
+ AutoFastDifferentiation
+
+Chooses [FastDifferentiation.jl](https://github.com/brianguenter/FastDifferentiation.jl).
+"""
+struct AutoFastDifferentiation <: AbstractSymbolicDifferentiationMode end
+
+"""
+ AutoSparseFastDifferentiation
+
+Chooses [FastDifferentiation.jl](https://github.com/brianguenter/FastDifferentiation.jl) while exploiting sparsity.
+"""
+struct AutoSparseFastDifferentiation <: AbstractSparseSymbolicDifferentiationMode end
+
 export AutoChainRules,
  AutoDiffractor,
  AutoFiniteDiff,
@@ -248,5 +264,7 @@ export AutoChainRules,
  AutoSparseZygote,
  AutoSparseReverseDiff,
  AutoPolyesterForwardDiff,
- AutoSparsePolyesterForwardDiff
+ AutoSparsePolyesterForwardDiff,
+ AutoFastDifferentiation,
+ AutoSparseFastDifferentiation
 end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -98,4 +98,12 @@ struct CustomTag end
  adtype = AutoDiffractor()
  @test adtype isa ADTypes.AbstractADType
  @test adtype isa AutoDiffractor
+
+ adtype = AutoFastDifferentiation()
+ @test adtype isa ADTypes.AbstractADType
+ @test adtype isa AutoFastDifferentiation
+
+ adtype = AutoSparseFastDifferentiation()
+ @test adtype isa ADTypes.AbstractADType
+ @test adtype isa AutoSparseFastDifferentiation
 end