handle SciMLOperators

SciML · Dec 10, 2023 · 6ec50f5 · 6ec50f5
1 parent ce6980f
commit 6ec50f5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/misc_utils.jl b/src/misc_utils.jl
@@ -186,9 +186,11 @@ function get_differential_vars(f, u)
     differential_vars = nothing
     if hasproperty(f, :mass_matrix)
         mm = f.mass_matrix
+        mm = mm isa MatrixOperator ? mm.A : mm
+
         if mm isa UniformScaling
             return nothing
-        elseif isdiag(mm)
+        elseif !(mm isa SciMLOperators.AbstractSciMLOperators) && isdiag(mm)
             differential_vars = reshape(diag(mm)  .!= 0, size(u))
         else
             return DifferentialVarsUndefined()