overload base.resize<bang>

Author: vpuri3

src/differentiation/jaches_products.jl

@@ -228,6 +228,12 @@ function (L::FwdModeAutoDiffVecProd)(dv, v, p, t)
     L.vecprod!(dv, L.f, L.u, v, L.cache...)
 end
 
+function Base.resize!(L::FwdModeAutoDiffVecProd, n::Integer)
+    for v in L.cache
+        resize!(v, n)
+    end
+end
+
 function JacVec(f, u::AbstractArray, p = nothing, t = nothing;
                 autodiff = AutoForwardDiff(), tag = DeivVecTag(), kwargs...)
     cache, vecprod, vecprod! = if autodiff isa AutoFiniteDiff

src/differentiation/vecjac_products.jl

@@ -87,6 +87,12 @@ function (L::RevModeAutoDiffVecProd{ad, true, false})(dv, v, p, t) where {ad}
     L.vecprod!(dv, L.f, L.u, v, L.cache...)
 end
 
+function Base.resize!(L::RevModeAutoDiffVecProd, n::Integer)
+    for v in L.cache
+        resize!(v, n)
+    end
+end
+
 function VecJac(f, u::AbstractArray, p = nothing, t = nothing; autodiff = AutoFiniteDiff(),
                 kwargs...)
     vecprod, vecprod! = if autodiff isa AutoFiniteDiff