Skip to content

Commit b1a4b36

Browse files
fix: fix OptimizationProblem codegen with CSE
1 parent d45b2db commit b1a4b36

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/systems/optimization/optimizationsystem.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
354354

355355
f = let _f = eval_or_rgf(
356356
generate_function(
357-
sys, checkbounds = checkbounds, linenumbers = linenumbers,
357+
sys; checkbounds = checkbounds, linenumbers = linenumbers,
358358
expression = Val{true}, wrap_mtkparameters = false, cse);
359359
eval_expression,
360360
eval_module)
@@ -367,7 +367,7 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
367367
if grad
368368
_grad = let (grad_oop, grad_iip) = eval_or_rgf.(
369369
generate_gradient(
370-
sys, checkbounds = checkbounds,
370+
sys; checkbounds = checkbounds,
371371
linenumbers = linenumbers,
372372
parallel = parallel, expression = Val{true},
373373
wrap_mtkparameters = false, cse);
@@ -386,7 +386,7 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
386386
if hess
387387
_hess = let (hess_oop, hess_iip) = eval_or_rgf.(
388388
generate_hessian(
389-
sys, checkbounds = checkbounds,
389+
sys; checkbounds = checkbounds,
390390
linenumbers = linenumbers,
391391
sparse = sparse, parallel = parallel,
392392
expression = Val{true}, wrap_mtkparameters = false, cse);
@@ -413,9 +413,9 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
413413
if length(cstr) > 0
414414
@named cons_sys = ConstraintsSystem(cstr, dvs, ps; checks)
415415
cons_sys = complete(cons_sys)
416-
cons, lcons_, ucons_ = generate_function(cons_sys, checkbounds = checkbounds,
416+
cons, lcons_, ucons_ = generate_function(cons_sys; checkbounds = checkbounds,
417417
linenumbers = linenumbers,
418-
expression = Val{true}; wrap_mtkparameters = false, cse)
418+
expression = Val{true}, wrap_mtkparameters = false, cse)
419419
cons = let (cons_oop, cons_iip) = eval_or_rgf.(cons; eval_expression, eval_module)
420420
_cons(u, p) = cons_oop(u, p)
421421
_cons(resid, u, p) = cons_iip(resid, u, p)
@@ -443,7 +443,7 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
443443
if cons_h
444444
_cons_h = let (cons_hess_oop, cons_hess_iip) = eval_or_rgf.(
445445
generate_hessian(
446-
cons_sys, checkbounds = checkbounds,
446+
cons_sys; checkbounds = checkbounds,
447447
linenumbers = linenumbers,
448448
sparse = cons_sparse, parallel = parallel,
449449
expression = Val{true}, wrap_mtkparameters = false, cse);

0 commit comments

Comments
 (0)