From 176268d20c1ce0a92a44027b3e6c42c9f5ecabfb Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 20 Nov 2020 21:35:13 +0800
Subject: [PATCH] test(mge): disable gc in leak check

GitOrigin-RevId: b650a46e02e82793cc668e22bc7bb6d9225e01f1
---
 imperative/python/test/unit/core/test_autodiff.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/imperative/python/test/unit/core/test_autodiff.py b/imperative/python/test/unit/core/test_autodiff.py
index 30a0cc23..5a6c3004 100644
--- a/imperative/python/test/unit/core/test_autodiff.py
+++ b/imperative/python/test/unit/core/test_autodiff.py
@@ -163,12 +163,16 @@ def test_release():
     def check(f):
         n = 0
         d = None
-        for i in range(3):
-            f()
-            m = len(gc.get_objects())
-            d = m - n
-            n = m
-        assert d == 0
+        gc.disable()
+        try:
+            for i in range(3):
+                f()
+                m = len(gc.get_objects())
+                d = m - n
+                n = m
+            assert d == 0
+        finally:
+            gc.enable()
 
     x = TensorWrapper([0.0])
     dy = TensorWrapper(np.ones_like(x.numpy()))