Skip to content

Commit 92f0202

Browse files
committed
using class name instead of user provided name when redirect
Signed-off-by: Chendi.Xue <chendi.xue@intel.com>
1 parent e40b501 commit 92f0202

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm/model_executor/custom_op.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ class CustomOp(nn.Module):
1818

1919
def __new__(cls, *args, **kwargs):
2020
try:
21-
op_name = cls.name
21+
op_name = cls.__name__
2222
except AttributeError:
2323
raise TypeError(
2424
f"Cannot instantiate '{cls.__name__}': its 'name' attribute "
@@ -30,6 +30,8 @@ def __new__(cls, *args, **kwargs):
3030
op_cls_to_instantiate = cls
3131
else:
3232
op_cls_to_instantiate = cls.op_registry_oot[op_name]
33+
logger.debug("Instantiating custom op: %s using %s", op_name,
34+
str(op_cls_to_instantiate))
3335
return super().__new__(op_cls_to_instantiate)
3436

3537
def __init__(self):

0 commit comments

Comments
 (0)