Skip to content

Commit 12c2bab

Browse files
feat: allow passing cachesyms to generate_update_A and generate_update_b
1 parent a0ce384 commit 12c2bab

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/systems/codegen.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1187,10 +1187,10 @@ $GENERATE_X_KWARGS
11871187
All other keyword arguments are forwarded to [`build_function_wrapper`](@ref).
11881188
"""
11891189
function generate_update_A(sys::System, A::AbstractMatrix; expression = Val{true},
1190-
wrap_gfw = Val{false}, eval_expression = false, eval_module = @__MODULE__, kwargs...)
1190+
wrap_gfw = Val{false}, eval_expression = false, eval_module = @__MODULE__, cachesyms = (), kwargs...)
11911191
ps = reorder_parameters(sys)
11921192

1193-
res = build_function_wrapper(sys, A, ps...; p_start = 1, expression = Val{true},
1193+
res = build_function_wrapper(sys, A, ps..., cachesyms...; p_start = 1, expression = Val{true},
11941194
similarto = typeof(A), kwargs...)
11951195
return maybe_compile_function(expression, wrap_gfw, (1, 1, is_split(sys)), res;
11961196
eval_expression, eval_module)
@@ -1209,10 +1209,10 @@ $GENERATE_X_KWARGS
12091209
All other keyword arguments are forwarded to [`build_function_wrapper`](@ref).
12101210
"""
12111211
function generate_update_b(sys::System, b::AbstractVector; expression = Val{true},
1212-
wrap_gfw = Val{false}, eval_expression = false, eval_module = @__MODULE__, kwargs...)
1212+
wrap_gfw = Val{false}, eval_expression = false, eval_module = @__MODULE__, cachesyms = (), kwargs...)
12131213
ps = reorder_parameters(sys)
12141214

1215-
res = build_function_wrapper(sys, b, ps...; p_start = 1, expression = Val{true},
1215+
res = build_function_wrapper(sys, b, ps..., cachesyms...; p_start = 1, expression = Val{true},
12161216
similarto = typeof(b), kwargs...)
12171217
return maybe_compile_function(expression, wrap_gfw, (1, 1, is_split(sys)), res;
12181218
eval_expression, eval_module)

0 commit comments

Comments
 (0)