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