Nits.

python/tvm/meta_schedule/apply_history_best.py

@@ -25,6 +25,7 @@
 
 from . import _ffi_api
 from .database import Database
+from .utils import make_logging_func
 
 logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
 
@@ -43,7 +44,7 @@ class ApplyHistoryBest(Object):
 
     def __init__(self, database: Database) -> None:
         self.__init_handle_by_constructor__(
-            _ffi_api.ApplyHistoryBest, database, logger  # type: ignore # pylint: disable=no-member
+            _ffi_api.ApplyHistoryBest, database, make_logging_func(logger)  # type: ignore # pylint: disable=no-member
         )
 
     def query(

python/tvm/meta_schedule/runner/rpc_runner.py

@@ -39,9 +39,9 @@
     run_evaluator_common,
 )
 
-
 logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
 
+
 T_CREATE_SESSION = Callable[  # pylint: disable=invalid-name
     [RPCConfig],  # The RPC configuration
     RPCSession,  # The RPC Session

python/tvm/meta_schedule/task_scheduler/task_scheduler.py

@@ -32,6 +32,9 @@
 from ..tune_context import TuneContext
 
 
+logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
+
+
 @register_object("meta_schedule.TaskScheduler")
 class TaskScheduler(Object):
     """The abstract task scheduler interface.
@@ -52,8 +55,6 @@ class TaskScheduler(Object):
         The cost model used for search.
     measure_callbacks: List[MeasureCallback] = None
         The list of measure callbacks of the scheduler.
-    logger: Optional[logging.Logger]
-        The logger of the task scheduler.
     num_trials_already : int
         The number of trials already conducted.
     """
@@ -66,7 +67,6 @@ class TaskScheduler(Object):
     cost_model: Optional[CostModel]
     measure_callbacks: List[MeasureCallback]
     num_trials_already: int
-    logger: Optional[logging.Logger]
 
     def tune(self) -> None:
         """Auto-tuning."""
@@ -136,7 +136,6 @@ def __init__(
         max_trials: int,
         cost_model: Optional[CostModel] = None,
         measure_callbacks: Optional[List[MeasureCallback]] = None,
-        logger: Optional[logging.Logger] = None,
         f_tune: Callable = None,
         f_initialize_task: Callable = None,
         f_touch_task: Callable = None,
@@ -181,7 +180,6 @@ class PyTaskScheduler:
             "max_trials",
             "cost_model",
             "measure_callbacks",
-            "logger",
         ],
         "methods": [
             "tune",
@@ -201,7 +199,6 @@ def __init__(
         max_trials: int,
         cost_model: Optional[CostModel] = None,
         measure_callbacks: Optional[List[MeasureCallback]] = None,
-        logger: Optional[logging.Logger] = None,
     ):
         self.tasks = tasks
         self.builder = builder
@@ -210,7 +207,6 @@ def __init__(
         self.max_trials = max_trials
         self.cost_model = cost_model
         self.measure_callbacks = measure_callbacks
-        self.logger = logger
 
     def tune(self) -> None:
         """Auto-tuning."""

python/tvm/meta_schedule/testing/relay_workload.py

@@ -30,7 +30,6 @@
 from tvm.runtime import NDArray, load_param_dict, save_param_dict
 from tvm.target import Target
 
-
 logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
 
 

python/tvm/meta_schedule/tune.py

@@ -46,14 +46,13 @@
 from .tune_context import TuneContext
 from .utils import autotvm_silencer
 
+logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
 
 FnSpaceGenerator = Callable[[], SpaceGenerator]
 FnScheduleRule = Callable[[], List[ScheduleRule]]
 FnPostproc = Callable[[], List[Postproc]]
 FnMutatorProb = Callable[[], Dict[Mutator, float]]
 
-logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
-
 
 class DefaultLLVM:
     """Default tuning configuration for LLVM."""