Add AutoDiffractor

SciML · Mar 11, 2024 · e3ab11a · e3ab11a
1 parent 4bd7704
commit e3ab11a
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -13,9 +13,10 @@ Just run the following code in a Julia REPL to find out:
 julia> using ADTypes
 
 julia> names(ADTypes)
-16-element Vector{Symbol}:
+17-element Vector{Symbol}:
  :ADTypes
  :AutoChainRules
+ :AutoDiffractor
  :AutoEnzyme
  :AutoFiniteDiff
  :AutoFiniteDifferences

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