Promote ReverseDiff compile field to type

Author: avik-pal

Project.toml

@@ -3,7 +3,7 @@ uuid = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 authors = [
     "Vaibhav Dixit <[email protected]>, Guillaume Dalle and contributors",
 ]
-version = "1.4.0"
+version = "1.5.0"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/ADTypes.jl

@@ -16,6 +16,9 @@ abstract type AbstractADType end
 
 Base.broadcastable(ad::AbstractADType) = Ref(ad)
 
+@inline _unwrap_val(::Val{T}) where {T} = T
+@inline _unwrap_val(x) = x
+
 include("mode.jl")
 include("dense.jl")
 include("sparse.jl")

src/dense.jl

@@ -186,14 +186,19 @@ Defined by [ADTypes.jl](https://github.com/SciML/ADTypes.jl).
 
 # Constructors
 
-    AutoReverseDiff(; compile=false)
+    AutoReverseDiff(; compile::Union{Val, Bool} = Val(false))
 
 # Fields
 
-  - `compile::Bool`: whether to [compile the tape](https://juliadiff.org/ReverseDiff.jl/api/#ReverseDiff.compile) prior to differentiation
+  - `compile::Union{Val, Bool}`: whether to [compile the tape](https://juliadiff.org/ReverseDiff.jl/api/#ReverseDiff.compile) prior to differentiation
 """
-Base.@kwdef struct AutoReverseDiff <: AbstractADType
-    compile::Bool = false
+struct AutoReverseDiff{C} <: AbstractADType
+    compile::Bool  # this field if left for legacy reasons
+
+    function AutoReverseDiff(; compile::Union{Val, Bool} = Val(false))
+        _compile = _unwrap_val(compile)
+        return new{_compile}(_compile)
+    end
 end
 
 mode(::AutoReverseDiff) = ReverseMode()

src/symbols.jl

@@ -22,8 +22,8 @@ ADTypes.AutoZygote()
 Auto(package::Symbol, args...; kws...) = Auto(Val(package), args...; kws...)
 
 for backend in (:ChainRules, :Diffractor, :Enzyme, :FastDifferentiation,
-                :FiniteDiff, :FiniteDifferences, :ForwardDiff, :PolyesterForwardDiff,
-                :ReverseDiff, :Symbolics, :Tapir, :Tracker, :Zygote)
-    @eval Auto(::Val{$(QuoteNode(backend))}, args...; kws...) = $(Symbol(:Auto, backend))(args...; kws...)
+    :FiniteDiff, :FiniteDifferences, :ForwardDiff, :PolyesterForwardDiff,
+    :ReverseDiff, :Symbolics, :Tapir, :Tracker, :Zygote)
+    @eval Auto(::Val{$(QuoteNode(backend))}, args...; kws...) = $(Symbol(:Auto, backend))(
+        args...; kws...)
 end
-

test/dense.jl

@@ -113,7 +113,7 @@ end
 end
 
 @testset "AutoReverseDiff" begin
-    ad = AutoReverseDiff()
+    ad = @inferred AutoReverseDiff()
     @test ad isa AbstractADType
     @test ad isa AutoReverseDiff
     @test mode(ad) isa ReverseMode
@@ -124,6 +124,12 @@ end
     @test ad isa AutoReverseDiff
     @test mode(ad) isa ReverseMode
     @test ad.compile
+
+    ad = @inferred AutoReverseDiff(; compile = Val(true))
+    @test ad isa AbstractADType
+    @test ad isa AutoReverseDiff
+    @test mode(ad) isa ReverseMode
+    @test ad.compile
 end
 
 @testset "AutoSymbolics" begin