fix for newest OrdinaryDiffEq

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "MultiScaleArrays"
 uuid = "f9640e96-87f6-5992-9c3b-0743c6a49ffa"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "1.8.0"
+version = "1.8.1"
 
 [deps]
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"

src/diffeq.jl

@@ -169,6 +169,26 @@ function remove_node_grad_config!(cache,grad_config::ForwardDiff.DerivativeConfi
   nothing
 end
 
+function add_node_grad_config!(cache,grad_config::AbstractArray,i,x)
+    cache.grad_config = ForwardDiff.Dual{typeof(ForwardDiff.Tag(cache.tf,eltype(cache.du1)))}.(cache.du1, cache.du1)
+    nothing
+end
+
+function add_node_grad_config!(cache,grad_config::AbstractArray,i,x,I...)
+  cache.grad_config = ForwardDiff.Dual{typeof(ForwardDiff.Tag(cache.tf,eltype(cache.du1)))}.(cache.du1, cache.du1)
+  nothing
+end
+
+function remove_node_grad_config!(cache,grad_config::AbstractArray,i,x)
+    cache.grad_config = ForwardDiff.Dual{typeof(ForwardDiff.Tag(cache.tf,eltype(cache.du1)))}.(cache.du1, cache.du1)
+    nothing
+end
+
+function remove_node_grad_config!(cache,grad_config::AbstractArray,i,x,I...)
+  cache.grad_config = ForwardDiff.Dual{typeof(ForwardDiff.Tag(cache.tf,eltype(cache.du1)))}.(cache.du1, cache.du1)
+  nothing
+end
+
 function add_node_grad_config!(cache,grad_config::FiniteDiff.GradientCache,i,x,I...)
   grad_config.fx !== nothing && add_node!(grad_config.fx, recursivecopy(x), I...)
   grad_config.c1 !== nothing && add_node!(grad_config.c1, recursivecopy(x), I...)

src/math.jl

@@ -14,7 +14,17 @@ Broadcast.BroadcastStyle(::Type{<:AMSA}) = AMSAStyle()
 
 @inline function Base.copy(bc::Broadcast.Broadcasted{<:AMSAStyle})
     first_amsa = find_amsa(bc)
-    out = similar(first_amsa)
+
+    out = similar(first_amsa,Base.Broadcast._broadcast_getindex_eltype(bc))
+
+    #=
+    ElType = Base.Broadcast.combine_eltypes(bc.f, bc.args)
+    if Base.isconcretetype(ElType)
+        # We can trust it and defer to the simpler `copyto!`
+        return copyto!(similar(bc, ElType), bc)
+    end
+    =#
+
     copyto!(out,bc)
     out
 end