Fix after rebase, and reorganize the TVMScript folder structure.

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: Siyuan Feng <[email protected]>

include/tvm/relax/tuning_api.h

@@ -297,6 +297,13 @@ class TuningRecord : public runtime::ObjectRef {
   TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS(TuningRecord, runtime::ObjectRef, TuningRecordNode);
 };
 
+/*! \brief The equality check for Workload */
+struct WorkloadEqual {
+  bool operator()(const meta_schedule::Workload& a, const meta_schedule::Workload& b) const {
+    return a->shash == b->shash && tvm::StructuralEqual()(a->mod, b->mod);
+  }
+};
+
 /* \brief The abstract interface of database. */
 class DatabaseNode : public runtime::Object {
  public:

include/tvm/tir/analysis.h

@@ -260,10 +260,6 @@ const PrimFuncNode* FindEntryFunc(const IRModule& mod, GlobalVar* result_g_var);
  * \return The anchor block if found, nullptr otherwise.
  */
 const tir::BlockNode* FindAnchorBlock(const IRModule& mod);
- * \brief Annotate Op Pattern Kind for PrimFunc, which is used in relax FuseOps.
- * \param func The PrimFunc to be analyzed.
- * \return The Op Pattern Kind.
- */
 
 // Pass variants of verification analysis
 // directly throws RuntimeError when verification fails.

python/tvm/script/__init__.py

@@ -14,15 +14,15 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-"""TVM Script APIs of TVM Python Package, aimed to support TIR"""
-from . import _parser, parser_v1
+"""TVM Script APIs of TVM Python Package"""
+from . import parser, parser_v1
 
 #############
-from ._parser import ir
-from ._parser import ir_module
-from ._parser import parse as from_source_v2
-from ._parser import tir
-from ._parser import relax
+from .parser import ir
+from .parser import ir_module
+from .parser import parse as from_source_v2
+from .parser import tir
+from .parser import relax
 
 #############
 from .parser_v1 import from_source as from_source_v1