@@ -156,13 +156,7 @@ def __init__(
156156 if register_meta :
157157 library .impl (name , _get_meta_impl (self ), "Meta" )
158158
159- if register_impl :
160- if dispatch_key is None :
161- dispatch_key = default_dispatch_keys ()
162- elif isinstance (dispatch_key , str ):
163- dispatch_key = [dispatch_key ]
164- for k in dispatch_key :
165- library .impl (name , _create_impl_trampoline (self ), k )
159+ # Note: Implementation registration removed as eager functionality is not used by tests
166160
167161 fq_name = f"{ library .ns } .{ name } "
168162 ALL_CUSTOM_OP_REGS [fq_name ] = self
@@ -992,29 +986,6 @@ def meta(*args):
992986 return meta
993987
994988
995- def _create_impl_trampoline (op : CustomOp ):
996- # Import lazily when an implementation trampoline is requested to avoid
997- # circular dependency between base objects and eager runtime goo.
998- from .eager import (
999- eager_dispatch ,
1000- )
1001-
1002- def handler (* args ):
1003- eager_override = op .eager_execute (* args )
1004- if eager_override is not NotImplemented :
1005- return eager_override
1006-
1007- ksel = EagerKernelSelection (op , args )
1008- op .select (ksel )
1009- if logger .isEnabledFor (logging .DEBUG ):
1010- logging .debug (
1011- "Dispatch on %s for specialization %s" , op .name , ksel .spec_key
1012- )
1013- return eager_dispatch (ksel )
1014-
1015- return handler
1016-
1017-
1018989def _define_signature_in_library (lib : torch .library .Library , signature : str ) -> str :
1019990 """Helper to define a schema in the library.
1020991
0 commit comments