Make dual

JuliaDiff · Jul 28, 2024 · 61cc9e4 · 61cc9e4
1 parent 09c4a97
commit 61cc9e4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/DifferentiationInterface/ext/DifferentiationInterfaceForwardDiffExt/utils.jl b/DifferentiationInterface/ext/DifferentiationInterfaceForwardDiffExt/utils.jl
@@ -19,10 +19,18 @@ function make_dual(::Type{T}, x::Number, dx::Number) where {T}
     return Dual{T}(x, dx)
 end
 
+function make_dual(::Type{T}, x, dx) where {T}
+    return Dual{T}.(x, dx)
+end
+
 function make_dual(::Type{T}, x::Number, dx::Batch{B,<:Number}) where {T,B}
     return Dual{T}(x, dx.elements...)
 end
 
+function make_dual(::Type{T}, x, dx::Batch{B}) where {T,B}
+    return Dual{T}.(x, dx.elements...)
+end
+
 function make_dual!(::Type{T}, xdual, x, dx) where {T}
     return xdual .= Dual{T}.(x, dx)
 end