apache · tqchen · Oct 16, 2025 · Oct 13, 2025 · Oct 13, 2025 · Oct 14, 2025
diff --git a/3rdparty/tvm-ffi b/3rdparty/tvm-ffi
diff --git a/include/tvm/ir/name_supply.h b/include/tvm/ir/name_supply.h
@@ -86,6 +86,12 @@ class NameSupplyNode : public Object {
   std::string prefix_;
 
   static constexpr const bool _type_mutable = true;
+
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<NameSupplyNode>();
+  }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.NameSupply", NameSupplyNode, Object);
 
  private:

diff --git a/include/tvm/meta_schedule/database.h b/include/tvm/meta_schedule/database.h
@@ -391,6 +391,8 @@ class PyDatabaseNode : public DatabaseNode {
     // `f_query_schedule` is not registered
     // `f_query_ir_module` is not registered
     // `f_size` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PyDatabaseNode>();
   }
 
   bool HasWorkload(const IRModule& mod) final {

diff --git a/include/tvm/meta_schedule/feature_extractor.h b/include/tvm/meta_schedule/feature_extractor.h
@@ -40,7 +40,8 @@ class FeatureExtractorNode : public runtime::Object {
   virtual ~FeatureExtractorNode() = default;
 
   static void RegisterReflection() {
-    // No fields to register
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<FeatureExtractorNode>();
   }
 
   /*!
@@ -79,6 +80,8 @@ class PyFeatureExtractorNode : public FeatureExtractorNode {
   static void RegisterReflection() {
     // `f_extract_from` is not registered
     // `f_as_string` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PyFeatureExtractorNode>();
   }
 
   ffi::Array<tvm::runtime::Tensor> ExtractFrom(

diff --git a/include/tvm/meta_schedule/measure_callback.h b/include/tvm/meta_schedule/measure_callback.h
@@ -43,7 +43,8 @@ class MeasureCallbackNode : public runtime::Object {
   virtual ~MeasureCallbackNode() = default;
 
   static void RegisterReflection() {
-    // No fields to register
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<MeasureCallbackNode>();
   }
 
   /*!
@@ -95,6 +96,8 @@ class PyMeasureCallbackNode : public MeasureCallbackNode {
   static void RegisterReflection() {
     // `f_apply` is not registered
     // `f_as_string` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PyMeasureCallbackNode>();
   }
 
   void Apply(const TaskScheduler& task_scheduler,                     //

diff --git a/include/tvm/meta_schedule/mutator.h b/include/tvm/meta_schedule/mutator.h
@@ -41,7 +41,8 @@ class MutatorNode : public runtime::Object {
   virtual ~MutatorNode() = default;
 
   static void RegisterReflection() {
-    // No fields to register
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<MutatorNode>();
   }
 
   /*!

diff --git a/include/tvm/meta_schedule/postproc.h b/include/tvm/meta_schedule/postproc.h
@@ -40,7 +40,8 @@ class PostprocNode : public runtime::Object {
   virtual ~PostprocNode() = default;
 
   static void RegisterReflection() {
-    // No fields to register
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PostprocNode>();
   }
 
   /*!
@@ -199,6 +200,8 @@ class PyPostprocNode : public PostprocNode {
     // `f_apply` is not registered
     // `f_clone` is not registered
     // `f_as_string` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PyPostprocNode>();
   }
 
   void InitializeWithTuneContext(const TuneContext& context) final;

diff --git a/include/tvm/meta_schedule/profiler.h b/include/tvm/meta_schedule/profiler.h
@@ -60,8 +60,8 @@ class ProfilerNode : public runtime::Object {
   ffi::Function total_timer;
 
   static void RegisterReflection() {
-    // `stats_sec` is not registered
-    // `total_timer` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<ProfilerNode>();
   }
 
   static constexpr const bool _type_mutable = true;

diff --git a/include/tvm/meta_schedule/runner.h b/include/tvm/meta_schedule/runner.h
@@ -128,6 +128,8 @@ class RunnerFutureNode : public runtime::Object {
   static void RegisterReflection() {
     // `f_done` is not registered
     // `f_result` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<RunnerFutureNode>();
   }
 
   /*!
@@ -189,6 +191,11 @@ class RunnerNode : public runtime::Object {
    */
   virtual ffi::Array<RunnerFuture> Run(ffi::Array<RunnerInput> runner_inputs) = 0;
 
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<RunnerNode>();
+  }
+
   static constexpr const bool _type_mutable = true;
   TVM_FFI_DECLARE_OBJECT_INFO("meta_schedule.Runner", RunnerNode, runtime::Object);
 };
