Merge pull request #39 from prbzrg/support-dot

ChrisRackauckas · web-flow · commit c9b269273b2b · 2024-04-11T07:35:34.000-04:00
Support dot broadcast
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
