fixes for DAEs and complex u0

SciML · Nov 29, 2024 · 29e2b1d · 29e2b1d
1 parent f9b5445
commit 29e2b1d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl
@@ -67,7 +67,8 @@ function DiffEqBase.prepare_alg(
     sparsity_detector = isnothing(sparsity) ? TracerSparsityDetector() : ADTypes.KnownJacobianSparsityDetector(sparsity)
     color_alg = DiffEqBase.has_colorvec(prob.f) ? ADTypes.ConstantColoringAlgorithm(sparsity, prob.f.colorvec) : GreedyColoringAlgorithm()
 
-    if !(u0 isa SVector)
+    # don't use sparsity if the vectors are SVectors, or the eltype of u0 is Complex
+    if !(u0 isa SVector) && !(eltype(u0) <: Complex)
         autodiff = AutoSparse(autodiff, sparsity_detector = sparsity_detector, coloring_algorithm = color_alg)
     end
 

diff --git a/lib/OrdinaryDiffEqDifferentiation/src/derivative_wrappers.jl b/lib/OrdinaryDiffEqDifferentiation/src/derivative_wrappers.jl
@@ -116,7 +116,10 @@ function build_jac_config(alg, f::F1, uf::F2, du1, uprev,
                 @. @view(jac_prototype[idxs]) = @view(f.mass_matrix[idxs])
             end
         end
-        uf = SciMLBase.@set uf.f = SciMLBase.unwrapped_f(uf.f)
+        if uf.f.f isa FunctionWrappersWrappers.FunctionWrappersWrapper 
+            uf = SciMLBase.@set uf.f = SciMLBase.unwrapped_f(uf.f)
+        end
+
         jac_config = DI.prepare_jacobian(uf, du1, alg_autodiff(alg), u)
     else 
         jac_config = nothing