Support precompilation of callbacks

JuliaGameTheoreticPlanning · Apr 16, 2024 · 8794e95 · 8794e95
1 parent d3c2e97
commit 8794e95
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/parametric_problem.jl b/src/parametric_problem.jl
@@ -77,6 +77,7 @@ function ParametricMCP(
     lower_bounds::Vector,
     upper_bounds::Vector;
     compute_sensitivities = true,
+    compile_callbacks = true,
 ) where {T<:Union{FD.Node,Symbolics.Num}}
     problem_size = Internals.check_dimensions(f_symbolic, z_symbolic, lower_bounds, upper_bounds)
 
@@ -118,7 +119,7 @@ function ParametricMCP(
     end
 
     parameter_dimension = length(θ_symbolic)
-    ParametricMCP(
+    mcp = ParametricMCP(
         f!,
         jacobian_z!,
         jacobian_θ!,
@@ -127,6 +128,12 @@ function ParametricMCP(
         parameter_dimension,
         problem_size,
     )
+
+    if compile_callbacks
+        compile_callbacks(mcp)
+    end
+
+    mcp
 end
 
 """