added optimize kwarg to model_typed and model_warntype

Author: torfjelde

src/debug_utils.jl

@@ -680,34 +680,34 @@ function has_static_constraints(
 end
 
 """
-    model_warntype(model[, varinfo, context])
+    model_warntype(model[, varinfo, context]; optimize=true)
 
 Check the type stability of the model's evaluator, warning about any potential issues.
 
 This simply calls `@code_warntype` on the model's evaluator, filling in internal arguments where needed.
 """
-function model_warntype(model::Model, varinfo::AbstractVarInfo=VarInfo(model), context::AbstractContext=DefaultContext())
+function model_warntype(model::Model, varinfo::AbstractVarInfo=VarInfo(model), context::AbstractContext=DefaultContext(); optimize::Bool=true)
     args, kwargs = DynamicPPL.make_evaluate_args_and_kwargs(model, varinfo, context)
     return if isempty(kwargs)
-        InteractiveUtils.@code_warntype model.f(args...)
+        InteractiveUtils.@code_warntype optimize=optimize model.f(args...)
     else
-        InteractiveUtils.@code_warntype model.f(args...; kwargs...)
+        InteractiveUtils.@code_warntype optimize=optimize model.f(args...; kwargs...)
     end
 end
 
 """
-    model_typed(model[, varinfo, context])
+    model_typed(model[, varinfo, context]; optimize=true)
 
 Return the type inference for the model's evaluator.
 
 This simply calls `@code_typed` on the model's evaluator, filling in internal arguments where needed.
 """
-function model_typed(model::Model, varinfo::AbstractVarInfo=VarInfo(model), context::AbstractContext=DefaultContext())
+function model_typed(model::Model, varinfo::AbstractVarInfo=VarInfo(model), context::AbstractContext=DefaultContext(); optimize::Bool=true)
     args, kwargs = DynamicPPL.make_evaluate_args_and_kwargs(model, varinfo, context)
     return if isempty(kwargs)
-        InteractiveUtils.@code_typed model.f(args...)
+        InteractiveUtils.@code_typed optimize=optimize model.f(args...)
     else
-        InteractiveUtils.@code_typed model.f(args...; kwargs...)
+        InteractiveUtils.@code_typed optimize=optimize model.f(args...; kwargs...)
     end
 end