compiled

Author: masahi

python/tvm/meta_schedule/schedule_rule/__init__.py

@@ -22,7 +22,7 @@
 from .add_rfactor import AddRFactor
 from .auto_inline import AutoInline
 from .cross_thread_reduction import CrossThreadReduction
-from .multi_level_tiling import MultiLevelTiling, ReuseType
+from .multi_level_tiling import MultiLevelTiling, MultiLevelTilingVNNI, ReuseType
 from .parallel_vectorize_unroll import ParallelizeVectorizeUnroll
 from .random_compute_location import RandomComputeLocation
 from .schedule_rule import PyScheduleRule, ScheduleRule

src/meta_schedule/schedule_rule/multi_level_tiling.h

@@ -176,7 +176,7 @@ class MultiLevelTilingNode : public ScheduleRuleNode {
   }
 
   static constexpr const char* _type_key = "meta_schedule.MultiLevelTiling";
-  TVM_DECLARE_FINAL_OBJECT_INFO(MultiLevelTilingNode, ScheduleRuleNode);
+  TVM_DECLARE_BASE_OBJECT_INFO(MultiLevelTilingNode, ScheduleRuleNode);
 };
 
 }  // namespace meta_schedule

src/meta_schedule/schedule_rule/multi_level_tiling_vnni.cc

@@ -262,8 +262,10 @@ class MultiLevelTilingVNNINode : public MultiLevelTilingNode {
     return MultiLevelTilingNode::ApplySubRules(states);
   }
 
+public:
+
   static constexpr const char* _type_key = "meta_schedule.MultiLevelTilingVNNI";
-  // TVM_DECLARE_FINAL_OBJECT_INFO(MultiLevelTilingVNNINode, MultiLevelTilingNode);
+  TVM_DECLARE_FINAL_OBJECT_INFO(MultiLevelTilingVNNINode, MultiLevelTilingNode);
 };
 
 // Constructor