Skip to content

Commit 46b1867

Browse files
authored
[Wave] Remove unused code in iree/turbine/runtime (#41)
Signed-off-by: Harsh Menon <[email protected]>
1 parent a472d1a commit 46b1867

File tree

4 files changed

+1
-612
lines changed

4 files changed

+1
-612
lines changed

iree/turbine/runtime/op_reg/base.py

Lines changed: 1 addition & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -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-
1018989
def _define_signature_in_library(lib: torch.library.Library, signature: str) -> str:
1019990
"""Helper to define a schema in the library.
1020991

iree/turbine/runtime/op_reg/compiler.py

Lines changed: 0 additions & 220 deletions
This file was deleted.

0 commit comments

Comments
 (0)