Skip to content

Commit cf80a82

Browse files
authored
[Fix] Set DRefObj and CUDAIPCMemoryObj as mutable (#18294)
This PR marks `DRefObj` and `CUDAIPCMemoryObj` as a mutable object classes. The flags are missed during previous macro refactor.
1 parent abc8ae8 commit cf80a82

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

include/tvm/runtime/disco/cuda_ipc_memory.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,8 @@ class CUDAIPCMemoryObj : public Object {
6969
std::vector<void*> barrier_out;
7070
/*! \brief The integer buffer flag for all-reduce. */
7171
int barrier_flag;
72+
73+
static constexpr const bool _type_mutable = true;
7274
TVM_FFI_DECLARE_OBJECT_INFO("tvm.runtime.disco.cuda_ipc_memory", CUDAIPCMemoryObj, Object);
7375
};
7476

include/tvm/runtime/disco/session.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,7 @@ class DRefObj : public Object {
151151

152152
static constexpr const uint32_t _type_index = TypeIndex::kRuntimeDiscoDRef;
153153
static const constexpr bool _type_final = true;
154+
static constexpr const bool _type_mutable = true;
154155
TVM_FFI_DECLARE_OBJECT_INFO_STATIC("runtime.disco.DRef", DRefObj, Object);
155156

156157
/*! \brief The id of the register */

0 commit comments

Comments
 (0)