Prelude import altered

apache · Jan 4, 2021 · 2948252 · 2948252
1 parent ae3f3b1
commit 2948252
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/tvm/relay/testing/__init__.py b/python/tvm/relay/testing/__init__.py
@@ -24,6 +24,7 @@
 from tvm import te
 from tvm import relay
 from tvm.relay import op
+from tvm.relay.prelude import Prelude
 from tvm.testing import enabled_targets
 
 from . import mlp
@@ -51,8 +52,6 @@ def run_opt_pass(expr, opt_pass, import_prelude=False):
     assert isinstance(opt_pass, tvm.transform.Pass)
     mod = tvm.IRModule.from_expr(expr)
     if import_prelude:
-        from tvm.relay import Prelude
-
         Prelude(mod)
     mod = relay.transform.InferType()(mod)
     mod = opt_pass(mod)