User-Interface: Tune-TE (apache#527)

* fix a lot of issues * Add tune-te
zxybazh · Jan 7, 2022 · c521ce1 · c521ce1
1 parent 28a8537
commit c521ce1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/python/tvm/meta_schedule/__init__.py b/python/tvm/meta_schedule/__init__.py
@@ -38,4 +38,4 @@
 from .tune_context import TuneContext
 from .search_strategy import MeasureCandidate
 from .tune_context import TuneContext
-from .tune import tune_tir
+from .tune import tune_tir, tune_te
diff --git a/python/tvm/meta_schedule/tune.py b/python/tvm/meta_schedule/tune.py
@@ -70,6 +70,15 @@
     TaskScheduler,
 ]
 
+TYPE_F_TUNE_CONTEXT = Callable[  # pylint: disable=invalid-name
+    [
+        IRModule,
+        Target,
+        SearchStrategyConfig,
+        str,
+    ],
+    TuneContext,
+]
 
 class DefaultLLVM:
     """Default tuning configuration for LLVM."""

diff --git a/src/meta_schedule/search_strategy/replay_func.cc b/src/meta_schedule/search_strategy/replay_func.cc
@@ -17,7 +17,6 @@
  * under the License.
  */
 #include "../utils.h"
-#include "tvm/meta_schedule/postproc.h"
 
 namespace tvm {
 namespace meta_schedule {