some more

Author: ChrisRackauckas

ext/DiffEqBaseForwardDiffExt.jl

@@ -6,7 +6,7 @@ using DiffEqBase: Void, FunctionWrappersWrappers, OrdinaryDiffEqTag, AbstractTim
     RecursiveArrayTools, reduce_tup, _promote_tspan, has_continuous_callback
 import DiffEqBase: hasdualpromote, wrapfun_oop, wrapfun_iip, prob2dtmin,
                    promote_tspan, anyeltypedual, isdualtype, value, ODE_DEFAULT_NORM,
-                   InternalITP, nextfloat_tdir, DualEltypeChecker
+                   InternalITP, nextfloat_tdir, DualEltypeChecker, sse
 
 eltypedual(x) = eltype(x) <: ForwardDiff.Dual
 isdualtype(::Type{<:ForwardDiff.Dual}) = true
@@ -500,7 +500,6 @@ value(x::ForwardDiff.Dual) = value(ForwardDiff.value(x))
 unitfulvalue(x::Type{ForwardDiff.Dual{T, V, N}}) where {T, V, N} = V
 unitfulvalue(x::ForwardDiff.Dual) = unitfulvalue(ForwardDiff.unitfulvalue(x))
 
-sse(x::Number) = abs2(x)
 sse(x::ForwardDiff.Dual) = sse(ForwardDiff.value(x)) + sum(sse, ForwardDiff.partials(x))
 totallength(x::Number) = 1
 function totallength(x::ForwardDiff.Dual)

src/utils.jl

@@ -2,6 +2,7 @@
 value(x) = x
 unitfulvalue(x) = x
 isdistribution(u0) = false
+sse(x::Number) = abs2(x)
 
 _vec(v) = vec(v)
 _vec(v::Number) = v