From a207b27e8ec57d7eba98ba9b77fed5f9120932a7 Mon Sep 17 00:00:00 2001 From: William Moses Date: Thu, 28 Nov 2024 16:07:13 -0500 Subject: [PATCH] Speed up invoke (#2138) * Speed up invoke * handler specialize --- src/compiler/interpreter.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/compiler/interpreter.jl b/src/compiler/interpreter.jl index e1204a709f..1e442482be 100644 --- a/src/compiler/interpreter.jl +++ b/src/compiler/interpreter.jl @@ -759,13 +759,13 @@ end end function abstract_call_known( - @nospecialize(interp::EnzymeInterpreter), + interp::EnzymeInterpreter{Handler}, @nospecialize(f), arginfo::ArgInfo, si::StmtInfo, sv::AbsIntState, max_methods::Int = get_max_methods(interp, f, sv), -) +) where Handler (; fargs, argtypes) = arginfo @@ -889,7 +889,7 @@ function abstract_call_known( end return Base.@invoke abstract_call_known( interp::AbstractInterpreter, - f, + f::Any, arginfo::ArgInfo, si::StmtInfo, sv::AbsIntState,