Merge pull request #2657 from jClugstor/bug_fix_2656

ChrisRackauckas · web-flow · commit f3d421860418 · 2025-04-02T14:00:03.000-04:00
Fix typo, add test for 2656
diff --git a/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl
@@ -122,7 +122,7 @@ function prepare_user_sparsity(ad_alg, prob)
 
                 if !isnothing(jac_prototype)
                     for idx in idxs
-                        jac_prototype[idx] = f.mass_matrix[idx]
+                        jac_prototype[idx] = prob.f.mass_matrix[idx]
                     end
                 end
             end
diff --git a/test/interface/sparsediff_tests.jl b/test/interface/sparsediff_tests.jl
@@ -64,6 +64,8 @@ for f in [f_oop, f_ip]
                     [
                         ODEFunction(f, colorvec = colors,
                             jac_prototype = jac_sp),
+                        ODEFunction(f, colorvec = colors,
+                            jac_prototype = jac_sp, mass_matrix = I(length(u0))),
                         ODEFunction(f, jac_prototype = jac_sp),
                         ODEFunction(f, colorvec = colors,
                             sparsity = jac_sp)

Original file line number	Diff line number	Diff line change
`@@ -64,6 +64,8 @@ for f in [f_oop, f_ip]`
`64`	`64`	`[`
`65`	`65`	`ODEFunction(f, colorvec = colors,`
`66`	`66`	`jac_prototype = jac_sp),`
	`67`	`+ ODEFunction(f, colorvec = colors,`
	`68`	`+ jac_prototype = jac_sp, mass_matrix = I(length(u0))),`
`67`	`69`	`ODEFunction(f, jac_prototype = jac_sp),`
`68`	`70`	`ODEFunction(f, colorvec = colors,`
`69`	`71`	`sparsity = jac_sp)`