Skip to content

Commit ea52a85

Browse files
committed
Mark _evaluate!! as noinline
1 parent eed80e5 commit ea52a85

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

Diff for: src/model.jl

+7-1
Original file line numberDiff line numberDiff line change
@@ -931,7 +931,13 @@ end
931931
932932
Evaluate the `model` with the arguments matching the given `context` and `varinfo` object.
933933
"""
934-
function _evaluate!!(model::Model, varinfo::AbstractVarInfo, context::AbstractContext)
934+
@noinline function _evaluate!!(
935+
model::Model, varinfo::AbstractVarInfo, context::AbstractContext
936+
)
937+
# NOTE(penelopeysm): This @noinline is a workaround for
938+
# https://github.com/EnzymeAD/Enzyme.jl/issues/2337 and only really affects
939+
# the smallest models (e.g. those with a single parameter). This call can
940+
# be removed once that issue is fixed.
935941
args, kwargs = make_evaluate_args_and_kwargs(model, varinfo, context)
936942
return model.f(args...; kwargs...)
937943
end

0 commit comments

Comments
 (0)