From 510031766046297398dd48ee0e4a3e28bf566596 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 18 Jan 2021 17:04:11 +0800
Subject: [PATCH] feat(mge): make Function available at autodiff.Function

GitOrigin-RevId: 7c9ab65351c2967520c40b5d1c531dd65c7a1008
---
 imperative/python/megengine/autodiff/__init__.py                | 1 +
 imperative/python/megengine/quantization/fake_quant.py          | 2 +-
 imperative/python/megengine/quantization/internal_fake_quant.py | 2 +-
 imperative/python/megengine/quantization/utils.py               | 2 +-
 imperative/python/test/unit/core/test_function.py               | 2 +-
 5 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/imperative/python/megengine/autodiff/__init__.py b/imperative/python/megengine/autodiff/__init__.py
index 3def6f814..10553b392 100644
--- a/imperative/python/megengine/autodiff/__init__.py
+++ b/imperative/python/megengine/autodiff/__init__.py
@@ -6,4 +6,5 @@
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+from ..core.autodiff.grad import Function
 from .grad_manager import GradManager
diff --git a/imperative/python/megengine/quantization/fake_quant.py b/imperative/python/megengine/quantization/fake_quant.py
index 66bc56b6d..6b035d531 100644
--- a/imperative/python/megengine/quantization/fake_quant.py
+++ b/imperative/python/megengine/quantization/fake_quant.py
@@ -11,7 +11,7 @@ from typing import Iterable
 import numpy as np
 
 from .. import functional as F
-from ..core.autodiff.grad import Function
+from ..autodiff import Function
 from ..core.tensor.dtype import _metadata_dict, get_quantized_dtype
 from ..module import Module
 from ..tensor import Parameter, Tensor
diff --git a/imperative/python/megengine/quantization/internal_fake_quant.py b/imperative/python/megengine/quantization/internal_fake_quant.py
index ab980406e..1f246f482 100644
--- a/imperative/python/megengine/quantization/internal_fake_quant.py
+++ b/imperative/python/megengine/quantization/internal_fake_quant.py
@@ -12,7 +12,7 @@ from functools import partial
 import numpy as np
 
 from .. import functional as F
-from ..core.autodiff.grad import Function
+from ..autodiff import Function
 from .fake_quant import _FakeQuantize
 from .observer import MinMaxObserver
 from .qconfig import QConfig
diff --git a/imperative/python/megengine/quantization/utils.py b/imperative/python/megengine/quantization/utils.py
index 3c297cf7f..7f36c25b3 100644
--- a/imperative/python/megengine/quantization/utils.py
+++ b/imperative/python/megengine/quantization/utils.py
@@ -12,8 +12,8 @@ from typing import Dict
 import numpy as np
 
 from .. import functional as F
+from ..autodiff import Function
 from ..core._imperative_rt.core2 import apply
-from ..core.autodiff.grad import Function
 from ..core.ops import builtin
 from ..core.tensor import megbrain_graph
 from ..core.tensor.dtype import _metadata_dict
diff --git a/imperative/python/test/unit/core/test_function.py b/imperative/python/test/unit/core/test_function.py
index accc75106..de4094323 100644
--- a/imperative/python/test/unit/core/test_function.py
+++ b/imperative/python/test/unit/core/test_function.py
@@ -15,7 +15,7 @@ import megengine.optimizer as optimizer
 from megengine import Parameter
 from megengine import Tensor as tensor
 from megengine import tensor
-from megengine.core.autodiff.grad import Function
+from megengine.autodiff import Function
 from megengine.module import Module
 
 
-- 
GitLab