@@ -222,6 +229,8 @@ class PyRunnerNode : public RunnerNode {
 
   static void RegisterReflection() {
     // `f_run` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PyRunnerNode>();
   }
 
   ffi::Array<RunnerFuture> Run(ffi::Array<RunnerInput> runner_inputs) final {

diff --git a/include/tvm/meta_schedule/schedule_rule.h b/include/tvm/meta_schedule/schedule_rule.h
@@ -43,7 +43,8 @@ class ScheduleRuleNode : public runtime::Object {
   virtual ~ScheduleRuleNode() = default;
 
   static void RegisterReflection() {
-    // No fields to register
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<ScheduleRuleNode>();
   }
 
   /*!
@@ -337,6 +338,8 @@ class PyScheduleRuleNode : public ScheduleRuleNode {
     // `f_apply` is not registered
     // `f_as_string` is not registered
     // `f_clone` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PyScheduleRuleNode>();
   }
 
   void InitializeWithTuneContext(const TuneContext& context) final;

diff --git a/include/tvm/meta_schedule/search_strategy.h b/include/tvm/meta_schedule/search_strategy.h
@@ -249,6 +249,8 @@ class PySearchStrategyNode : public SearchStrategyNode {
     // `f_generate_measure_candidates` is not registered
     // `f_notify_runner_results` is not registered
     // `f_clone` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PySearchStrategyNode>();
   }
 
   void InitializeWithTuneContext(const TuneContext& context) final;

diff --git a/include/tvm/meta_schedule/space_generator.h b/include/tvm/meta_schedule/space_generator.h
@@ -227,6 +227,8 @@ class PySpaceGeneratorNode : public SpaceGeneratorNode {
     // `f_initialize_with_tune_context` is not registered
     // `f_generate_design_space` is not registered
     // `f_clone` is not registered
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<PySpaceGeneratorNode>();
   }
 
   void InitializeWithTuneContext(const TuneContext& context) final;

diff --git a/include/tvm/relax/nested_msg.h b/include/tvm/relax/nested_msg.h
@@ -639,7 +639,7 @@ struct TypeTraits<relax::NestedMsg<T>> : public TypeTraitsBase {
   }
 
   TVM_FFI_INLINE static relax::NestedMsg<T> MoveFromAnyAfterCheck(TVMFFIAny* src) {
-    return relax::NestedMsg<T>(details::AnyUnsafe::MoveTVMFFIAnyToAny(std::move(*src)));
+    return relax::NestedMsg<T>(details::AnyUnsafe::MoveTVMFFIAnyToAny(src));
   }
 
   static std::optional<relax::NestedMsg<T>> TryCastFromAnyView(const TVMFFIAny* src) {
@@ -673,6 +673,14 @@ struct TypeTraits<relax::NestedMsg<T>> : public TypeTraitsBase {
   TVM_FFI_INLINE static std::string TypeStr() {
     return "NestedMsg<" + details::Type2Str<T>::v() + ">";
   }
+
+  TVM_FFI_INLINE static std::string TypeSchema() {
+    std::ostringstream oss;
+    oss << R"({"type":"NestedMsg","args":[)";
+    oss << details::TypeSchema<T>::v();
+    oss << "]}";
+    return oss.str();
+  }
 };
 }  // namespace ffi
 }  // namespace tvm

diff --git a/include/tvm/runtime/data_type.h b/include/tvm/runtime/data_type.h
@@ -497,6 +497,10 @@ struct TypeTraits<runtime::DataType> : public TypeTraitsBase {
   }
 
   TVM_FFI_INLINE static std::string TypeStr() { return ffi::StaticTypeKey::kTVMFFIDataType; }
+
+  TVM_FFI_INLINE static std::string TypeSchema() {
+    return R"({"type":")" + std::string(ffi::StaticTypeKey::kTVMFFIDataType) + R"("})";
+  }
 };
 
 }  // namespace ffi

diff --git a/include/tvm/runtime/tensor.h b/include/tvm/runtime/tensor.h
@@ -74,6 +74,7 @@ class Tensor : public tvm::ffi::Tensor {
   static Tensor FromDLPackVersioned(DLManagedTensorVersioned* tensor) {
     return tvm::ffi::Tensor::FromDLPackVersioned(tensor, kAllocAlignment, true);
   }
+  inline const DLTensor* operator->() const { return this->get(); }
   /*!
    * \brief Copy data content from another array.
    * \param other The source array to be copied from.

diff --git a/include/tvm/tir/block_scope.h b/include/tvm/tir/block_scope.h
@@ -264,7 +264,8 @@ class BlockScopeNode : public Object {
   std::unordered_map<Buffer, ffi::Array<StmtSRef>, ObjectPtrHash, ObjectPtrEqual> buffer_writers;
 
   static void RegisterReflection() {
-    // No fields to register as they are not visited
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<BlockScopeNode>();
   }
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tir.BlockScope", BlockScopeNode, Object);
 

diff --git a/include/tvm/tir/schedule/schedule.h b/include/tvm/tir/schedule/schedule.h
@@ -51,7 +51,8 @@ enum class BufferIndexType : int32_t {
 class BlockRVNode : public runtime::Object {
  public:
   static void RegisterReflection() {
-    // No fields to register as they are not visited
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<BlockRVNode>();
   }
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tir.BlockRV", BlockRVNode, runtime::Object);
 };
@@ -73,7 +74,8 @@ class BlockRV : public runtime::ObjectRef {
 class LoopRVNode : public runtime::Object {
  public:
   static void RegisterReflection() {
-    // No fields to register as they are not visited
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<LoopRVNode>();
   }
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tir.LoopRV", LoopRVNode, runtime::Object);
 };

diff --git a/python/tvm/testing/_ffi_api.py b/python/tvm/testing/_ffi_api.py
@@ -17,5 +17,8 @@
 """FFI APIs for tvm.testing"""
 import tvm_ffi
 
+# must import testing before init_ffi_api
+import tvm_ffi.testing
+
 
 tvm_ffi.init_ffi_api("testing", __name__)
diff --git a/src/arith/presburger_set.cc b/src/arith/presburger_set.cc
@@ -277,6 +277,7 @@ PresburgerSet MakePresburgerSet(const PrimExpr& constraint) { return PresburgerS
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  PresburgerSetNode::RegisterReflection();
   refl::GlobalDef().def("arith.PresburgerSet", MakePresburgerSet);
 }
 

diff --git a/src/ir/name_supply.cc b/src/ir/name_supply.cc
@@ -93,6 +93,7 @@ std::string NameSupplyNode::GetUniqueName(std::string name, bool add_underscore)
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  NameSupplyNode::RegisterReflection();
   refl::GlobalDef()
       .def("ir.NameSupply", [](ffi::String prefix) { return NameSupply(prefix); })
       .def_method("ir.NameSupply_FreshName", &NameSupplyNode::FreshName)

diff --git a/src/meta_schedule/measure_callback/add_to_database.cc b/src/meta_schedule/measure_callback/add_to_database.cc
@@ -56,6 +56,12 @@ class AddToDatabaseNode : public MeasureCallbackNode {
           /*args_info=*/candidate->args_info));
     }
   }
