Support dot broadcast

prbzrg · prbzrg · commit ba9a4766b3c0 · 2024-04-11T14:51:35.000+03:30
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
