From aaf62e47e64d592be770e915a7aa59d41eddb729 Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Tue, 11 Feb 2020 21:41:47 -0800
Subject: [PATCH] Fix optimize

---
 python/tvm/relay/build_module.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/relay/build_module.py b/python/tvm/relay/build_module.py
index fa812cb35703..8b347883fe3a 100644
--- a/python/tvm/relay/build_module.py
+++ b/python/tvm/relay/build_module.py
@@ -275,7 +275,7 @@ def optimize(mod, target=None, params=None):
     params : dict
         The parameters of the final graph.
     """
-    if isinstance(mod, _Module):
+    if isinstance(mod, IRModule):
         func = mod["main"]
     elif isinstance(mod, _expr.Function):
         func = mod