+
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<AddToDatabaseNode>();
+  }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.AddToDatabase", AddToDatabaseNode,
                                     MeasureCallbackNode);
 };
@@ -67,6 +73,7 @@ MeasureCallback MeasureCallback::AddToDatabase() {
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  refl::ObjectDef<AddToDatabaseNode>();
   refl::GlobalDef().def("meta_schedule.MeasureCallbackAddToDatabase",
                         MeasureCallback::AddToDatabase);
 }

diff --git a/src/meta_schedule/measure_callback/remove_build_artifact.cc b/src/meta_schedule/measure_callback/remove_build_artifact.cc
@@ -37,6 +37,12 @@ class RemoveBuildArtifactNode : public MeasureCallbackNode {
       }
     }
   }
+
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<RemoveBuildArtifactNode>();
+  }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.RemoveBuildArtifact", RemoveBuildArtifactNode,
                                     MeasureCallbackNode);
 };
@@ -48,6 +54,7 @@ MeasureCallback MeasureCallback::RemoveBuildArtifact() {
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  RemoveBuildArtifactNode::RegisterReflection();
   refl::GlobalDef().def("meta_schedule.MeasureCallbackRemoveBuildArtifact",
                         MeasureCallback::RemoveBuildArtifact);
 }

diff --git a/src/meta_schedule/measure_callback/update_cost_model.cc b/src/meta_schedule/measure_callback/update_cost_model.cc
@@ -54,6 +54,12 @@ class UpdateCostModelNode : public MeasureCallbackNode {
     }
     cost_model->Update(task->ctx, pruned_candidate, pruned_runner_result);
   }
