Merge pull request #29 from SciML/gd/diffractor

Add AutoDiffractor
SciML · Mar 11, 2024 · 9f17b82 · 9f17b82
2 parents 59ded48 + 41d5a38
commit 9f17b82
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/ADTypes.jl b/src/ADTypes.jl
@@ -222,7 +222,15 @@ Base.@kwdef struct AutoSparseReverseDiff <: AbstractSparseReverseMode
  compile::Bool = false
 end
 
+"""
+ AutoDiffractor
+
+Chooses [Diffractor.jl](https://github.com/JuliaDiff/Diffractor.jl).
+"""
+struct AutoDiffractor <: AbstractADType end
+
 export AutoChainRules,
+ AutoDiffractor,
  AutoFiniteDiff,
  AutoFiniteDifferences,
  AutoForwardDiff,

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -91,4 +91,8 @@ struct CustomTag end
  adtype = AutoEnzyme(; mode = Val(:Reverse))
  @test adtype isa ADTypes.AbstractADType
  @test adtype isa AutoEnzyme{Val{:Reverse}}
+
+ adtype = AutoDiffractor()
+ @test adtype isa ADTypes.AbstractADType
+ @test adtype isa AutoDiffractor
 end