addressing comments

Author: mkatanbaf

python/tvm/contrib/micro/meta_schedule/test_autotune_ms.py

@@ -76,15 +76,15 @@ def get_module():
 )
 def test_micro_tuning_with_meta_schedule(platform, options):
     if platform == "crt":
-        target = tvm.target.target.micro(model="host", options="-num-cores=1")
+        target = tvm.target.target.micro(model="host")
     else:
         boards_file = (
             pathlib.Path(tvm.micro.get_microtvm_template_projects("zephyr")) / "boards.json"
         )
         with open(boards_file) as f:
             boards = json.load(f)
         target = tvm.target.target.micro(
-            model=boards[options["board"]]["model"], options="-mcpu=cortex-m4 -num-cores=1"
+            model=boards[options["board"]]["model"], options="-mcpu=cortex-m4"
         )
 
     work_dir = utils.tempdir()
@@ -156,7 +156,7 @@ def test_micro_tuning_with_meta_schedule(platform, options):
 
     # Build reference model (without tuning)
     dev = tvm.cpu()
-    target = tvm.target.target.micro(model="host", options="-num-cores=1")
+    target = tvm.target.target.micro(model="host")
     with tvm.transform.PassContext(
         opt_level=3, config={"tir.disable_vectorize": True}, disabled_pass=["AlterOpLayout"]
     ):

python/tvm/rpc/server.py

@@ -319,7 +319,7 @@ def __init__(
         load_library=None,
         custom_addr=None,
         silent=False,
-        reuse_addr=False,
+        reuse_addr=True,
         timeout=None,
     ):
 
@@ -377,7 +377,7 @@ def _popen_start_rpc_server(
     silent=False,
     no_fork=False,
     server_init_callback=None,
-    reuse_addr=False,
+    reuse_addr=True,
     timeout=None,
 ):
     if no_fork:
@@ -488,7 +488,7 @@ def __init__(
         silent=False,
         no_fork=False,
         server_init_callback=None,
-        reuse_addr=False,
+        reuse_addr=True,
         timeout=None,
     ):
         try:

python/tvm/rpc/tracker.py

@@ -387,9 +387,7 @@ class PopenTrackerServerState(object):
 
     current = None
 
-    def __init__(
-        self, host, port=9190, port_end=9199, silent=False, reuse_addr=False, timeout=None
-    ):
+    def __init__(self, host, port=9190, port_end=9199, silent=False, reuse_addr=True, timeout=None):
         if silent:
             logger.setLevel(logging.WARN)
 
@@ -419,7 +417,7 @@ def __init__(
 
 
 def _popen_start_tracker_server(
-    host, port=9190, port_end=9199, silent=False, reuse_addr=False, timeout=None
+    host, port=9190, port_end=9199, silent=False, reuse_addr=True, timeout=None
 ):
     # This is a function that will be sent to the
     # Popen worker to run on a separate process.
@@ -458,7 +456,7 @@ class Tracker(object):
     """
 
     def __init__(
-        self, host="0.0.0.0", port=9190, port_end=9199, silent=False, reuse_addr=False, timeout=None
+        self, host="0.0.0.0", port=9190, port_end=9199, silent=False, reuse_addr=True, timeout=None
     ):
         if silent:
             logger.setLevel(logging.WARN)

src/meta_schedule/mutator/mutator.cc

@@ -82,8 +82,7 @@ Map<Mutator, FloatImm> Mutator::DefaultMicro() {
   return Map<Mutator, FloatImm>{
       {Mutator::MutateTileSize(), FloatImm(DataType::Float(64), 0.9)},
       {Mutator::MutateComputeLocation(), FloatImm(DataType::Float(64), 0.05)},
-      {Mutator::MutateUnroll(), FloatImm(DataType::Float(64), 0.03)},
-      {Mutator::MutateParallel(/*max_jobs_per_core=*/16), FloatImm(DataType::Float(64), 0.02)}};
+      {Mutator::MutateUnroll(), FloatImm(DataType::Float(64), 0.03)}};
 }
 
 TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)

src/meta_schedule/schedule_rule/schedule_rule.cc

@@ -271,11 +271,6 @@ Array<ScheduleRule> ScheduleRule::DefaultMicro() {
           Map<String, ObjectRef>{{"req", String("may")},
                                  {"levels", Array<Integer>{1, 2}},
                                  {"scope", String("global")}}),
-      ScheduleRule::ParallelizeVectorizeUnroll(
-          /*max_jobs_per_core=*/16,
-          /*max_vectorize_extent=*/1,
-          /*unroll_max_steps=*/Array<Integer>{0, 16, 64, 512},
-          /*unroll_explicit=*/true),
   };
 }
 

src/target/target_kind.cc

@@ -309,7 +309,6 @@ TVM_REGISTER_TARGET_KIND("c", kDLCPU)
     .add_attr_option<String>("march")
     .add_attr_option<Integer>("workspace-byte-alignment")
     .add_attr_option<Integer>("constants-byte-alignment")
-    .add_attr_option<Integer>("num-cores")
     .set_default_keys({"cpu"})
     .set_target_parser(tvm::target::parsers::cpu::ParseTarget);