diff --git a/include/tvm/runtime/disco/cuda_ipc_memory.h b/include/tvm/runtime/disco/cuda_ipc_memory.h index e1cc74ddfe13..a6bfbd866b06 100644 --- a/include/tvm/runtime/disco/cuda_ipc_memory.h +++ b/include/tvm/runtime/disco/cuda_ipc_memory.h @@ -69,6 +69,8 @@ class CUDAIPCMemoryObj : public Object { std::vector barrier_out; /*! \brief The integer buffer flag for all-reduce. */ int barrier_flag; + + static constexpr const bool _type_mutable = true; TVM_FFI_DECLARE_OBJECT_INFO("tvm.runtime.disco.cuda_ipc_memory", CUDAIPCMemoryObj, Object); }; diff --git a/include/tvm/runtime/disco/session.h b/include/tvm/runtime/disco/session.h index 471c4567afca..283d75740c4f 100644 --- a/include/tvm/runtime/disco/session.h +++ b/include/tvm/runtime/disco/session.h @@ -151,6 +151,7 @@ class DRefObj : public Object { static constexpr const uint32_t _type_index = TypeIndex::kRuntimeDiscoDRef; static const constexpr bool _type_final = true; + static constexpr const bool _type_mutable = true; TVM_FFI_DECLARE_OBJECT_INFO_STATIC("runtime.disco.DRef", DRefObj, Object); /*! \brief The id of the register */