From 0b88ec3cffda14ec206355e273c275fe7d24dc7e Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 24 Aug 2020 17:48:16 +0800
Subject: [PATCH] refactor(mge/imperative): use apply.super in distributed

GitOrigin-RevId: e53276bd46ff52a6cde14db77d3069137db9c839
---
 imperative/python/megengine/functional/distributed.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/imperative/python/megengine/functional/distributed.py b/imperative/python/megengine/functional/distributed.py
index c52705716..71fafd5c0 100644
--- a/imperative/python/megengine/functional/distributed.py
+++ b/imperative/python/megengine/functional/distributed.py
@@ -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()
-- 
GitLab