From ad65c9ba63855d8be038265665a9998ec8dd1f94 Mon Sep 17 00:00:00 2001
From: spectrometerHBH <spectrometer@sjtu.edu.cn>
Date: Sat, 2 Oct 2021 03:55:54 +0000
Subject: [PATCH] fix

---
 python/tvm/script/parser.py                        |  2 +-
 tests/python/unittest/test_meta_schedule_runner.py | 14 +++++++-------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/python/tvm/script/parser.py b/python/tvm/script/parser.py
index 53aa11f2548b..d5e79e8676c1 100644
--- a/python/tvm/script/parser.py
+++ b/python/tvm/script/parser.py
@@ -344,7 +344,7 @@ def B(...):
                 __tvm_meta__ = ...
 
             # returns an IRModule
-            mod = MyMod()
+            mod = MyMod
         """
         if len(node.funcs) == 1:
             return self.transform(next(iter(node.funcs.values())))
diff --git a/tests/python/unittest/test_meta_schedule_runner.py b/tests/python/unittest/test_meta_schedule_runner.py
index 8c1debb071c3..c2c180a201ac 100644
--- a/tests/python/unittest/test_meta_schedule_runner.py
+++ b/tests/python/unittest/test_meta_schedule_runner.py
@@ -177,7 +177,7 @@ def test_meta_schedule_rpc_single_run():
 def test_meta_schedule_local_single_run():
     """Test meta schedule local runner for a single run"""
     # Build the module
-    mod = MatmulModule()
+    mod = MatmulModule
     builder = LocalBuilder()
     (builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
     assert builder_result.artifact_path is not None
@@ -285,9 +285,9 @@ def test_meta_schedule_local_multiple_runs():
     """Test meta schedule local runner for multiple runs"""
     # Build the module
     mods = [
-        MatmulModule(),
-        MatmulReluModule(),
-        BatchMatmulModule(),
+        MatmulModule,
+        MatmulReluModule,
+        BatchMatmulModule,
     ]
     builder = LocalBuilder()
     builder_inputs = [BuilderInput(mod, Target("llvm")) for mod in mods]
@@ -408,7 +408,7 @@ def timeout_session_creator(  # pylint: disable=unused-variable
 
 def test_meta_schedule_local_runner_time_out():
     """Test meta schedule Local Runner time out"""
-    mod = MatmulModule()
+    mod = MatmulModule
     builder = LocalBuilder()
     (builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
     assert builder_result.artifact_path is not None
@@ -508,7 +508,7 @@ def exception_session_creator(  # pylint: disable=unused-variable
 
 def test_meta_schedule_local_runner_exception():
     """Test meta schedule Local Runner time out"""
-    mod = MatmulModule()
+    mod = MatmulModule
     builder = LocalBuilder()
     (builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
     assert builder_result.artifact_path is not None
@@ -833,7 +833,7 @@ def test_run_evaluator(
         return costs
 
     # Build the module
-    mod = AddModule()
+    mod = AddModule
     builder = LocalBuilder()
     (builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
     assert builder_result.artifact_path is not None