提交 0b88ec3c 编写于 作者: M Megvii Engine Team

refactor(mge/imperative): use apply.super in distributed

GitOrigin-RevId: e53276bd46ff52a6cde14db77d3069137db9c839
上级 33295900
...@@ -19,7 +19,7 @@ from ..core.autodiff.grad import ( ...@@ -19,7 +19,7 @@ from ..core.autodiff.grad import (
) )
from ..core.ops.builtin import CollectiveComm, Copy, RemoteRecv, RemoteSend from ..core.ops.builtin import CollectiveComm, Copy, RemoteRecv, RemoteSend
from ..core.tensor.core import apply from ..core.tensor.core import apply
from ..core.tensor.tensor import Tensor, tensor_apply from ..core.tensor.tensor import Tensor
from ..distributed.group import ( from ..distributed.group import (
WORLD, WORLD,
Group, Group,
...@@ -48,7 +48,7 @@ __all__ = [ ...@@ -48,7 +48,7 @@ __all__ = [
@apply.register() @apply.register()
def _(op: RemoteSend, *args: Tensor): def _(op: RemoteSend, *args: Tensor):
ret = tensor_apply(op, *args) ret = apply.super(op, *args)
# set extra information # set extra information
tracer_set = dict() tracer_set = dict()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册