Skip to content

Commit 1f62414

Browse files
feat: allow passing cachesyms to generate_update_A and generate_update_b
1 parent 2033f28 commit 1f62414

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/systems/codegen.jl

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

1195-
res = build_function_wrapper(sys, A, ps...; p_start = 1, expression = Val{true},
1195+
res = build_function_wrapper(
1196+
sys, A, ps..., cachesyms...; p_start = 1, expression = Val{true},
11961197
similarto = typeof(A), kwargs...)
11971198
return maybe_compile_function(expression, wrap_gfw, (1, 1, is_split(sys)), res;
11981199
eval_expression, eval_module)
@@ -1211,10 +1212,11 @@ $GENERATE_X_KWARGS
12111212
All other keyword arguments are forwarded to [`build_function_wrapper`](@ref).
12121213
"""
12131214
function generate_update_b(sys::System, b::AbstractVector; expression = Val{true},
1214-
wrap_gfw = Val{false}, eval_expression = false, eval_module = @__MODULE__, kwargs...)
1215+
wrap_gfw = Val{false}, eval_expression = false, eval_module = @__MODULE__, cachesyms = (), kwargs...)
12151216
ps = reorder_parameters(sys)
12161217

1217-
res = build_function_wrapper(sys, b, ps...; p_start = 1, expression = Val{true},
1218+
res = build_function_wrapper(
1219+
sys, b, ps..., cachesyms...; p_start = 1, expression = Val{true},
12181220
similarto = typeof(b), kwargs...)
12191221
return maybe_compile_function(expression, wrap_gfw, (1, 1, is_split(sys)), res;
12201222
eval_expression, eval_module)

0 commit comments

Comments
 (0)