Merge pull request #195 from JuliaDiff/tags

allow for setting tags in operators

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "SparseDiffTools"
 uuid = "47a9eef4-7e08-11e9-0b38-333d64bd3804"
 authors = ["Pankaj Mishra <[email protected]>", "Chris Rackauckas <[email protected]>"]
-version = "1.25.1"
+version = "1.25.2"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/differentiation/jaches_products.jl

@@ -195,10 +195,10 @@ struct JacVec{F,T1,T2,xType}
     autodiff::Bool
 end
 
-function JacVec(f, x::AbstractArray; autodiff = true)
+function JacVec(f, x::AbstractArray, tag = DeivVecTag(); autodiff = true)
     if autodiff
-        cache1 = Dual{typeof(ForwardDiff.Tag(DeivVecTag(),eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
-        cache2 = Dual{typeof(ForwardDiff.Tag(DeivVecTag(),eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
+        cache1 = Dual{typeof(ForwardDiff.Tag(tag,eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
+        cache2 = Dual{typeof(ForwardDiff.Tag(tag,eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
     else
         cache1 = similar(x)
         cache2 = similar(x)
@@ -264,10 +264,10 @@ struct HesVecGrad{G,T1,T2,uType}
     autodiff::Bool
 end
 
-function HesVecGrad(g, x::AbstractArray; autodiff = false)
+function HesVecGrad(g, x::AbstractArray, tag = DeivVecTag(); autodiff = false)
     if autodiff
-        cache1 = Dual{typeof(ForwardDiff.Tag(DeivVecTag(),eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
-        cache2 = Dual{typeof(ForwardDiff.Tag(DeivVecTag(),eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
+        cache1 = Dual{typeof(ForwardDiff.Tag(tag,eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
+        cache2 = Dual{typeof(ForwardDiff.Tag(tag,eltype(x))),eltype(x),1}.(x, ForwardDiff.Partials.(tuple.(x)))
     else
         cache1 = similar(x)
         cache2 = similar(x)