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

test(mge/test): reorganize utils test files

GitOrigin-RevId: 08d27a8b1d18fa4d25b5c6f34486d1e8d4f0fae3
上级 53075cd3
......@@ -28,6 +28,7 @@ from megengine.functional import exp, log
from megengine.jit import exclude_from_trace, trace
from megengine.module import Module
from megengine.random import normal, uniform
from megengine.utils.naming import auto_naming
@pytest.mark.parametrize("trace_mode", [False, True])
......@@ -139,6 +140,8 @@ def test_dump():
def f(a, b):
return a + b
# prevent from remaining scope from exception test
auto_naming.clear()
a = tensor([2])
b = tensor([4])
y = f(a, b).numpy()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册