+
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<UpdateCostModelNode>();
+  }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.UpdateCostModel", UpdateCostModelNode,
                                     MeasureCallbackNode);
 };
@@ -65,6 +71,7 @@ MeasureCallback MeasureCallback::UpdateCostModel() {
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  UpdateCostModelNode::RegisterReflection();
   refl::GlobalDef().def("meta_schedule.MeasureCallbackUpdateCostModel",
                         MeasureCallback::UpdateCostModel);
 }

diff --git a/src/meta_schedule/postproc/disallow_async_strided_mem_copy.cc b/src/meta_schedule/postproc/disallow_async_strided_mem_copy.cc
@@ -173,6 +173,12 @@ class DisallowAsyncStridedMemCopyNode : public PostprocNode {
         ffi::make_object<DisallowAsyncStridedMemCopyNode>(*this);
     return Postproc(n);
   }
+
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<DisallowAsyncStridedMemCopyNode>();
+  }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.DisallowAsyncStridedMemCopy",
                                     DisallowAsyncStridedMemCopyNode, PostprocNode);
 
@@ -188,6 +194,7 @@ Postproc Postproc::DisallowAsyncStridedMemCopy() {
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  DisallowAsyncStridedMemCopyNode::RegisterReflection();
   refl::GlobalDef().def("meta_schedule.PostprocDisallowAsyncStridedMemCopy",
                         Postproc::DisallowAsyncStridedMemCopy);
 }

diff --git a/src/meta_schedule/postproc/disallow_dynamic_loop.cc b/src/meta_schedule/postproc/disallow_dynamic_loop.cc
@@ -74,6 +74,12 @@ class DisallowDynamicLoopNode : public PostprocNode {
     ObjectPtr<DisallowDynamicLoopNode> n = ffi::make_object<DisallowDynamicLoopNode>(*this);
     return Postproc(n);
   }
+
+  static void RegisterReflection() {
+    namespace refl = tvm::ffi::reflection;
+    refl::ObjectDef<DisallowDynamicLoopNode>();
+  }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.DisallowDynamicLoop", DisallowDynamicLoopNode,
                                     PostprocNode);
 };
@@ -85,6 +91,7 @@ Postproc Postproc::DisallowDynamicLoop() {
 
 TVM_FFI_STATIC_INIT_BLOCK() {
   namespace refl = tvm::ffi::reflection;
+  DisallowDynamicLoopNode::RegisterReflection();
   refl::GlobalDef().def("meta_schedule.PostprocDisallowDynamicLoop", Postproc::DisallowDynamicLoop);
 }
 

diff --git a/src/meta_schedule/postproc/rewrite_cooperative_fetch.cc b/src/meta_schedule/postproc/rewrite_cooperative_fetch.cc
@@ -139,6 +139,7 @@ class RewriteCooperativeFetchNode : public PostprocNode {
     ObjectPtr<RewriteCooperativeFetchNode> n = ffi::make_object<RewriteCooperativeFetchNode>(*this);
     return Postproc(n);
   }
+
   TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.RewriteCooperativeFetch",
                                     RewriteCooperativeFetchNode, PostprocNode);