Incorrect prototype was being passed to the NonlinearFunction

Author: avik-pal

ext/BoundaryValueDiffEqOrdinaryDiffEqExt.jl

@@ -108,11 +108,10 @@ end
     if @load_preference("PrecompileShootingNLLS", VERSION≥v"1.10-")
         append!(algs,
             [
-                Shooting(Tsit5();
-                    nlsolve = LevenbergMarquardt(;
-                        autodiff = AutoForwardDiff(chunksize = 2))),
-                Shooting(Tsit5();
-                    nlsolve = GaussNewton(; autodiff = AutoForwardDiff(chunksize = 2))),
+                Shooting(Tsit5(); nlsolve = LevenbergMarquardt(),
+                    jac_alg = BVPJacobianAlgorithm(AutoForwardDiff(; chunksize = 2))),
+                Shooting(Tsit5(); nlsolve = GaussNewton(),
+                    jac_alg = BVPJacobianAlgorithm(AutoForwardDiff(; chunksize = 2))),
             ])
     end
 

src/solve/single_shooting.jl

@@ -58,8 +58,7 @@ function __solve(prob::BVProblem, alg_::Shooting; odesolve_kwargs = (;),
             alg.jac_alg.diffmode, loss_fnₚ)
     end
 
-    nlf = NonlinearFunction{iip}(loss_fn; prob.f.jac_prototype, resid_prototype,
-        jac = jac_fn)
+    nlf = NonlinearFunction{iip}(loss_fn; jac_prototype, resid_prototype, jac = jac_fn)
     nlprob = if length(resid_prototype) == length(u0)
         NonlinearProblem(nlf, vec(u0), prob.p)
     else