Working on Evo Search (apache#542)

zxybazh · Jan 7, 2022 · 309e75e · 309e75e
1 parent 096e093
commit 309e75e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/tvm/meta_schedule/__init__.py b/python/tvm/meta_schedule/__init__.py
@@ -36,4 +36,3 @@
 )
 from .tune import tune_te, tune_tir, tune_relay
 from .tune_context import TuneContext
-from .tune import tune_tir, tune_te
diff --git a/tests/python/unittest/test_meta_schedule_search_strategy.py b/tests/python/unittest/test_meta_schedule_search_strategy.py
@@ -269,4 +269,5 @@ def predict(
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    test_meta_schedule_evolutionary_search()
+    # sys.exit(pytest.main([__file__] + sys.argv[1:]))