diff --git a/python/tvm/meta_schedule/__init__.py b/python/tvm/meta_schedule/__init__.py
index 4dc82b584860d..e5a7ea87975d7 100644
--- 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
index e6684d507f2ef..86a38a2192106 100644
--- 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
index c562c78bd1874..44eb0436edb79 100644
--- 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")