User-Interface: Tune-TIR (apache#525)

* User-Interface: Tune-TIR * fix fix fix
zxybazh · Jan 7, 2022 · 28a8537 · 28a8537
1 parent 63bccca
commit 28a8537
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/python/tvm/meta_schedule/__init__.py b/python/tvm/meta_schedule/__init__.py
@@ -37,3 +37,5 @@
 from .tune import tune_te, tune_tir, tune_relay
 from .tune_context import TuneContext
 from .search_strategy import MeasureCandidate
+from .tune_context import TuneContext
+from .tune import tune_tir
diff --git a/src/meta_schedule/search_strategy/replay_func.cc b/src/meta_schedule/search_strategy/replay_func.cc
@@ -17,6 +17,7 @@
  * under the License.
  */
 #include "../utils.h"
+#include "tvm/meta_schedule/postproc.h"
 
 namespace tvm {
 namespace meta_schedule {

diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
@@ -258,6 +258,7 @@ TVM_REGISTER_TARGET_KIND("llvm", kDLCPU)
     .add_attr_option<Bool>("link-params", Bool(false))
     .add_attr_option<Bool>("unpacked-api")
     .add_attr_option<String>("interface-api")
+    .add_attr_option<Integer>("num-cores")
     // Fast math flags, see https://llvm.org/docs/LangRef.html#fast-math-flags
     .add_attr_option<Bool>("fast-math")  // implies all the below
     .add_attr_option<Bool>("fast-math-nnan")