Rebase.

include/tvm/meta_schedule/builder.h

@@ -1,4 +1,3 @@
-
 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file

python/tvm/meta_schedule/__init__.py

@@ -16,3 +16,4 @@
 # under the License.
 """Package `tvm.meta_schedule`. The meta schedule infrastructure."""
 from . import builder
+from . import tune_context

python/tvm/meta_schedule/tune_context.py

@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-"""The tune context of meta schedule."""
+"""Meta Schedule tuning context."""
 
 from typing import Optional
 
@@ -36,8 +36,8 @@ class TuneContext(Object):
     the same task can interact with each other through tune context. Most classes have a function
     to initialize with a tune context.
 
-    Members
-    -------
+    Parameters
+    ----------
     mod : Optional[IRModule] = None
         The workload to be optimized.
     target : Optional[Target] = None

tests/python/meta_schedule/test_meta_schedule_tune_context.py → tests/python/unittest/test_meta_schedule_tune_context.py

@@ -23,7 +23,7 @@
 from tvm import tir
 from tvm.script import ty
 from tvm.target import Target
-from tvm.meta_schedule import TuneContext
+from tvm.meta_schedule.tune_context import TuneContext
 
 # pylint: disable=invalid-name,no-member,line-too-long,too-many-nested-blocks,missing-docstring
 
@@ -44,7 +44,7 @@ def main(a: ty.handle, b: ty.handle, c: ty.handle) -> None:  # pylint: disable=n
 # pylint: enable=invalid-name,no-member,line-too-long,too-many-nested-blocks,missing-docstring
 
 
-def test_tune_context():
+def test_tune_context_create():
     mod = Matmul()
     context = TuneContext(mod, Target("llvm"), "Test Task")
     assert context.num_threads > 0