Merge pull request #101 from JuliaDiff/restructure

use ArrayInterface.restructure

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.5.0"
+version = "1.6.0"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -19,7 +19,7 @@ VertexSafeGraphs = "19fa3120-7c27-5ec5-8db8-b0b0aa330d6f"
 
 [compat]
 Adapt = "1"
-ArrayInterface = "2"
+ArrayInterface = "2.8"
 Compat = "2.2, 3"
 DataStructures = "0.17"
 FiniteDiff = "2"

src/differentiation/compute_jacobian_ad.jl

@@ -31,15 +31,15 @@ function ForwardColorJacCache(f,x,_chunksize = nothing;
     end
 
     p = adapt.(typeof(x),generate_chunked_partials(x,colorvec,chunksize))
-    t = reshape(Dual{ForwardDiff.Tag(f,eltype(vec(x)))}.(vec(x),first(p)),size(x)...)
-
+    _t = Dual{ForwardDiff.Tag(f,eltype(vec(x)))}.(vec(x),first(p))
+    t = ArrayInterface.restructure(x,_t)
     if dx isa Nothing
         fx = similar(t)
         _dx = similar(x)
     else
         tup = first(first(p)) .* false
-        pi = adapt.(typeof(dx),[tup for i in 1:length(dx)])
-        fx = reshape(Dual{ForwardDiff.Tag(f,eltype(vec(x)))}.(vec(dx),pi),size(dx)...)
+        _pi = adapt.(typeof(dx),[tup for i in 1:length(dx)])
+        fx = reshape(Dual{ForwardDiff.Tag(f,eltype(vec(x)))}.(vec(dx),_pi),size(dx)...)
         _dx = dx
     end