Browse Source

fix(mge/subgraph): fix mem leak

GitOrigin-RevId: dc7bfe6265
release-1.7
Megvii Engine Team chenyuanzhao 3 years ago
parent
commit
4f86b2c503
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/python/src/ops.cpp

+ 1
- 0
imperative/python/src/ops.cpp View File

@@ -450,6 +450,7 @@ void _init_py_op_def(py::module m) {
py_type.tp_richcompare = PyOp(OpDef)::tp_richcompare; py_type.tp_richcompare = PyOp(OpDef)::tp_richcompare;
py_type.tp_getset = py_op::py_getsetters; py_type.tp_getset = py_op::py_getsetters;
py_type.tp_repr = py_op::py_repr; py_type.tp_repr = py_op::py_repr;
py_type.tp_dealloc = py_dealloc_generic<PyOp(OpDef)>;
mgb_assert(PyType_Ready(&py_type) >= 0); mgb_assert(PyType_Ready(&py_type) >= 0);
m.add_object("OpDef", reinterpret_cast<PyObject*>(&py_type)); m.add_object("OpDef", reinterpret_cast<PyObject*>(&py_type));
} }


Loading…
Cancel
Save