Merge pull request #39 from prbzrg/support-dot

Support dot broadcast
SciML · Apr 11, 2024 · c9b2692 · c9b2692
2 parents dd87ccf + ba9a476
commit c9b2692
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/ADTypes.jl b/src/ADTypes.jl
@@ -249,6 +249,8 @@ Chooses [FastDifferentiation.jl](https://github.com/brianguenter/FastDifferentia
 """
 struct AutoSparseFastDifferentiation <: AbstractSparseSymbolicDifferentiationMode end
 
+Base.broadcastable(ad::AbstractADType) = Ref(ad)
+
 export AutoChainRules,
        AutoDiffractor,
        AutoFiniteDiff,

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -106,4 +106,6 @@ struct CustomTag end
     adtype = AutoSparseFastDifferentiation()
     @test adtype isa ADTypes.AbstractADType
     @test adtype isa AutoSparseFastDifferentiation
+
+    @test identity.(adtype) == adtype
 end