fix rosenbrock direct callout

SciML · Aug 7, 2024 · 40dc728 · 40dc728
1 parent 2e162ad
commit 40dc728
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/OrdinaryDiffEqCore/src/solve.jl b/lib/OrdinaryDiffEqCore/src/solve.jl
@@ -100,7 +100,7 @@ function DiffEqBase.__init(
 
     if alg isa OrdinaryDiffEqRosenbrockAdaptiveAlgorithm &&
        # https://github.com/SciML/OrdinaryDiffEq.jl/pull/2079 fixes this for Rosenbrock23 and 32
-       !(alg isa Union{Rosenbrock23, Rosenbrock32}) &&
+       !only_diagonal_mass_matrix(alg) &&
        prob.f.mass_matrix isa AbstractMatrix &&
        all(isequal(0), prob.f.mass_matrix)
         # technically this should also warn for zero operators but those are hard to check for