diff --git a/thunder/core/interpreter.py b/thunder/core/interpreter.py index 228e1a4b9f..235c6689ea 100644 --- a/thunder/core/interpreter.py +++ b/thunder/core/interpreter.py @@ -7041,7 +7041,7 @@ def _impl(fn, *args, **kwargs): if ( unbound_fn_candidate is not None and isinstance(unbound_fn_candidate, (WrapperDescriptorType, MethodDescriptorType)) - and unbound_fn_candidate.__get__(slf) == fn + and unbound_fn_candidate.__get__(slf, type(slf)) == fn ): unbound_fn = unbound_fn_candidate break