Skip to content

Commit 9f17b82

Browse files
Merge pull request #29 from SciML/gd/diffractor
Add AutoDiffractor
2 parents 59ded48 + 41d5a38 commit 9f17b82

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

src/ADTypes.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,15 @@ Base.@kwdef struct AutoSparseReverseDiff <: AbstractSparseReverseMode
222222
compile::Bool = false
223223
end
224224

225+
"""
226+
AutoDiffractor
227+
228+
Chooses [Diffractor.jl](https://github.com/JuliaDiff/Diffractor.jl).
229+
"""
230+
struct AutoDiffractor <: AbstractADType end
231+
225232
export AutoChainRules,
233+
AutoDiffractor,
226234
AutoFiniteDiff,
227235
AutoFiniteDifferences,
228236
AutoForwardDiff,

test/runtests.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,4 +91,8 @@ struct CustomTag end
9191
adtype = AutoEnzyme(; mode = Val(:Reverse))
9292
@test adtype isa ADTypes.AbstractADType
9393
@test adtype isa AutoEnzyme{Val{:Reverse}}
94+
95+
adtype = AutoDiffractor()
96+
@test adtype isa ADTypes.AbstractADType
97+
@test adtype isa AutoDiffractor
9498
end

0 commit comments

Comments
 (0)