(tvm-ffi) branch main updated: [EXTRA] Allow callers to disable unique in-place mutation (#776)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm-ffi.git


The following commit(s) were added to refs/heads/main by this push:
     new bb786912 [EXTRA] Allow callers to disable unique in-place mutation 
(#776)
bb786912 is described below

commit bb786912ad60af2335d82a60c3a7224a3eb0269a
Author: Tianqi Chen <tqc...@users.noreply.github.com>
AuthorDate: Fri Sep 11 12:57:40 2026 -0400

    [EXTRA] Allow callers to disable unique in-place mutation (#776)
    
    Native handlers can pass their in-place permission directly to
    MaybeInplaceMutateIfUniqueExpected. Add an optional allow_inplace flag,
    defaulting to true; false dispatches ordinary mutation without checking
    uniqueness.
---
 include/tvm/ffi/extra/structural_mutate.h | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/include/tvm/ffi/extra/structural_mutate.h 
b/include/tvm/ffi/extra/structural_mutate.h
index a50ea694..ce4b48d9 100644
--- a/include/tvm/ffi/extra/structural_mutate.h
+++ b/include/tvm/ffi/extra/structural_mutate.h
@@ -406,21 +406,24 @@ class StructuralMutatorObj : public Object {
   }
 
   /*!
-   * \brief Mutate a value, using in-place mutation only for a uniquely owned 
object.
+   * \brief Mutate a value, using in-place mutation only for a uniquely owned 
object
+   *        and \p allow_inplace set to true.
    * \tparam T The declared replacement type.
    * \param value The borrowed value to mutate.
+   * \param allow_inplace Whether in-place mutation is permitted. Defaults to 
true.
+   *        If false, use ordinary mutation without checking uniqueness.
    * \return The replacement or unchanged marker, or an Error if mutation 
failed.
    *
-   * \note The caller must already know the entire path from the root is 
uniquely
-   *       owned, either through an owning moved-in root or while handling a
-   *       ``__s_maybe_inplace_mutate__`` hook. This method checks only \p 
value
-   *       itself, not its ancestors.
+   * \note When \p allow_inplace is true, the caller must already know the 
entire
+   *       path from the root is uniquely owned, either through an owning 
moved-in
+   *       root or while handling a ``__s_maybe_inplace_mutate__`` hook. This 
method
+   *       checks only \p value itself, not its ancestors.
    */
   template <typename T = Any>
   TVM_FFI_INLINE Expected<UnchangedOr<T>> MaybeInplaceMutateIfUniqueExpected(
-      AnyView value) noexcept {
+      AnyView value, bool allow_inplace = true) noexcept {
     const Object* obj = value.as<Object>();
-    if (obj != nullptr && obj->unique()) {
+    if (allow_inplace && obj != nullptr && obj->unique()) {
       return MaybeInplaceMutateExpected<T>(value);
     }
     return MutateExpected<T>(value);