// clang-format off template<> struct EnumTrait { static constexpr const char *name = "AdaptivePooling.Mode"; static constexpr std::underlying_type_t max = 3 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"MAX", "AVERAGE", "AVERAGE_COUNT_EXCLUDE_PADDING"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("MAX"), AdaptivePooling::Mode::MAX}, {normalize_enum("AVERAGE"), AdaptivePooling::Mode::AVERAGE}, {normalize_enum("AVERAGE_COUNT_EXCLUDE_PADDING"), AdaptivePooling::Mode::AVERAGE_COUNT_EXCLUDE_PADDING}}; template<> PyObject* EnumWrapper::pyobj_insts[3] = {nullptr}; void _init_py_AdaptivePooling_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.AdaptivePooling.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("AdaptivePooling.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Mode::MAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MAX", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Mode::AVERAGE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "AVERAGE", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Mode::AVERAGE_COUNT_EXCLUDE_PADDING; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "AVERAGE_COUNT_EXCLUDE_PADDING", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "AdaptivePooling.Format"; static constexpr std::underlying_type_t max = 18 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"NCHW", "NHWC", "NHWCD4", "NCHW4", "NCHW8", "NCHW32", "NCHW88", "NCHW44", "NCHW44_DOT", "NCHW4_NCHW32", "NCHW32_NCHW4", "NCHW4_NCHW", "NHWC_NCHW", "NHWC_NCHW4_IC_SMALL", "NCHW_NCHW4_IC_SMALL", "CHWN4", "NCHW64", "NCHW4_NHWC"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("NCHW"), AdaptivePooling::Format::NCHW}, {normalize_enum("NHWC"), AdaptivePooling::Format::NHWC}, {normalize_enum("NHWCD4"), AdaptivePooling::Format::NHWCD4}, {normalize_enum("NCHW4"), AdaptivePooling::Format::NCHW4}, {normalize_enum("NCHW8"), AdaptivePooling::Format::NCHW8}, {normalize_enum("NCHW32"), AdaptivePooling::Format::NCHW32}, {normalize_enum("NCHW88"), AdaptivePooling::Format::NCHW88}, {normalize_enum("NCHW44"), AdaptivePooling::Format::NCHW44}, {normalize_enum("NCHW44_DOT"), AdaptivePooling::Format::NCHW44_DOT}, {normalize_enum("NCHW4_NCHW32"), AdaptivePooling::Format::NCHW4_NCHW32}, {normalize_enum("NCHW32_NCHW4"), AdaptivePooling::Format::NCHW32_NCHW4}, {normalize_enum("NCHW4_NCHW"), AdaptivePooling::Format::NCHW4_NCHW}, {normalize_enum("NHWC_NCHW"), AdaptivePooling::Format::NHWC_NCHW}, {normalize_enum("NHWC_NCHW4_IC_SMALL"), AdaptivePooling::Format::NHWC_NCHW4_IC_SMALL}, {normalize_enum("NCHW_NCHW4_IC_SMALL"), AdaptivePooling::Format::NCHW_NCHW4_IC_SMALL}, {normalize_enum("CHWN4"), AdaptivePooling::Format::CHWN4}, {normalize_enum("NCHW64"), AdaptivePooling::Format::NCHW64}, {normalize_enum("NCHW4_NHWC"), AdaptivePooling::Format::NCHW4_NHWC}}; template<> PyObject* EnumWrapper::pyobj_insts[18] = {nullptr}; void _init_py_AdaptivePooling_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.AdaptivePooling.Format", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Format").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("AdaptivePooling.Format").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NHWC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWC", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NHWCD4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWCD4", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW8", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW32", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW88; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW88", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW44; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW44", inst) >= 0); EnumWrapper::pyobj_insts[7] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW44_DOT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW44_DOT", inst) >= 0); EnumWrapper::pyobj_insts[8] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW4_NCHW32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4_NCHW32", inst) >= 0); EnumWrapper::pyobj_insts[9] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW32_NCHW4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW32_NCHW4", inst) >= 0); EnumWrapper::pyobj_insts[10] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW4_NCHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4_NCHW", inst) >= 0); EnumWrapper::pyobj_insts[11] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NHWC_NCHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWC_NCHW", inst) >= 0); EnumWrapper::pyobj_insts[12] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NHWC_NCHW4_IC_SMALL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWC_NCHW4_IC_SMALL", inst) >= 0); EnumWrapper::pyobj_insts[13] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW_NCHW4_IC_SMALL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW_NCHW4_IC_SMALL", inst) >= 0); EnumWrapper::pyobj_insts[14] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::CHWN4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CHWN4", inst) >= 0); EnumWrapper::pyobj_insts[15] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW64; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW64", inst) >= 0); EnumWrapper::pyobj_insts[16] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = AdaptivePooling::Format::NCHW4_NHWC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4_NHWC", inst) >= 0); EnumWrapper::pyobj_insts[17] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(AdaptivePooling) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"format", serialization::dump(opdef.format)}, {"shape", serialization::dump(opdef.shape)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("shape"); if (iter != state.end()) { opdef.shape = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(AdaptivePooling) int PyOp(AdaptivePooling)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "format", "shape", "scope", NULL}; PyObject *mode = NULL, *format = NULL, *shape = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOO", const_cast(kwlist), &mode, &format, &shape, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (shape) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().shape = py::cast(py::handle(shape)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(AdaptivePooling)::py_getsetters[] = { {const_cast("mode"), py_get_generic(AdaptivePooling, mode), py_set_generic(AdaptivePooling, mode), const_cast("mode"), NULL}, {const_cast("format"), py_get_generic(AdaptivePooling, format), py_set_generic(AdaptivePooling, format), const_cast("format"), NULL}, {const_cast("shape"), py_get_generic(AdaptivePooling, shape), py_set_generic(AdaptivePooling, shape), const_cast("shape"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(AdaptivePooling)::tp_methods[] = { {const_cast("__getstate__"), PyOp(AdaptivePooling)::getstate, METH_NOARGS, "AdaptivePooling getstate"}, {const_cast("__setstate__"), PyOp(AdaptivePooling)::setstate, METH_VARARGS, "AdaptivePooling setstate"}, {NULL} /* Sentinel */ }; void _init_py_AdaptivePooling(py::module m) { using py_op = PyOp(AdaptivePooling); auto& py_type = PyOpType(AdaptivePooling); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.AdaptivePooling"; py_type.tp_basicsize = sizeof(PyOp(AdaptivePooling)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "AdaptivePooling"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_AdaptivePooling_Mode(py_type); _init_py_AdaptivePooling_Format(py_type); PyType_Modified(&py_type); m.add_object("AdaptivePooling", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(AdaptivePooling::typeinfo(), &py_type).second); } PyOpDefBegin(AddAxis) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(AddAxis) int PyOp(AddAxis)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "scope", NULL}; PyObject *axis = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &axis, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(AddAxis)::py_getsetters[] = { {const_cast("axis"), py_get_generic(AddAxis, axis), py_set_generic(AddAxis, axis), const_cast("axis"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(AddAxis)::tp_methods[] = { {const_cast("__getstate__"), PyOp(AddAxis)::getstate, METH_NOARGS, "AddAxis getstate"}, {const_cast("__setstate__"), PyOp(AddAxis)::setstate, METH_VARARGS, "AddAxis setstate"}, {NULL} /* Sentinel */ }; void _init_py_AddAxis(py::module m) { using py_op = PyOp(AddAxis); auto& py_type = PyOpType(AddAxis); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.AddAxis"; py_type.tp_basicsize = sizeof(PyOp(AddAxis)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "AddAxis"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("AddAxis", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(AddAxis::typeinfo(), &py_type).second); } PyOpDefBegin(Argmax) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Argmax) int PyOp(Argmax)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "scope", NULL}; PyObject *axis = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &axis, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Argmax)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Argmax, axis), py_set_generic(Argmax, axis), const_cast("axis"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Argmax)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Argmax)::getstate, METH_NOARGS, "Argmax getstate"}, {const_cast("__setstate__"), PyOp(Argmax)::setstate, METH_VARARGS, "Argmax setstate"}, {NULL} /* Sentinel */ }; void _init_py_Argmax(py::module m) { using py_op = PyOp(Argmax); auto& py_type = PyOpType(Argmax); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Argmax"; py_type.tp_basicsize = sizeof(PyOp(Argmax)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Argmax"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Argmax", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Argmax::typeinfo(), &py_type).second); } PyOpDefBegin(Argmin) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Argmin) int PyOp(Argmin)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "scope", NULL}; PyObject *axis = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &axis, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Argmin)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Argmin, axis), py_set_generic(Argmin, axis), const_cast("axis"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Argmin)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Argmin)::getstate, METH_NOARGS, "Argmin getstate"}, {const_cast("__setstate__"), PyOp(Argmin)::setstate, METH_VARARGS, "Argmin setstate"}, {NULL} /* Sentinel */ }; void _init_py_Argmin(py::module m) { using py_op = PyOp(Argmin); auto& py_type = PyOpType(Argmin); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Argmin"; py_type.tp_basicsize = sizeof(PyOp(Argmin)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Argmin"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Argmin", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Argmin::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Argsort.Order"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"ASCENDING", "DESCENDING"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("ASCENDING"), Argsort::Order::ASCENDING}, {normalize_enum("DESCENDING"), Argsort::Order::DESCENDING}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_Argsort_Order(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Argsort.Order", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Order").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Argsort.Order").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Argsort::Order::ASCENDING; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ASCENDING", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Argsort::Order::DESCENDING; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DESCENDING", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Order", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Argsort) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"order", serialization::dump(opdef.order)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("order"); if (iter != state.end()) { opdef.order = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Argsort) int PyOp(Argsort)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"order", "scope", NULL}; PyObject *order = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &order, &scope)) return -1; if (order) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().order = py::cast(py::handle(order)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Argsort)::py_getsetters[] = { {const_cast("order"), py_get_generic(Argsort, order), py_set_generic(Argsort, order), const_cast("order"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Argsort)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Argsort)::getstate, METH_NOARGS, "Argsort getstate"}, {const_cast("__setstate__"), PyOp(Argsort)::setstate, METH_VARARGS, "Argsort setstate"}, {NULL} /* Sentinel */ }; void _init_py_Argsort(py::module m) { using py_op = PyOp(Argsort); auto& py_type = PyOpType(Argsort); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Argsort"; py_type.tp_basicsize = sizeof(PyOp(Argsort)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Argsort"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Argsort_Order(py_type); PyType_Modified(&py_type); m.add_object("Argsort", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Argsort::typeinfo(), &py_type).second); } PyOpDefBegin(AssertEqual) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"maxerr", serialization::dump(opdef.maxerr)}, {"verbose", serialization::dump(opdef.verbose)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("maxerr"); if (iter != state.end()) { opdef.maxerr = serialization::load(iter->second); } } { auto&& iter = state.find("verbose"); if (iter != state.end()) { opdef.verbose = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(AssertEqual) int PyOp(AssertEqual)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"maxerr", "verbose", "scope", NULL}; PyObject *maxerr = NULL, *verbose = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &maxerr, &verbose, &scope)) return -1; if (maxerr) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().maxerr = py::cast(py::handle(maxerr)); } CATCH_ALL(-1) } if (verbose) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().verbose = py::cast(py::handle(verbose)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(AssertEqual)::py_getsetters[] = { {const_cast("maxerr"), py_get_generic(AssertEqual, maxerr), py_set_generic(AssertEqual, maxerr), const_cast("maxerr"), NULL}, {const_cast("verbose"), py_get_generic(AssertEqual, verbose), py_set_generic(AssertEqual, verbose), const_cast("verbose"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(AssertEqual)::tp_methods[] = { {const_cast("__getstate__"), PyOp(AssertEqual)::getstate, METH_NOARGS, "AssertEqual getstate"}, {const_cast("__setstate__"), PyOp(AssertEqual)::setstate, METH_VARARGS, "AssertEqual setstate"}, {NULL} /* Sentinel */ }; void _init_py_AssertEqual(py::module m) { using py_op = PyOp(AssertEqual); auto& py_type = PyOpType(AssertEqual); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.AssertEqual"; py_type.tp_basicsize = sizeof(PyOp(AssertEqual)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "AssertEqual"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("AssertEqual", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(AssertEqual::typeinfo(), &py_type).second); } PyOpDefBegin(AtlasRuntime) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"buf", serialization::dump(opdef.buf)}, {"buf_size", serialization::dump(opdef.buf_size)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("buf"); if (iter != state.end()) { opdef.buf = serialization::load(iter->second); } } { auto&& iter = state.find("buf_size"); if (iter != state.end()) { opdef.buf_size = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(AtlasRuntime) int PyOp(AtlasRuntime)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"buf", "buf_size", "scope", NULL}; PyObject *buf = NULL, *buf_size = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &buf, &buf_size, &scope)) return -1; if (buf) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf = py::cast(py::handle(buf)); } CATCH_ALL(-1) } if (buf_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf_size = py::cast(py::handle(buf_size)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(AtlasRuntime)::py_getsetters[] = { {const_cast("buf"), py_get_generic(AtlasRuntime, buf), py_set_generic(AtlasRuntime, buf), const_cast("buf"), NULL}, {const_cast("buf_size"), py_get_generic(AtlasRuntime, buf_size), py_set_generic(AtlasRuntime, buf_size), const_cast("buf_size"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(AtlasRuntime)::tp_methods[] = { {const_cast("__getstate__"), PyOp(AtlasRuntime)::getstate, METH_NOARGS, "AtlasRuntime getstate"}, {const_cast("__setstate__"), PyOp(AtlasRuntime)::setstate, METH_VARARGS, "AtlasRuntime setstate"}, {NULL} /* Sentinel */ }; void _init_py_AtlasRuntime(py::module m) { using py_op = PyOp(AtlasRuntime); auto& py_type = PyOpType(AtlasRuntime); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.AtlasRuntime"; py_type.tp_basicsize = sizeof(PyOp(AtlasRuntime)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "AtlasRuntime"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("AtlasRuntime", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(AtlasRuntime::typeinfo(), &py_type).second); } PyOpDefBegin(Barrier) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"comp_node", serialization::dump(opdef.comp_node)}, {"nr_outputs", serialization::dump(opdef.nr_outputs)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } { auto&& iter = state.find("nr_outputs"); if (iter != state.end()) { opdef.nr_outputs = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Barrier) int PyOp(Barrier)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"comp_node", "nr_outputs", "scope", NULL}; PyObject *comp_node = NULL, *nr_outputs = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &comp_node, &nr_outputs, &scope)) return -1; if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (nr_outputs) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nr_outputs = py::cast(py::handle(nr_outputs)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Barrier)::py_getsetters[] = { {const_cast("comp_node"), py_get_generic(Barrier, comp_node), py_set_generic(Barrier, comp_node), const_cast("comp_node"), NULL}, {const_cast("nr_outputs"), py_get_generic(Barrier, nr_outputs), py_set_generic(Barrier, nr_outputs), const_cast("nr_outputs"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Barrier)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Barrier)::getstate, METH_NOARGS, "Barrier getstate"}, {const_cast("__setstate__"), PyOp(Barrier)::setstate, METH_VARARGS, "Barrier setstate"}, {NULL} /* Sentinel */ }; void _init_py_Barrier(py::module m) { using py_op = PyOp(Barrier); auto& py_type = PyOpType(Barrier); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Barrier"; py_type.tp_basicsize = sizeof(PyOp(Barrier)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Barrier"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Barrier", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Barrier::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "BatchConvBias.NonlineMode"; static constexpr std::underlying_type_t max = 4 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"IDENTITY", "RELU", "SIGMOID", "H_SWISH"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("IDENTITY"), BatchConvBias::NonlineMode::IDENTITY}, {normalize_enum("RELU"), BatchConvBias::NonlineMode::RELU}, {normalize_enum("SIGMOID"), BatchConvBias::NonlineMode::SIGMOID}, {normalize_enum("H_SWISH"), BatchConvBias::NonlineMode::H_SWISH}}; template<> PyObject* EnumWrapper::pyobj_insts[4] = {nullptr}; void _init_py_BatchConvBias_NonlineMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchConvBias.NonlineMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("NonlineMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchConvBias.NonlineMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::NonlineMode::IDENTITY; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "IDENTITY", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::NonlineMode::RELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RELU", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::NonlineMode::SIGMOID; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SIGMOID", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::NonlineMode::H_SWISH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "H_SWISH", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "NonlineMode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "BatchConvBias.Mode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"CROSS_CORRELATION", "CONVOLUTION"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("CROSS_CORRELATION"), BatchConvBias::Mode::CROSS_CORRELATION}, {normalize_enum("CONVOLUTION"), BatchConvBias::Mode::CONVOLUTION}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_BatchConvBias_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchConvBias.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchConvBias.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Mode::CROSS_CORRELATION; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CROSS_CORRELATION", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Mode::CONVOLUTION; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CONVOLUTION", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "BatchConvBias.Sparse"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DENSE", "GROUP"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DENSE"), BatchConvBias::Sparse::DENSE}, {normalize_enum("GROUP"), BatchConvBias::Sparse::GROUP}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_BatchConvBias_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchConvBias.Sparse", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Sparse").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchConvBias.Sparse").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Sparse::DENSE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DENSE", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Sparse::GROUP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "GROUP", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_BatchConvBias_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "BatchConvBias.ComputeMode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DEFAULT", "FLOAT32"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DEFAULT"), BatchConvBias::ComputeMode::DEFAULT}, {normalize_enum("FLOAT32"), BatchConvBias::ComputeMode::FLOAT32}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_BatchConvBias_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchConvBias.ComputeMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("ComputeMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchConvBias.ComputeMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::ComputeMode::DEFAULT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DEFAULT", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::ComputeMode::FLOAT32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT32", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "BatchConvBias.Strategy"; static constexpr std::underlying_type_t max = (1llu << 4) - 1; }; template<> PyTypeObject* BitCombinedEnumWrapper::type = nullptr; template<> const char* BitCombinedEnumWrapper::members[] = {"HEURISTIC", "PROFILE", "REPRODUCIBLE", "OPTIMIZED"}; template<> std::unordered_map BitCombinedEnumWrapper::mem2value = {{normalize_enum("HEURISTIC"), BatchConvBias::Strategy::HEURISTIC}, {normalize_enum("PROFILE"), BatchConvBias::Strategy::PROFILE}, {normalize_enum("REPRODUCIBLE"), BatchConvBias::Strategy::REPRODUCIBLE}, {normalize_enum("OPTIMIZED"), BatchConvBias::Strategy::OPTIMIZED}}; template<> PyObject* BitCombinedEnumWrapper::pyobj_insts[4] = {nullptr}; void _init_py_BatchConvBias_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)BitCombinedEnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)BitCombinedEnumWrapper::py_repr}, {Py_tp_richcompare, (void*)BitCombinedEnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {Py_tp_new, (void*)BitCombinedEnumWrapper::py_new_combined_enum}, {Py_nb_or, (void*)BitCombinedEnumWrapper::py_or}, {Py_nb_and, (void*)BitCombinedEnumWrapper::py_and}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchConvBias.Strategy", // basicsize sizeof(BitCombinedEnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Strategy").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchConvBias.Strategy").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Strategy::HEURISTIC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "HEURISTIC", inst) >= 0); BitCombinedEnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Strategy::PROFILE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "PROFILE", inst) >= 0); BitCombinedEnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Strategy::REPRODUCIBLE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REPRODUCIBLE", inst) >= 0); BitCombinedEnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchConvBias::Strategy::OPTIMIZED; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "OPTIMIZED", inst) >= 0); BitCombinedEnumWrapper::pyobj_insts[3] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(BatchConvBias) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"nonlineMode", serialization::dump(opdef.nonlineMode)}, {"mode", serialization::dump(opdef.mode)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"format", serialization::dump(opdef.format)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)}, {"dtype", serialization::dump(opdef.dtype)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("nonlineMode"); if (iter != state.end()) { opdef.nonlineMode = serialization::load(iter->second); } } { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchConvBias) int PyOp(BatchConvBias)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"nonlineMode", "mode", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "sparse", "format", "compute_mode", "strategy", "workspace_limit", "dtype", "scope", NULL}; PyObject *nonlineMode = NULL, *mode = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *format = NULL, *compute_mode = NULL, *strategy = NULL, *workspace_limit = NULL, *dtype = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOOOO", const_cast(kwlist), &nonlineMode, &mode, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &sparse, &format, &compute_mode, &strategy, &workspace_limit, &dtype, &scope)) return -1; if (nonlineMode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nonlineMode = py::cast(py::handle(nonlineMode)); } CATCH_ALL(-1) } if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchConvBias)::py_getsetters[] = { {const_cast("nonlineMode"), py_get_generic(BatchConvBias, nonlineMode), py_set_generic(BatchConvBias, nonlineMode), const_cast("nonlineMode"), NULL}, {const_cast("mode"), py_get_generic(BatchConvBias, mode), py_set_generic(BatchConvBias, mode), const_cast("mode"), NULL}, {const_cast("pad_h"), py_get_generic(BatchConvBias, pad_h), py_set_generic(BatchConvBias, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(BatchConvBias, pad_w), py_set_generic(BatchConvBias, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(BatchConvBias, stride_h), py_set_generic(BatchConvBias, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(BatchConvBias, stride_w), py_set_generic(BatchConvBias, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(BatchConvBias, dilate_h), py_set_generic(BatchConvBias, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(BatchConvBias, dilate_w), py_set_generic(BatchConvBias, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(BatchConvBias, sparse), py_set_generic(BatchConvBias, sparse), const_cast("sparse"), NULL}, {const_cast("format"), py_get_generic(BatchConvBias, format), py_set_generic(BatchConvBias, format), const_cast("format"), NULL}, {const_cast("compute_mode"), py_get_generic(BatchConvBias, compute_mode), py_set_generic(BatchConvBias, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("strategy"), py_get_generic(BatchConvBias, strategy), py_set_generic(BatchConvBias, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(BatchConvBias, workspace_limit), py_set_generic(BatchConvBias, workspace_limit), const_cast("workspace_limit"), NULL}, {const_cast("dtype"), py_get_generic(BatchConvBias, dtype), py_set_generic(BatchConvBias, dtype), const_cast("dtype"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchConvBias)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchConvBias)::getstate, METH_NOARGS, "BatchConvBias getstate"}, {const_cast("__setstate__"), PyOp(BatchConvBias)::setstate, METH_VARARGS, "BatchConvBias setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchConvBias(py::module m) { using py_op = PyOp(BatchConvBias); auto& py_type = PyOpType(BatchConvBias); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchConvBias"; py_type.tp_basicsize = sizeof(PyOp(BatchConvBias)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchConvBias"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_BatchConvBias_NonlineMode(py_type); _init_py_BatchConvBias_Mode(py_type); _init_py_BatchConvBias_Sparse(py_type); _init_py_BatchConvBias_Format(py_type); _init_py_BatchConvBias_ComputeMode(py_type); _init_py_BatchConvBias_Strategy(py_type); PyType_Modified(&py_type); m.add_object("BatchConvBias", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchConvBias::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "BatchNorm.ParamDim"; static constexpr std::underlying_type_t max = 4 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DIM_11HW", "DIM_1CHW", "DIM_1C11", "DIM_111C"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DIM_11HW"), BatchNorm::ParamDim::DIM_11HW}, {normalize_enum("DIM_1CHW"), BatchNorm::ParamDim::DIM_1CHW}, {normalize_enum("DIM_1C11"), BatchNorm::ParamDim::DIM_1C11}, {normalize_enum("DIM_111C"), BatchNorm::ParamDim::DIM_111C}}; template<> PyObject* EnumWrapper::pyobj_insts[4] = {nullptr}; void _init_py_BatchNorm_ParamDim(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchNorm.ParamDim", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("ParamDim").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchNorm.ParamDim").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchNorm::ParamDim::DIM_11HW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DIM_11HW", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchNorm::ParamDim::DIM_1CHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DIM_1CHW", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchNorm::ParamDim::DIM_1C11; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DIM_1C11", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchNorm::ParamDim::DIM_111C; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DIM_111C", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ParamDim", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "BatchNorm.FwdMode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"TRAINING", "INFERENCE"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("TRAINING"), BatchNorm::FwdMode::TRAINING}, {normalize_enum("INFERENCE"), BatchNorm::FwdMode::INFERENCE}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_BatchNorm_FwdMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchNorm.FwdMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("FwdMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchNorm.FwdMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchNorm::FwdMode::TRAINING; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "TRAINING", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchNorm::FwdMode::INFERENCE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "INFERENCE", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "FwdMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(BatchNorm) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"param_dim", serialization::dump(opdef.param_dim)}, {"fwd_mode", serialization::dump(opdef.fwd_mode)}, {"epsilon", serialization::dump(opdef.epsilon)}, {"avg_factor", serialization::dump(opdef.avg_factor)}, {"scale", serialization::dump(opdef.scale)}, {"bias", serialization::dump(opdef.bias)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("param_dim"); if (iter != state.end()) { opdef.param_dim = serialization::load(iter->second); } } { auto&& iter = state.find("fwd_mode"); if (iter != state.end()) { opdef.fwd_mode = serialization::load(iter->second); } } { auto&& iter = state.find("epsilon"); if (iter != state.end()) { opdef.epsilon = serialization::load(iter->second); } } { auto&& iter = state.find("avg_factor"); if (iter != state.end()) { opdef.avg_factor = serialization::load(iter->second); } } { auto&& iter = state.find("scale"); if (iter != state.end()) { opdef.scale = serialization::load(iter->second); } } { auto&& iter = state.find("bias"); if (iter != state.end()) { opdef.bias = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchNorm) int PyOp(BatchNorm)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"param_dim", "fwd_mode", "epsilon", "avg_factor", "scale", "bias", "scope", NULL}; PyObject *param_dim = NULL, *fwd_mode = NULL, *epsilon = NULL, *avg_factor = NULL, *scale = NULL, *bias = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOO", const_cast(kwlist), ¶m_dim, &fwd_mode, &epsilon, &avg_factor, &scale, &bias, &scope)) return -1; if (param_dim) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().param_dim = py::cast(py::handle(param_dim)); } CATCH_ALL(-1) } if (fwd_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().fwd_mode = py::cast(py::handle(fwd_mode)); } CATCH_ALL(-1) } if (epsilon) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().epsilon = py::cast(py::handle(epsilon)); } CATCH_ALL(-1) } if (avg_factor) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().avg_factor = py::cast(py::handle(avg_factor)); } CATCH_ALL(-1) } if (scale) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().scale = py::cast(py::handle(scale)); } CATCH_ALL(-1) } if (bias) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bias = py::cast(py::handle(bias)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchNorm)::py_getsetters[] = { {const_cast("param_dim"), py_get_generic(BatchNorm, param_dim), py_set_generic(BatchNorm, param_dim), const_cast("param_dim"), NULL}, {const_cast("fwd_mode"), py_get_generic(BatchNorm, fwd_mode), py_set_generic(BatchNorm, fwd_mode), const_cast("fwd_mode"), NULL}, {const_cast("epsilon"), py_get_generic(BatchNorm, epsilon), py_set_generic(BatchNorm, epsilon), const_cast("epsilon"), NULL}, {const_cast("avg_factor"), py_get_generic(BatchNorm, avg_factor), py_set_generic(BatchNorm, avg_factor), const_cast("avg_factor"), NULL}, {const_cast("scale"), py_get_generic(BatchNorm, scale), py_set_generic(BatchNorm, scale), const_cast("scale"), NULL}, {const_cast("bias"), py_get_generic(BatchNorm, bias), py_set_generic(BatchNorm, bias), const_cast("bias"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchNorm)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchNorm)::getstate, METH_NOARGS, "BatchNorm getstate"}, {const_cast("__setstate__"), PyOp(BatchNorm)::setstate, METH_VARARGS, "BatchNorm setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchNorm(py::module m) { using py_op = PyOp(BatchNorm); auto& py_type = PyOpType(BatchNorm); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchNorm"; py_type.tp_basicsize = sizeof(PyOp(BatchNorm)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchNorm"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_BatchNorm_ParamDim(py_type); _init_py_BatchNorm_FwdMode(py_type); PyType_Modified(&py_type); m.add_object("BatchNorm", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchNorm::typeinfo(), &py_type).second); } void _init_py_BatchNormBackward_ParamDim(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ParamDim", reinterpret_cast(e_type)) >= 0); } void _init_py_BatchNormBackward_FwdMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "FwdMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(BatchNormBackward) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"param_dim", serialization::dump(opdef.param_dim)}, {"fwd_mode", serialization::dump(opdef.fwd_mode)}, {"epsilon", serialization::dump(opdef.epsilon)}, {"avg_factor", serialization::dump(opdef.avg_factor)}, {"scale", serialization::dump(opdef.scale)}, {"bias", serialization::dump(opdef.bias)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("param_dim"); if (iter != state.end()) { opdef.param_dim = serialization::load(iter->second); } } { auto&& iter = state.find("fwd_mode"); if (iter != state.end()) { opdef.fwd_mode = serialization::load(iter->second); } } { auto&& iter = state.find("epsilon"); if (iter != state.end()) { opdef.epsilon = serialization::load(iter->second); } } { auto&& iter = state.find("avg_factor"); if (iter != state.end()) { opdef.avg_factor = serialization::load(iter->second); } } { auto&& iter = state.find("scale"); if (iter != state.end()) { opdef.scale = serialization::load(iter->second); } } { auto&& iter = state.find("bias"); if (iter != state.end()) { opdef.bias = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchNormBackward) int PyOp(BatchNormBackward)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"param_dim", "fwd_mode", "epsilon", "avg_factor", "scale", "bias", "scope", NULL}; PyObject *param_dim = NULL, *fwd_mode = NULL, *epsilon = NULL, *avg_factor = NULL, *scale = NULL, *bias = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOO", const_cast(kwlist), ¶m_dim, &fwd_mode, &epsilon, &avg_factor, &scale, &bias, &scope)) return -1; if (param_dim) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().param_dim = py::cast(py::handle(param_dim)); } CATCH_ALL(-1) } if (fwd_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().fwd_mode = py::cast(py::handle(fwd_mode)); } CATCH_ALL(-1) } if (epsilon) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().epsilon = py::cast(py::handle(epsilon)); } CATCH_ALL(-1) } if (avg_factor) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().avg_factor = py::cast(py::handle(avg_factor)); } CATCH_ALL(-1) } if (scale) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().scale = py::cast(py::handle(scale)); } CATCH_ALL(-1) } if (bias) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bias = py::cast(py::handle(bias)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchNormBackward)::py_getsetters[] = { {const_cast("param_dim"), py_get_generic(BatchNormBackward, param_dim), py_set_generic(BatchNormBackward, param_dim), const_cast("param_dim"), NULL}, {const_cast("fwd_mode"), py_get_generic(BatchNormBackward, fwd_mode), py_set_generic(BatchNormBackward, fwd_mode), const_cast("fwd_mode"), NULL}, {const_cast("epsilon"), py_get_generic(BatchNormBackward, epsilon), py_set_generic(BatchNormBackward, epsilon), const_cast("epsilon"), NULL}, {const_cast("avg_factor"), py_get_generic(BatchNormBackward, avg_factor), py_set_generic(BatchNormBackward, avg_factor), const_cast("avg_factor"), NULL}, {const_cast("scale"), py_get_generic(BatchNormBackward, scale), py_set_generic(BatchNormBackward, scale), const_cast("scale"), NULL}, {const_cast("bias"), py_get_generic(BatchNormBackward, bias), py_set_generic(BatchNormBackward, bias), const_cast("bias"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchNormBackward)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchNormBackward)::getstate, METH_NOARGS, "BatchNormBackward getstate"}, {const_cast("__setstate__"), PyOp(BatchNormBackward)::setstate, METH_VARARGS, "BatchNormBackward setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchNormBackward(py::module m) { using py_op = PyOp(BatchNormBackward); auto& py_type = PyOpType(BatchNormBackward); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchNormBackward"; py_type.tp_basicsize = sizeof(PyOp(BatchNormBackward)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchNormBackward"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_BatchNormBackward_ParamDim(py_type); _init_py_BatchNormBackward_FwdMode(py_type); PyType_Modified(&py_type); m.add_object("BatchNormBackward", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchNormBackward::typeinfo(), &py_type).second); } PyOpDefBegin(BatchedIncrMeshIndexing) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchedIncrMeshIndexing) int PyOp(BatchedIncrMeshIndexing)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchedIncrMeshIndexing)::py_getsetters[] = { {const_cast("items"), py_get_generic(BatchedIncrMeshIndexing, items), py_set_generic(BatchedIncrMeshIndexing, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchedIncrMeshIndexing)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchedIncrMeshIndexing)::getstate, METH_NOARGS, "BatchedIncrMeshIndexing getstate"}, {const_cast("__setstate__"), PyOp(BatchedIncrMeshIndexing)::setstate, METH_VARARGS, "BatchedIncrMeshIndexing setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchedIncrMeshIndexing(py::module m) { using py_op = PyOp(BatchedIncrMeshIndexing); auto& py_type = PyOpType(BatchedIncrMeshIndexing); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchedIncrMeshIndexing"; py_type.tp_basicsize = sizeof(PyOp(BatchedIncrMeshIndexing)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchedIncrMeshIndexing"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("BatchedIncrMeshIndexing", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchedIncrMeshIndexing::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "BatchedMatrixMul.ComputeMode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DEFAULT", "FLOAT32"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DEFAULT"), BatchedMatrixMul::ComputeMode::DEFAULT}, {normalize_enum("FLOAT32"), BatchedMatrixMul::ComputeMode::FLOAT32}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_BatchedMatrixMul_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchedMatrixMul.ComputeMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("ComputeMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchedMatrixMul.ComputeMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::ComputeMode::DEFAULT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DEFAULT", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::ComputeMode::FLOAT32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT32", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "BatchedMatrixMul.Format"; static constexpr std::underlying_type_t max = 5 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DEFAULT", "MK4", "MK8", "MK4_DOT", "N32K4_DOT"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DEFAULT"), BatchedMatrixMul::Format::DEFAULT}, {normalize_enum("MK4"), BatchedMatrixMul::Format::MK4}, {normalize_enum("MK8"), BatchedMatrixMul::Format::MK8}, {normalize_enum("MK4_DOT"), BatchedMatrixMul::Format::MK4_DOT}, {normalize_enum("N32K4_DOT"), BatchedMatrixMul::Format::N32K4_DOT}}; template<> PyObject* EnumWrapper::pyobj_insts[5] = {nullptr}; void _init_py_BatchedMatrixMul_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.BatchedMatrixMul.Format", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Format").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("BatchedMatrixMul.Format").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::Format::DEFAULT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DEFAULT", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::Format::MK4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MK4", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::Format::MK8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MK8", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::Format::MK4_DOT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MK4_DOT", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = BatchedMatrixMul::Format::N32K4_DOT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "N32K4_DOT", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_BatchedMatrixMul_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(BatchedMatrixMul) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"transposeA", serialization::dump(opdef.transposeA)}, {"transposeB", serialization::dump(opdef.transposeB)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"format", serialization::dump(opdef.format)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)}, {"dimA", serialization::dump(opdef.dimA)}, {"dimB", serialization::dump(opdef.dimB)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("transposeA"); if (iter != state.end()) { opdef.transposeA = serialization::load(iter->second); } } { auto&& iter = state.find("transposeB"); if (iter != state.end()) { opdef.transposeB = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } { auto&& iter = state.find("dimA"); if (iter != state.end()) { opdef.dimA = serialization::load(iter->second); } } { auto&& iter = state.find("dimB"); if (iter != state.end()) { opdef.dimB = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchedMatrixMul) int PyOp(BatchedMatrixMul)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"transposeA", "transposeB", "compute_mode", "format", "strategy", "workspace_limit", "dimA", "dimB", "scope", NULL}; PyObject *transposeA = NULL, *transposeB = NULL, *compute_mode = NULL, *format = NULL, *strategy = NULL, *workspace_limit = NULL, *dimA = NULL, *dimB = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOO", const_cast(kwlist), &transposeA, &transposeB, &compute_mode, &format, &strategy, &workspace_limit, &dimA, &dimB, &scope)) return -1; if (transposeA) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().transposeA = py::cast(py::handle(transposeA)); } CATCH_ALL(-1) } if (transposeB) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().transposeB = py::cast(py::handle(transposeB)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (dimA) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dimA = py::cast(py::handle(dimA)); } CATCH_ALL(-1) } if (dimB) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dimB = py::cast(py::handle(dimB)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchedMatrixMul)::py_getsetters[] = { {const_cast("transposeA"), py_get_generic(BatchedMatrixMul, transposeA), py_set_generic(BatchedMatrixMul, transposeA), const_cast("transposeA"), NULL}, {const_cast("transposeB"), py_get_generic(BatchedMatrixMul, transposeB), py_set_generic(BatchedMatrixMul, transposeB), const_cast("transposeB"), NULL}, {const_cast("compute_mode"), py_get_generic(BatchedMatrixMul, compute_mode), py_set_generic(BatchedMatrixMul, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("format"), py_get_generic(BatchedMatrixMul, format), py_set_generic(BatchedMatrixMul, format), const_cast("format"), NULL}, {const_cast("strategy"), py_get_generic(BatchedMatrixMul, strategy), py_set_generic(BatchedMatrixMul, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(BatchedMatrixMul, workspace_limit), py_set_generic(BatchedMatrixMul, workspace_limit), const_cast("workspace_limit"), NULL}, {const_cast("dimA"), py_get_generic(BatchedMatrixMul, dimA), py_set_generic(BatchedMatrixMul, dimA), const_cast("dimA"), NULL}, {const_cast("dimB"), py_get_generic(BatchedMatrixMul, dimB), py_set_generic(BatchedMatrixMul, dimB), const_cast("dimB"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchedMatrixMul)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchedMatrixMul)::getstate, METH_NOARGS, "BatchedMatrixMul getstate"}, {const_cast("__setstate__"), PyOp(BatchedMatrixMul)::setstate, METH_VARARGS, "BatchedMatrixMul setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchedMatrixMul(py::module m) { using py_op = PyOp(BatchedMatrixMul); auto& py_type = PyOpType(BatchedMatrixMul); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchedMatrixMul"; py_type.tp_basicsize = sizeof(PyOp(BatchedMatrixMul)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchedMatrixMul"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_BatchedMatrixMul_ComputeMode(py_type); _init_py_BatchedMatrixMul_Format(py_type); _init_py_BatchedMatrixMul_Strategy(py_type); PyType_Modified(&py_type); m.add_object("BatchedMatrixMul", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchedMatrixMul::typeinfo(), &py_type).second); } PyOpDefBegin(BatchedMeshIndexing) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchedMeshIndexing) int PyOp(BatchedMeshIndexing)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchedMeshIndexing)::py_getsetters[] = { {const_cast("items"), py_get_generic(BatchedMeshIndexing, items), py_set_generic(BatchedMeshIndexing, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchedMeshIndexing)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchedMeshIndexing)::getstate, METH_NOARGS, "BatchedMeshIndexing getstate"}, {const_cast("__setstate__"), PyOp(BatchedMeshIndexing)::setstate, METH_VARARGS, "BatchedMeshIndexing setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchedMeshIndexing(py::module m) { using py_op = PyOp(BatchedMeshIndexing); auto& py_type = PyOpType(BatchedMeshIndexing); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchedMeshIndexing"; py_type.tp_basicsize = sizeof(PyOp(BatchedMeshIndexing)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchedMeshIndexing"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("BatchedMeshIndexing", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchedMeshIndexing::typeinfo(), &py_type).second); } PyOpDefBegin(BatchedSetMeshIndexing) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BatchedSetMeshIndexing) int PyOp(BatchedSetMeshIndexing)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BatchedSetMeshIndexing)::py_getsetters[] = { {const_cast("items"), py_get_generic(BatchedSetMeshIndexing, items), py_set_generic(BatchedSetMeshIndexing, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BatchedSetMeshIndexing)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BatchedSetMeshIndexing)::getstate, METH_NOARGS, "BatchedSetMeshIndexing getstate"}, {const_cast("__setstate__"), PyOp(BatchedSetMeshIndexing)::setstate, METH_VARARGS, "BatchedSetMeshIndexing setstate"}, {NULL} /* Sentinel */ }; void _init_py_BatchedSetMeshIndexing(py::module m) { using py_op = PyOp(BatchedSetMeshIndexing); auto& py_type = PyOpType(BatchedSetMeshIndexing); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BatchedSetMeshIndexing"; py_type.tp_basicsize = sizeof(PyOp(BatchedSetMeshIndexing)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BatchedSetMeshIndexing"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("BatchedSetMeshIndexing", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BatchedSetMeshIndexing::typeinfo(), &py_type).second); } PyOpDefBegin(BetaRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(BetaRNG) int PyOp(BetaRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "handle", "scope", NULL}; PyObject *seed = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &seed, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(BetaRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(BetaRNG, seed), py_set_generic(BetaRNG, seed), const_cast("seed"), NULL}, {const_cast("handle"), py_get_generic(BetaRNG, handle), py_set_generic(BetaRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(BetaRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(BetaRNG)::getstate, METH_NOARGS, "BetaRNG getstate"}, {const_cast("__setstate__"), PyOp(BetaRNG)::setstate, METH_VARARGS, "BetaRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_BetaRNG(py::module m) { using py_op = PyOp(BetaRNG); auto& py_type = PyOpType(BetaRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.BetaRNG"; py_type.tp_basicsize = sizeof(PyOp(BetaRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "BetaRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("BetaRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(BetaRNG::typeinfo(), &py_type).second); } PyOpDefBegin(Borrow) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"comp_node", serialization::dump(opdef.comp_node)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Borrow) int PyOp(Borrow)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"comp_node", "scope", NULL}; PyObject *comp_node = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &comp_node, &scope)) return -1; if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Borrow)::py_getsetters[] = { {const_cast("comp_node"), py_get_generic(Borrow, comp_node), py_set_generic(Borrow, comp_node), const_cast("comp_node"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Borrow)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Borrow)::getstate, METH_NOARGS, "Borrow getstate"}, {const_cast("__setstate__"), PyOp(Borrow)::setstate, METH_VARARGS, "Borrow setstate"}, {NULL} /* Sentinel */ }; void _init_py_Borrow(py::module m) { using py_op = PyOp(Borrow); auto& py_type = PyOpType(Borrow); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Borrow"; py_type.tp_basicsize = sizeof(PyOp(Borrow)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Borrow"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Borrow", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Borrow::typeinfo(), &py_type).second); } PyOpDefBegin(Broadcast) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"shape", serialization::dump(opdef.shape)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("shape"); if (iter != state.end()) { opdef.shape = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Broadcast) int PyOp(Broadcast)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"shape", "scope", NULL}; PyObject *shape = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &shape, &scope)) return -1; if (shape) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().shape = py::cast(py::handle(shape)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Broadcast)::py_getsetters[] = { {const_cast("shape"), py_get_generic(Broadcast, shape), py_set_generic(Broadcast, shape), const_cast("shape"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Broadcast)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Broadcast)::getstate, METH_NOARGS, "Broadcast getstate"}, {const_cast("__setstate__"), PyOp(Broadcast)::setstate, METH_VARARGS, "Broadcast setstate"}, {NULL} /* Sentinel */ }; void _init_py_Broadcast(py::module m) { using py_op = PyOp(Broadcast); auto& py_type = PyOpType(Broadcast); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Broadcast"; py_type.tp_basicsize = sizeof(PyOp(Broadcast)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Broadcast"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Broadcast", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Broadcast::typeinfo(), &py_type).second); } PyOpDefBegin(CambriconRuntime) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"buf", serialization::dump(opdef.buf)}, {"buf_size", serialization::dump(opdef.buf_size)}, {"symbol", serialization::dump(opdef.symbol)}, {"tensor_dim_mutable", serialization::dump(opdef.tensor_dim_mutable)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("buf"); if (iter != state.end()) { opdef.buf = serialization::load(iter->second); } } { auto&& iter = state.find("buf_size"); if (iter != state.end()) { opdef.buf_size = serialization::load(iter->second); } } { auto&& iter = state.find("symbol"); if (iter != state.end()) { opdef.symbol = serialization::load(iter->second); } } { auto&& iter = state.find("tensor_dim_mutable"); if (iter != state.end()) { opdef.tensor_dim_mutable = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(CambriconRuntime) int PyOp(CambriconRuntime)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"buf", "buf_size", "symbol", "tensor_dim_mutable", "scope", NULL}; PyObject *buf = NULL, *buf_size = NULL, *symbol = NULL, *tensor_dim_mutable = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &buf, &buf_size, &symbol, &tensor_dim_mutable, &scope)) return -1; if (buf) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf = py::cast(py::handle(buf)); } CATCH_ALL(-1) } if (buf_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf_size = py::cast(py::handle(buf_size)); } CATCH_ALL(-1) } if (symbol) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().symbol = py::cast(py::handle(symbol)); } CATCH_ALL(-1) } if (tensor_dim_mutable) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().tensor_dim_mutable = py::cast(py::handle(tensor_dim_mutable)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(CambriconRuntime)::py_getsetters[] = { {const_cast("buf"), py_get_generic(CambriconRuntime, buf), py_set_generic(CambriconRuntime, buf), const_cast("buf"), NULL}, {const_cast("buf_size"), py_get_generic(CambriconRuntime, buf_size), py_set_generic(CambriconRuntime, buf_size), const_cast("buf_size"), NULL}, {const_cast("symbol"), py_get_generic(CambriconRuntime, symbol), py_set_generic(CambriconRuntime, symbol), const_cast("symbol"), NULL}, {const_cast("tensor_dim_mutable"), py_get_generic(CambriconRuntime, tensor_dim_mutable), py_set_generic(CambriconRuntime, tensor_dim_mutable), const_cast("tensor_dim_mutable"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(CambriconRuntime)::tp_methods[] = { {const_cast("__getstate__"), PyOp(CambriconRuntime)::getstate, METH_NOARGS, "CambriconRuntime getstate"}, {const_cast("__setstate__"), PyOp(CambriconRuntime)::setstate, METH_VARARGS, "CambriconRuntime setstate"}, {NULL} /* Sentinel */ }; void _init_py_CambriconRuntime(py::module m) { using py_op = PyOp(CambriconRuntime); auto& py_type = PyOpType(CambriconRuntime); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.CambriconRuntime"; py_type.tp_basicsize = sizeof(PyOp(CambriconRuntime)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "CambriconRuntime"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("CambriconRuntime", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(CambriconRuntime::typeinfo(), &py_type).second); } PyOpDefBegin(CheckNonFinite) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"scale", serialization::dump(opdef.scale)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("scale"); if (iter != state.end()) { opdef.scale = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(CheckNonFinite) int PyOp(CheckNonFinite)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"scale", "scope", NULL}; PyObject *scale = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &scale, &scope)) return -1; if (scale) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().scale = py::cast(py::handle(scale)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(CheckNonFinite)::py_getsetters[] = { {const_cast("scale"), py_get_generic(CheckNonFinite, scale), py_set_generic(CheckNonFinite, scale), const_cast("scale"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(CheckNonFinite)::tp_methods[] = { {const_cast("__getstate__"), PyOp(CheckNonFinite)::getstate, METH_NOARGS, "CheckNonFinite getstate"}, {const_cast("__setstate__"), PyOp(CheckNonFinite)::setstate, METH_VARARGS, "CheckNonFinite setstate"}, {NULL} /* Sentinel */ }; void _init_py_CheckNonFinite(py::module m) { using py_op = PyOp(CheckNonFinite); auto& py_type = PyOpType(CheckNonFinite); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.CheckNonFinite"; py_type.tp_basicsize = sizeof(PyOp(CheckNonFinite)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "CheckNonFinite"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("CheckNonFinite", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(CheckNonFinite::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "CollectiveComm.Mode"; static constexpr std::underlying_type_t max = 11 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"REDUCE_SUM", "BROADCAST", "ALL_GATHER", "REDUCE_SCATTER_SUM", "ALL_REDUCE_SUM", "ALL_REDUCE_MAX", "ALL_REDUCE_MIN", "ALL_REDUCE_PROD", "GATHER", "SCATTER", "ALL_TO_ALL"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("REDUCE_SUM"), CollectiveComm::Mode::REDUCE_SUM}, {normalize_enum("BROADCAST"), CollectiveComm::Mode::BROADCAST}, {normalize_enum("ALL_GATHER"), CollectiveComm::Mode::ALL_GATHER}, {normalize_enum("REDUCE_SCATTER_SUM"), CollectiveComm::Mode::REDUCE_SCATTER_SUM}, {normalize_enum("ALL_REDUCE_SUM"), CollectiveComm::Mode::ALL_REDUCE_SUM}, {normalize_enum("ALL_REDUCE_MAX"), CollectiveComm::Mode::ALL_REDUCE_MAX}, {normalize_enum("ALL_REDUCE_MIN"), CollectiveComm::Mode::ALL_REDUCE_MIN}, {normalize_enum("ALL_REDUCE_PROD"), CollectiveComm::Mode::ALL_REDUCE_PROD}, {normalize_enum("GATHER"), CollectiveComm::Mode::GATHER}, {normalize_enum("SCATTER"), CollectiveComm::Mode::SCATTER}, {normalize_enum("ALL_TO_ALL"), CollectiveComm::Mode::ALL_TO_ALL}}; template<> PyObject* EnumWrapper::pyobj_insts[11] = {nullptr}; void _init_py_CollectiveComm_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.CollectiveComm.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("CollectiveComm.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::REDUCE_SUM; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REDUCE_SUM", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::BROADCAST; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BROADCAST", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::ALL_GATHER; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ALL_GATHER", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::REDUCE_SCATTER_SUM; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REDUCE_SCATTER_SUM", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::ALL_REDUCE_SUM; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ALL_REDUCE_SUM", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::ALL_REDUCE_MAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ALL_REDUCE_MAX", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::ALL_REDUCE_MIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ALL_REDUCE_MIN", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::ALL_REDUCE_PROD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ALL_REDUCE_PROD", inst) >= 0); EnumWrapper::pyobj_insts[7] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::GATHER; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "GATHER", inst) >= 0); EnumWrapper::pyobj_insts[8] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::SCATTER; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SCATTER", inst) >= 0); EnumWrapper::pyobj_insts[9] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CollectiveComm::Mode::ALL_TO_ALL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ALL_TO_ALL", inst) >= 0); EnumWrapper::pyobj_insts[10] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(CollectiveComm) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"key", serialization::dump(opdef.key)}, {"nr_devices", serialization::dump(opdef.nr_devices)}, {"rank", serialization::dump(opdef.rank)}, {"is_root", serialization::dump(opdef.is_root)}, {"local_grad", serialization::dump(opdef.local_grad)}, {"addr", serialization::dump(opdef.addr)}, {"port", serialization::dump(opdef.port)}, {"dtype", serialization::dump(opdef.dtype)}, {"backend", serialization::dump(opdef.backend)}, {"comp_node", serialization::dump(opdef.comp_node)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("key"); if (iter != state.end()) { opdef.key = serialization::load(iter->second); } } { auto&& iter = state.find("nr_devices"); if (iter != state.end()) { opdef.nr_devices = serialization::load(iter->second); } } { auto&& iter = state.find("rank"); if (iter != state.end()) { opdef.rank = serialization::load(iter->second); } } { auto&& iter = state.find("is_root"); if (iter != state.end()) { opdef.is_root = serialization::load(iter->second); } } { auto&& iter = state.find("local_grad"); if (iter != state.end()) { opdef.local_grad = serialization::load(iter->second); } } { auto&& iter = state.find("addr"); if (iter != state.end()) { opdef.addr = serialization::load(iter->second); } } { auto&& iter = state.find("port"); if (iter != state.end()) { opdef.port = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } { auto&& iter = state.find("backend"); if (iter != state.end()) { opdef.backend = serialization::load(iter->second); } } { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(CollectiveComm) int PyOp(CollectiveComm)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "key", "nr_devices", "rank", "is_root", "local_grad", "addr", "port", "dtype", "backend", "comp_node", "scope", NULL}; PyObject *mode = NULL, *key = NULL, *nr_devices = NULL, *rank = NULL, *is_root = NULL, *local_grad = NULL, *addr = NULL, *port = NULL, *dtype = NULL, *backend = NULL, *comp_node = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOO", const_cast(kwlist), &mode, &key, &nr_devices, &rank, &is_root, &local_grad, &addr, &port, &dtype, &backend, &comp_node, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (key) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().key = py::cast(py::handle(key)); } CATCH_ALL(-1) } if (nr_devices) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nr_devices = py::cast(py::handle(nr_devices)); } CATCH_ALL(-1) } if (rank) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().rank = py::cast(py::handle(rank)); } CATCH_ALL(-1) } if (is_root) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().is_root = py::cast(py::handle(is_root)); } CATCH_ALL(-1) } if (local_grad) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().local_grad = py::cast(py::handle(local_grad)); } CATCH_ALL(-1) } if (addr) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().addr = py::cast(py::handle(addr)); } CATCH_ALL(-1) } if (port) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().port = py::cast(py::handle(port)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (backend) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().backend = py::cast(py::handle(backend)); } CATCH_ALL(-1) } if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(CollectiveComm)::py_getsetters[] = { {const_cast("mode"), py_get_generic(CollectiveComm, mode), py_set_generic(CollectiveComm, mode), const_cast("mode"), NULL}, {const_cast("key"), py_get_generic(CollectiveComm, key), py_set_generic(CollectiveComm, key), const_cast("key"), NULL}, {const_cast("nr_devices"), py_get_generic(CollectiveComm, nr_devices), py_set_generic(CollectiveComm, nr_devices), const_cast("nr_devices"), NULL}, {const_cast("rank"), py_get_generic(CollectiveComm, rank), py_set_generic(CollectiveComm, rank), const_cast("rank"), NULL}, {const_cast("is_root"), py_get_generic(CollectiveComm, is_root), py_set_generic(CollectiveComm, is_root), const_cast("is_root"), NULL}, {const_cast("local_grad"), py_get_generic(CollectiveComm, local_grad), py_set_generic(CollectiveComm, local_grad), const_cast("local_grad"), NULL}, {const_cast("addr"), py_get_generic(CollectiveComm, addr), py_set_generic(CollectiveComm, addr), const_cast("addr"), NULL}, {const_cast("port"), py_get_generic(CollectiveComm, port), py_set_generic(CollectiveComm, port), const_cast("port"), NULL}, {const_cast("dtype"), py_get_generic(CollectiveComm, dtype), py_set_generic(CollectiveComm, dtype), const_cast("dtype"), NULL}, {const_cast("backend"), py_get_generic(CollectiveComm, backend), py_set_generic(CollectiveComm, backend), const_cast("backend"), NULL}, {const_cast("comp_node"), py_get_generic(CollectiveComm, comp_node), py_set_generic(CollectiveComm, comp_node), const_cast("comp_node"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(CollectiveComm)::tp_methods[] = { {const_cast("__getstate__"), PyOp(CollectiveComm)::getstate, METH_NOARGS, "CollectiveComm getstate"}, {const_cast("__setstate__"), PyOp(CollectiveComm)::setstate, METH_VARARGS, "CollectiveComm setstate"}, {NULL} /* Sentinel */ }; void _init_py_CollectiveComm(py::module m) { using py_op = PyOp(CollectiveComm); auto& py_type = PyOpType(CollectiveComm); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.CollectiveComm"; py_type.tp_basicsize = sizeof(PyOp(CollectiveComm)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "CollectiveComm"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_CollectiveComm_Mode(py_type); PyType_Modified(&py_type); m.add_object("CollectiveComm", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(CollectiveComm::typeinfo(), &py_type).second); } PyOpDefBegin(Concat) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)}, {"comp_node", serialization::dump(opdef.comp_node)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Concat) int PyOp(Concat)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "comp_node", "scope", NULL}; PyObject *axis = NULL, *comp_node = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &axis, &comp_node, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Concat)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Concat, axis), py_set_generic(Concat, axis), const_cast("axis"), NULL}, {const_cast("comp_node"), py_get_generic(Concat, comp_node), py_set_generic(Concat, comp_node), const_cast("comp_node"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Concat)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Concat)::getstate, METH_NOARGS, "Concat getstate"}, {const_cast("__setstate__"), PyOp(Concat)::setstate, METH_VARARGS, "Concat setstate"}, {NULL} /* Sentinel */ }; void _init_py_Concat(py::module m) { using py_op = PyOp(Concat); auto& py_type = PyOpType(Concat); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Concat"; py_type.tp_basicsize = sizeof(PyOp(Concat)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Concat"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Concat", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Concat::typeinfo(), &py_type).second); } PyOpDefBegin(CondTake) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(CondTake) int PyOp(CondTake)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(CondTake)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(CondTake)::tp_methods[] = { {const_cast("__getstate__"), PyOp(CondTake)::getstate, METH_NOARGS, "CondTake getstate"}, {const_cast("__setstate__"), PyOp(CondTake)::setstate, METH_VARARGS, "CondTake setstate"}, {NULL} /* Sentinel */ }; void _init_py_CondTake(py::module m) { using py_op = PyOp(CondTake); auto& py_type = PyOpType(CondTake); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.CondTake"; py_type.tp_basicsize = sizeof(PyOp(CondTake)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "CondTake"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("CondTake", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(CondTake::typeinfo(), &py_type).second); } void _init_py_ConvBias_NonlineMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "NonlineMode", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvBias_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvBias_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvBias_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvBias_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvBias_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(ConvBias) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"nonlineMode", serialization::dump(opdef.nonlineMode)}, {"mode", serialization::dump(opdef.mode)}, {"sparse", serialization::dump(opdef.sparse)}, {"format", serialization::dump(opdef.format)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)}, {"dtype", serialization::dump(opdef.dtype)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("nonlineMode"); if (iter != state.end()) { opdef.nonlineMode = serialization::load(iter->second); } } { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ConvBias) int PyOp(ConvBias)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"nonlineMode", "mode", "sparse", "format", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "compute_mode", "strategy", "workspace_limit", "dtype", "scope", NULL}; PyObject *nonlineMode = NULL, *mode = NULL, *sparse = NULL, *format = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *compute_mode = NULL, *strategy = NULL, *workspace_limit = NULL, *dtype = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOOOO", const_cast(kwlist), &nonlineMode, &mode, &sparse, &format, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &compute_mode, &strategy, &workspace_limit, &dtype, &scope)) return -1; if (nonlineMode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nonlineMode = py::cast(py::handle(nonlineMode)); } CATCH_ALL(-1) } if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ConvBias)::py_getsetters[] = { {const_cast("nonlineMode"), py_get_generic(ConvBias, nonlineMode), py_set_generic(ConvBias, nonlineMode), const_cast("nonlineMode"), NULL}, {const_cast("mode"), py_get_generic(ConvBias, mode), py_set_generic(ConvBias, mode), const_cast("mode"), NULL}, {const_cast("sparse"), py_get_generic(ConvBias, sparse), py_set_generic(ConvBias, sparse), const_cast("sparse"), NULL}, {const_cast("format"), py_get_generic(ConvBias, format), py_set_generic(ConvBias, format), const_cast("format"), NULL}, {const_cast("pad_h"), py_get_generic(ConvBias, pad_h), py_set_generic(ConvBias, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(ConvBias, pad_w), py_set_generic(ConvBias, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(ConvBias, stride_h), py_set_generic(ConvBias, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(ConvBias, stride_w), py_set_generic(ConvBias, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(ConvBias, dilate_h), py_set_generic(ConvBias, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(ConvBias, dilate_w), py_set_generic(ConvBias, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("compute_mode"), py_get_generic(ConvBias, compute_mode), py_set_generic(ConvBias, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("strategy"), py_get_generic(ConvBias, strategy), py_set_generic(ConvBias, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(ConvBias, workspace_limit), py_set_generic(ConvBias, workspace_limit), const_cast("workspace_limit"), NULL}, {const_cast("dtype"), py_get_generic(ConvBias, dtype), py_set_generic(ConvBias, dtype), const_cast("dtype"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ConvBias)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ConvBias)::getstate, METH_NOARGS, "ConvBias getstate"}, {const_cast("__setstate__"), PyOp(ConvBias)::setstate, METH_VARARGS, "ConvBias setstate"}, {NULL} /* Sentinel */ }; void _init_py_ConvBias(py::module m) { using py_op = PyOp(ConvBias); auto& py_type = PyOpType(ConvBias); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ConvBias"; py_type.tp_basicsize = sizeof(PyOp(ConvBias)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ConvBias"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_ConvBias_NonlineMode(py_type); _init_py_ConvBias_Mode(py_type); _init_py_ConvBias_Sparse(py_type); _init_py_ConvBias_Format(py_type); _init_py_ConvBias_ComputeMode(py_type); _init_py_ConvBias_Strategy(py_type); PyType_Modified(&py_type); m.add_object("ConvBias", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ConvBias::typeinfo(), &py_type).second); } void _init_py_Convolution_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Convolution) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"format", serialization::dump(opdef.format)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Convolution) int PyOp(Convolution)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "sparse", "format", "compute_mode", "strategy", "workspace_limit", "scope", NULL}; PyObject *mode = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *format = NULL, *compute_mode = NULL, *strategy = NULL, *workspace_limit = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOO", const_cast(kwlist), &mode, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &sparse, &format, &compute_mode, &strategy, &workspace_limit, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Convolution)::py_getsetters[] = { {const_cast("mode"), py_get_generic(Convolution, mode), py_set_generic(Convolution, mode), const_cast("mode"), NULL}, {const_cast("pad_h"), py_get_generic(Convolution, pad_h), py_set_generic(Convolution, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(Convolution, pad_w), py_set_generic(Convolution, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(Convolution, stride_h), py_set_generic(Convolution, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(Convolution, stride_w), py_set_generic(Convolution, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(Convolution, dilate_h), py_set_generic(Convolution, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(Convolution, dilate_w), py_set_generic(Convolution, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(Convolution, sparse), py_set_generic(Convolution, sparse), const_cast("sparse"), NULL}, {const_cast("format"), py_get_generic(Convolution, format), py_set_generic(Convolution, format), const_cast("format"), NULL}, {const_cast("compute_mode"), py_get_generic(Convolution, compute_mode), py_set_generic(Convolution, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("strategy"), py_get_generic(Convolution, strategy), py_set_generic(Convolution, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(Convolution, workspace_limit), py_set_generic(Convolution, workspace_limit), const_cast("workspace_limit"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Convolution)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Convolution)::getstate, METH_NOARGS, "Convolution getstate"}, {const_cast("__setstate__"), PyOp(Convolution)::setstate, METH_VARARGS, "Convolution setstate"}, {NULL} /* Sentinel */ }; void _init_py_Convolution(py::module m) { using py_op = PyOp(Convolution); auto& py_type = PyOpType(Convolution); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Convolution"; py_type.tp_basicsize = sizeof(PyOp(Convolution)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Convolution"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Convolution_Mode(py_type); _init_py_Convolution_Sparse(py_type); _init_py_Convolution_Format(py_type); _init_py_Convolution_ComputeMode(py_type); _init_py_Convolution_Strategy(py_type); PyType_Modified(&py_type); m.add_object("Convolution", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Convolution::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Convolution3D.Mode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"CROSS_CORRELATION", "CONVOLUTION"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("CROSS_CORRELATION"), Convolution3D::Mode::CROSS_CORRELATION}, {normalize_enum("CONVOLUTION"), Convolution3D::Mode::CONVOLUTION}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_Convolution3D_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Convolution3D.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Convolution3D.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::Mode::CROSS_CORRELATION; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CROSS_CORRELATION", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::Mode::CONVOLUTION; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CONVOLUTION", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "Convolution3D.Sparse"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DENSE", "GROUP"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DENSE"), Convolution3D::Sparse::DENSE}, {normalize_enum("GROUP"), Convolution3D::Sparse::GROUP}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_Convolution3D_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Convolution3D.Sparse", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Sparse").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Convolution3D.Sparse").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::Sparse::DENSE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DENSE", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::Sparse::GROUP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "GROUP", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "Convolution3D.DataType"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"FLOAT", "FLOAT_IO16xC32"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("FLOAT"), Convolution3D::DataType::FLOAT}, {normalize_enum("FLOAT_IO16xC32"), Convolution3D::DataType::FLOAT_IO16xC32}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_Convolution3D_DataType(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Convolution3D.DataType", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("DataType").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Convolution3D.DataType").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::DataType::FLOAT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::DataType::FLOAT_IO16xC32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT_IO16xC32", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "DataType", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "Convolution3D.Format"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"NCDHW", "NDHWC"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("NCDHW"), Convolution3D::Format::NCDHW}, {normalize_enum("NDHWC"), Convolution3D::Format::NDHWC}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_Convolution3D_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Convolution3D.Format", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Format").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Convolution3D.Format").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::Format::NCDHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCDHW", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Convolution3D::Format::NDHWC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NDHWC", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution3D_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Convolution3D) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_d", serialization::dump(opdef.pad_d)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_d", serialization::dump(opdef.stride_d)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_d", serialization::dump(opdef.dilate_d)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"data_type", serialization::dump(opdef.data_type)}, {"format", serialization::dump(opdef.format)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_d"); if (iter != state.end()) { opdef.pad_d = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_d"); if (iter != state.end()) { opdef.stride_d = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_d"); if (iter != state.end()) { opdef.dilate_d = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("data_type"); if (iter != state.end()) { opdef.data_type = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Convolution3D) int PyOp(Convolution3D)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_d", "pad_h", "pad_w", "stride_d", "stride_h", "stride_w", "dilate_d", "dilate_h", "dilate_w", "sparse", "data_type", "format", "strategy", "workspace_limit", "scope", NULL}; PyObject *mode = NULL, *pad_d = NULL, *pad_h = NULL, *pad_w = NULL, *stride_d = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_d = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *data_type = NULL, *format = NULL, *strategy = NULL, *workspace_limit = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOOOOO", const_cast(kwlist), &mode, &pad_d, &pad_h, &pad_w, &stride_d, &stride_h, &stride_w, &dilate_d, &dilate_h, &dilate_w, &sparse, &data_type, &format, &strategy, &workspace_limit, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_d) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_d = py::cast(py::handle(pad_d)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_d) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_d = py::cast(py::handle(stride_d)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_d) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_d = py::cast(py::handle(dilate_d)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (data_type) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().data_type = py::cast(py::handle(data_type)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Convolution3D)::py_getsetters[] = { {const_cast("mode"), py_get_generic(Convolution3D, mode), py_set_generic(Convolution3D, mode), const_cast("mode"), NULL}, {const_cast("pad_d"), py_get_generic(Convolution3D, pad_d), py_set_generic(Convolution3D, pad_d), const_cast("pad_d"), NULL}, {const_cast("pad_h"), py_get_generic(Convolution3D, pad_h), py_set_generic(Convolution3D, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(Convolution3D, pad_w), py_set_generic(Convolution3D, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_d"), py_get_generic(Convolution3D, stride_d), py_set_generic(Convolution3D, stride_d), const_cast("stride_d"), NULL}, {const_cast("stride_h"), py_get_generic(Convolution3D, stride_h), py_set_generic(Convolution3D, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(Convolution3D, stride_w), py_set_generic(Convolution3D, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_d"), py_get_generic(Convolution3D, dilate_d), py_set_generic(Convolution3D, dilate_d), const_cast("dilate_d"), NULL}, {const_cast("dilate_h"), py_get_generic(Convolution3D, dilate_h), py_set_generic(Convolution3D, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(Convolution3D, dilate_w), py_set_generic(Convolution3D, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(Convolution3D, sparse), py_set_generic(Convolution3D, sparse), const_cast("sparse"), NULL}, {const_cast("data_type"), py_get_generic(Convolution3D, data_type), py_set_generic(Convolution3D, data_type), const_cast("data_type"), NULL}, {const_cast("format"), py_get_generic(Convolution3D, format), py_set_generic(Convolution3D, format), const_cast("format"), NULL}, {const_cast("strategy"), py_get_generic(Convolution3D, strategy), py_set_generic(Convolution3D, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(Convolution3D, workspace_limit), py_set_generic(Convolution3D, workspace_limit), const_cast("workspace_limit"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Convolution3D)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Convolution3D)::getstate, METH_NOARGS, "Convolution3D getstate"}, {const_cast("__setstate__"), PyOp(Convolution3D)::setstate, METH_VARARGS, "Convolution3D setstate"}, {NULL} /* Sentinel */ }; void _init_py_Convolution3D(py::module m) { using py_op = PyOp(Convolution3D); auto& py_type = PyOpType(Convolution3D); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Convolution3D"; py_type.tp_basicsize = sizeof(PyOp(Convolution3D)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Convolution3D"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Convolution3D_Mode(py_type); _init_py_Convolution3D_Sparse(py_type); _init_py_Convolution3D_DataType(py_type); _init_py_Convolution3D_Format(py_type); _init_py_Convolution3D_Strategy(py_type); PyType_Modified(&py_type); m.add_object("Convolution3D", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Convolution3D::typeinfo(), &py_type).second); } void _init_py_Convolution3DBackwardData_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution3DBackwardData_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution3DBackwardData_DataType(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "DataType", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution3DBackwardData_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_Convolution3DBackwardData_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Convolution3DBackwardData) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_d", serialization::dump(opdef.pad_d)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_d", serialization::dump(opdef.stride_d)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_d", serialization::dump(opdef.dilate_d)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"data_type", serialization::dump(opdef.data_type)}, {"format", serialization::dump(opdef.format)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_d"); if (iter != state.end()) { opdef.pad_d = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_d"); if (iter != state.end()) { opdef.stride_d = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_d"); if (iter != state.end()) { opdef.dilate_d = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("data_type"); if (iter != state.end()) { opdef.data_type = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Convolution3DBackwardData) int PyOp(Convolution3DBackwardData)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_d", "pad_h", "pad_w", "stride_d", "stride_h", "stride_w", "dilate_d", "dilate_h", "dilate_w", "sparse", "data_type", "format", "strategy", "workspace_limit", "scope", NULL}; PyObject *mode = NULL, *pad_d = NULL, *pad_h = NULL, *pad_w = NULL, *stride_d = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_d = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *data_type = NULL, *format = NULL, *strategy = NULL, *workspace_limit = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOOOOO", const_cast(kwlist), &mode, &pad_d, &pad_h, &pad_w, &stride_d, &stride_h, &stride_w, &dilate_d, &dilate_h, &dilate_w, &sparse, &data_type, &format, &strategy, &workspace_limit, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_d) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_d = py::cast(py::handle(pad_d)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_d) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_d = py::cast(py::handle(stride_d)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_d) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_d = py::cast(py::handle(dilate_d)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (data_type) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().data_type = py::cast(py::handle(data_type)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Convolution3DBackwardData)::py_getsetters[] = { {const_cast("mode"), py_get_generic(Convolution3DBackwardData, mode), py_set_generic(Convolution3DBackwardData, mode), const_cast("mode"), NULL}, {const_cast("pad_d"), py_get_generic(Convolution3DBackwardData, pad_d), py_set_generic(Convolution3DBackwardData, pad_d), const_cast("pad_d"), NULL}, {const_cast("pad_h"), py_get_generic(Convolution3DBackwardData, pad_h), py_set_generic(Convolution3DBackwardData, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(Convolution3DBackwardData, pad_w), py_set_generic(Convolution3DBackwardData, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_d"), py_get_generic(Convolution3DBackwardData, stride_d), py_set_generic(Convolution3DBackwardData, stride_d), const_cast("stride_d"), NULL}, {const_cast("stride_h"), py_get_generic(Convolution3DBackwardData, stride_h), py_set_generic(Convolution3DBackwardData, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(Convolution3DBackwardData, stride_w), py_set_generic(Convolution3DBackwardData, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_d"), py_get_generic(Convolution3DBackwardData, dilate_d), py_set_generic(Convolution3DBackwardData, dilate_d), const_cast("dilate_d"), NULL}, {const_cast("dilate_h"), py_get_generic(Convolution3DBackwardData, dilate_h), py_set_generic(Convolution3DBackwardData, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(Convolution3DBackwardData, dilate_w), py_set_generic(Convolution3DBackwardData, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(Convolution3DBackwardData, sparse), py_set_generic(Convolution3DBackwardData, sparse), const_cast("sparse"), NULL}, {const_cast("data_type"), py_get_generic(Convolution3DBackwardData, data_type), py_set_generic(Convolution3DBackwardData, data_type), const_cast("data_type"), NULL}, {const_cast("format"), py_get_generic(Convolution3DBackwardData, format), py_set_generic(Convolution3DBackwardData, format), const_cast("format"), NULL}, {const_cast("strategy"), py_get_generic(Convolution3DBackwardData, strategy), py_set_generic(Convolution3DBackwardData, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(Convolution3DBackwardData, workspace_limit), py_set_generic(Convolution3DBackwardData, workspace_limit), const_cast("workspace_limit"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Convolution3DBackwardData)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Convolution3DBackwardData)::getstate, METH_NOARGS, "Convolution3DBackwardData getstate"}, {const_cast("__setstate__"), PyOp(Convolution3DBackwardData)::setstate, METH_VARARGS, "Convolution3DBackwardData setstate"}, {NULL} /* Sentinel */ }; void _init_py_Convolution3DBackwardData(py::module m) { using py_op = PyOp(Convolution3DBackwardData); auto& py_type = PyOpType(Convolution3DBackwardData); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Convolution3DBackwardData"; py_type.tp_basicsize = sizeof(PyOp(Convolution3DBackwardData)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Convolution3DBackwardData"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Convolution3DBackwardData_Mode(py_type); _init_py_Convolution3DBackwardData_Sparse(py_type); _init_py_Convolution3DBackwardData_DataType(py_type); _init_py_Convolution3DBackwardData_Format(py_type); _init_py_Convolution3DBackwardData_Strategy(py_type); PyType_Modified(&py_type); m.add_object("Convolution3DBackwardData", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Convolution3DBackwardData::typeinfo(), &py_type).second); } void _init_py_ConvolutionBackwardData_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvolutionBackwardData_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvolutionBackwardData_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvolutionBackwardData_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } void _init_py_ConvolutionBackwardData_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(ConvolutionBackwardData) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"format", serialization::dump(opdef.format)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)}, {"dtype", serialization::dump(opdef.dtype)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ConvolutionBackwardData) int PyOp(ConvolutionBackwardData)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "sparse", "format", "compute_mode", "strategy", "workspace_limit", "dtype", "scope", NULL}; PyObject *mode = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *format = NULL, *compute_mode = NULL, *strategy = NULL, *workspace_limit = NULL, *dtype = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOOO", const_cast(kwlist), &mode, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &sparse, &format, &compute_mode, &strategy, &workspace_limit, &dtype, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ConvolutionBackwardData)::py_getsetters[] = { {const_cast("mode"), py_get_generic(ConvolutionBackwardData, mode), py_set_generic(ConvolutionBackwardData, mode), const_cast("mode"), NULL}, {const_cast("pad_h"), py_get_generic(ConvolutionBackwardData, pad_h), py_set_generic(ConvolutionBackwardData, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(ConvolutionBackwardData, pad_w), py_set_generic(ConvolutionBackwardData, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(ConvolutionBackwardData, stride_h), py_set_generic(ConvolutionBackwardData, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(ConvolutionBackwardData, stride_w), py_set_generic(ConvolutionBackwardData, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(ConvolutionBackwardData, dilate_h), py_set_generic(ConvolutionBackwardData, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(ConvolutionBackwardData, dilate_w), py_set_generic(ConvolutionBackwardData, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(ConvolutionBackwardData, sparse), py_set_generic(ConvolutionBackwardData, sparse), const_cast("sparse"), NULL}, {const_cast("format"), py_get_generic(ConvolutionBackwardData, format), py_set_generic(ConvolutionBackwardData, format), const_cast("format"), NULL}, {const_cast("compute_mode"), py_get_generic(ConvolutionBackwardData, compute_mode), py_set_generic(ConvolutionBackwardData, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("strategy"), py_get_generic(ConvolutionBackwardData, strategy), py_set_generic(ConvolutionBackwardData, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(ConvolutionBackwardData, workspace_limit), py_set_generic(ConvolutionBackwardData, workspace_limit), const_cast("workspace_limit"), NULL}, {const_cast("dtype"), py_get_generic(ConvolutionBackwardData, dtype), py_set_generic(ConvolutionBackwardData, dtype), const_cast("dtype"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ConvolutionBackwardData)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ConvolutionBackwardData)::getstate, METH_NOARGS, "ConvolutionBackwardData getstate"}, {const_cast("__setstate__"), PyOp(ConvolutionBackwardData)::setstate, METH_VARARGS, "ConvolutionBackwardData setstate"}, {NULL} /* Sentinel */ }; void _init_py_ConvolutionBackwardData(py::module m) { using py_op = PyOp(ConvolutionBackwardData); auto& py_type = PyOpType(ConvolutionBackwardData); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ConvolutionBackwardData"; py_type.tp_basicsize = sizeof(PyOp(ConvolutionBackwardData)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ConvolutionBackwardData"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_ConvolutionBackwardData_Mode(py_type); _init_py_ConvolutionBackwardData_Sparse(py_type); _init_py_ConvolutionBackwardData_Format(py_type); _init_py_ConvolutionBackwardData_ComputeMode(py_type); _init_py_ConvolutionBackwardData_Strategy(py_type); PyType_Modified(&py_type); m.add_object("ConvolutionBackwardData", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ConvolutionBackwardData::typeinfo(), &py_type).second); } PyOpDefBegin(Copy) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"comp_node", serialization::dump(opdef.comp_node)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Copy) int PyOp(Copy)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"comp_node", "scope", NULL}; PyObject *comp_node = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &comp_node, &scope)) return -1; if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Copy)::py_getsetters[] = { {const_cast("comp_node"), py_get_generic(Copy, comp_node), py_set_generic(Copy, comp_node), const_cast("comp_node"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Copy)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Copy)::getstate, METH_NOARGS, "Copy getstate"}, {const_cast("__setstate__"), PyOp(Copy)::setstate, METH_VARARGS, "Copy setstate"}, {NULL} /* Sentinel */ }; void _init_py_Copy(py::module m) { using py_op = PyOp(Copy); auto& py_type = PyOpType(Copy); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Copy"; py_type.tp_basicsize = sizeof(PyOp(Copy)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Copy"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Copy", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Copy::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Correlation.Format"; static constexpr std::underlying_type_t max = 20 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"NCHW", "NHWC", "NHWCD4", "NCHW4", "NCHW8", "NCHW32", "NCHW88", "NCHW44", "NCHW44_DOT", "NCHW_WINOGRAD", "NCHW88_WINOGRAD", "NCHW44_WINOGRAD", "NCHW4_NCHW32", "NCHW32_NCHW4", "NCHW4_NCHW", "NHWC_NCHW", "NHWC_NCHW4_IC_SMALL", "NCHW_NCHW4_IC_SMALL", "CHWN4", "NCHW4_NHWC"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("NCHW"), Correlation::Format::NCHW}, {normalize_enum("NHWC"), Correlation::Format::NHWC}, {normalize_enum("NHWCD4"), Correlation::Format::NHWCD4}, {normalize_enum("NCHW4"), Correlation::Format::NCHW4}, {normalize_enum("NCHW8"), Correlation::Format::NCHW8}, {normalize_enum("NCHW32"), Correlation::Format::NCHW32}, {normalize_enum("NCHW88"), Correlation::Format::NCHW88}, {normalize_enum("NCHW44"), Correlation::Format::NCHW44}, {normalize_enum("NCHW44_DOT"), Correlation::Format::NCHW44_DOT}, {normalize_enum("NCHW_WINOGRAD"), Correlation::Format::NCHW_WINOGRAD}, {normalize_enum("NCHW88_WINOGRAD"), Correlation::Format::NCHW88_WINOGRAD}, {normalize_enum("NCHW44_WINOGRAD"), Correlation::Format::NCHW44_WINOGRAD}, {normalize_enum("NCHW4_NCHW32"), Correlation::Format::NCHW4_NCHW32}, {normalize_enum("NCHW32_NCHW4"), Correlation::Format::NCHW32_NCHW4}, {normalize_enum("NCHW4_NCHW"), Correlation::Format::NCHW4_NCHW}, {normalize_enum("NHWC_NCHW"), Correlation::Format::NHWC_NCHW}, {normalize_enum("NHWC_NCHW4_IC_SMALL"), Correlation::Format::NHWC_NCHW4_IC_SMALL}, {normalize_enum("NCHW_NCHW4_IC_SMALL"), Correlation::Format::NCHW_NCHW4_IC_SMALL}, {normalize_enum("CHWN4"), Correlation::Format::CHWN4}, {normalize_enum("NCHW4_NHWC"), Correlation::Format::NCHW4_NHWC}}; template<> PyObject* EnumWrapper::pyobj_insts[20] = {nullptr}; void _init_py_Correlation_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Correlation.Format", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Format").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Correlation.Format").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NHWC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWC", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NHWCD4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWCD4", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW8", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW32", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW88; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW88", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW44; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW44", inst) >= 0); EnumWrapper::pyobj_insts[7] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW44_DOT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW44_DOT", inst) >= 0); EnumWrapper::pyobj_insts[8] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW_WINOGRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW_WINOGRAD", inst) >= 0); EnumWrapper::pyobj_insts[9] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW88_WINOGRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW88_WINOGRAD", inst) >= 0); EnumWrapper::pyobj_insts[10] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW44_WINOGRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW44_WINOGRAD", inst) >= 0); EnumWrapper::pyobj_insts[11] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW4_NCHW32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4_NCHW32", inst) >= 0); EnumWrapper::pyobj_insts[12] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW32_NCHW4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW32_NCHW4", inst) >= 0); EnumWrapper::pyobj_insts[13] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW4_NCHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4_NCHW", inst) >= 0); EnumWrapper::pyobj_insts[14] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NHWC_NCHW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWC_NCHW", inst) >= 0); EnumWrapper::pyobj_insts[15] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NHWC_NCHW4_IC_SMALL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NHWC_NCHW4_IC_SMALL", inst) >= 0); EnumWrapper::pyobj_insts[16] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW_NCHW4_IC_SMALL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW_NCHW4_IC_SMALL", inst) >= 0); EnumWrapper::pyobj_insts[17] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::CHWN4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CHWN4", inst) >= 0); EnumWrapper::pyobj_insts[18] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Correlation::Format::NCHW4_NHWC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NCHW4_NHWC", inst) >= 0); EnumWrapper::pyobj_insts[19] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Correlation) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"format", serialization::dump(opdef.format)}, {"kernel_size", serialization::dump(opdef.kernel_size)}, {"max_displacement", serialization::dump(opdef.max_displacement)}, {"stride1", serialization::dump(opdef.stride1)}, {"stride2", serialization::dump(opdef.stride2)}, {"pad_size", serialization::dump(opdef.pad_size)}, {"is_multiply", serialization::dump(opdef.is_multiply)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("kernel_size"); if (iter != state.end()) { opdef.kernel_size = serialization::load(iter->second); } } { auto&& iter = state.find("max_displacement"); if (iter != state.end()) { opdef.max_displacement = serialization::load(iter->second); } } { auto&& iter = state.find("stride1"); if (iter != state.end()) { opdef.stride1 = serialization::load(iter->second); } } { auto&& iter = state.find("stride2"); if (iter != state.end()) { opdef.stride2 = serialization::load(iter->second); } } { auto&& iter = state.find("pad_size"); if (iter != state.end()) { opdef.pad_size = serialization::load(iter->second); } } { auto&& iter = state.find("is_multiply"); if (iter != state.end()) { opdef.is_multiply = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Correlation) int PyOp(Correlation)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"format", "kernel_size", "max_displacement", "stride1", "stride2", "pad_size", "is_multiply", "scope", NULL}; PyObject *format = NULL, *kernel_size = NULL, *max_displacement = NULL, *stride1 = NULL, *stride2 = NULL, *pad_size = NULL, *is_multiply = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOO", const_cast(kwlist), &format, &kernel_size, &max_displacement, &stride1, &stride2, &pad_size, &is_multiply, &scope)) return -1; if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (kernel_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().kernel_size = py::cast(py::handle(kernel_size)); } CATCH_ALL(-1) } if (max_displacement) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().max_displacement = py::cast(py::handle(max_displacement)); } CATCH_ALL(-1) } if (stride1) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride1 = py::cast(py::handle(stride1)); } CATCH_ALL(-1) } if (stride2) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride2 = py::cast(py::handle(stride2)); } CATCH_ALL(-1) } if (pad_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_size = py::cast(py::handle(pad_size)); } CATCH_ALL(-1) } if (is_multiply) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().is_multiply = py::cast(py::handle(is_multiply)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Correlation)::py_getsetters[] = { {const_cast("format"), py_get_generic(Correlation, format), py_set_generic(Correlation, format), const_cast("format"), NULL}, {const_cast("kernel_size"), py_get_generic(Correlation, kernel_size), py_set_generic(Correlation, kernel_size), const_cast("kernel_size"), NULL}, {const_cast("max_displacement"), py_get_generic(Correlation, max_displacement), py_set_generic(Correlation, max_displacement), const_cast("max_displacement"), NULL}, {const_cast("stride1"), py_get_generic(Correlation, stride1), py_set_generic(Correlation, stride1), const_cast("stride1"), NULL}, {const_cast("stride2"), py_get_generic(Correlation, stride2), py_set_generic(Correlation, stride2), const_cast("stride2"), NULL}, {const_cast("pad_size"), py_get_generic(Correlation, pad_size), py_set_generic(Correlation, pad_size), const_cast("pad_size"), NULL}, {const_cast("is_multiply"), py_get_generic(Correlation, is_multiply), py_set_generic(Correlation, is_multiply), const_cast("is_multiply"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Correlation)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Correlation)::getstate, METH_NOARGS, "Correlation getstate"}, {const_cast("__setstate__"), PyOp(Correlation)::setstate, METH_VARARGS, "Correlation setstate"}, {NULL} /* Sentinel */ }; void _init_py_Correlation(py::module m) { using py_op = PyOp(Correlation); auto& py_type = PyOpType(Correlation); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Correlation"; py_type.tp_basicsize = sizeof(PyOp(Correlation)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Correlation"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Correlation_Format(py_type); PyType_Modified(&py_type); m.add_object("Correlation", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Correlation::typeinfo(), &py_type).second); } PyOpDefBegin(Cumsum) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)}, {"exclusive", serialization::dump(opdef.exclusive)}, {"reverse", serialization::dump(opdef.reverse)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("exclusive"); if (iter != state.end()) { opdef.exclusive = serialization::load(iter->second); } } { auto&& iter = state.find("reverse"); if (iter != state.end()) { opdef.reverse = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Cumsum) int PyOp(Cumsum)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "exclusive", "reverse", "scope", NULL}; PyObject *axis = NULL, *exclusive = NULL, *reverse = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOO", const_cast(kwlist), &axis, &exclusive, &reverse, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (exclusive) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().exclusive = py::cast(py::handle(exclusive)); } CATCH_ALL(-1) } if (reverse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().reverse = py::cast(py::handle(reverse)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Cumsum)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Cumsum, axis), py_set_generic(Cumsum, axis), const_cast("axis"), NULL}, {const_cast("exclusive"), py_get_generic(Cumsum, exclusive), py_set_generic(Cumsum, exclusive), const_cast("exclusive"), NULL}, {const_cast("reverse"), py_get_generic(Cumsum, reverse), py_set_generic(Cumsum, reverse), const_cast("reverse"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Cumsum)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Cumsum)::getstate, METH_NOARGS, "Cumsum getstate"}, {const_cast("__setstate__"), PyOp(Cumsum)::setstate, METH_VARARGS, "Cumsum setstate"}, {NULL} /* Sentinel */ }; void _init_py_Cumsum(py::module m) { using py_op = PyOp(Cumsum); auto& py_type = PyOpType(Cumsum); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Cumsum"; py_type.tp_basicsize = sizeof(PyOp(Cumsum)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Cumsum"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Cumsum", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Cumsum::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "CvtColor.Mode"; static constexpr std::underlying_type_t max = 32 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"RGB2GRAY", "RGB2YUV", "YUV2RGB", "GRAY2RGB", "RGBA2RGB", "RGBA2BGR", "RGBA2GRAY", "RGB2BGR", "BGR2GRAY", "BGR2RGB", "YUV2GRAY_NV21", "YUV2RGB_NV21", "YUV2BGR_NV21", "YUV2GRAY_NV12", "YUV2RGB_NV12", "YUV2BGR_NV12", "YUV2GRAY_YV12", "YUV2RGB_YV12", "YUV2BGR_YV12", "YUV2GRAY_YU12", "YUV2RGB_YU12", "YUV2BGR_YU12", "YCrCb2RGB", "YCrCb2BGR", "BT601_YUV2RGB_NV21", "BT601_YUV2BGR_NV21", "BT601_YUV2RGB_NV12", "BT601_YUV2BGR_NV12", "BT601_YUV2RGB_YV12", "BT601_YUV2BGR_YV12", "BT601_YUV2RGB_YU12", "BT601_YUV2BGR_YU12"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("RGB2GRAY"), CvtColor::Mode::RGB2GRAY}, {normalize_enum("RGB2YUV"), CvtColor::Mode::RGB2YUV}, {normalize_enum("YUV2RGB"), CvtColor::Mode::YUV2RGB}, {normalize_enum("GRAY2RGB"), CvtColor::Mode::GRAY2RGB}, {normalize_enum("RGBA2RGB"), CvtColor::Mode::RGBA2RGB}, {normalize_enum("RGBA2BGR"), CvtColor::Mode::RGBA2BGR}, {normalize_enum("RGBA2GRAY"), CvtColor::Mode::RGBA2GRAY}, {normalize_enum("RGB2BGR"), CvtColor::Mode::RGB2BGR}, {normalize_enum("BGR2GRAY"), CvtColor::Mode::BGR2GRAY}, {normalize_enum("BGR2RGB"), CvtColor::Mode::BGR2RGB}, {normalize_enum("YUV2GRAY_NV21"), CvtColor::Mode::YUV2GRAY_NV21}, {normalize_enum("YUV2RGB_NV21"), CvtColor::Mode::YUV2RGB_NV21}, {normalize_enum("YUV2BGR_NV21"), CvtColor::Mode::YUV2BGR_NV21}, {normalize_enum("YUV2GRAY_NV12"), CvtColor::Mode::YUV2GRAY_NV12}, {normalize_enum("YUV2RGB_NV12"), CvtColor::Mode::YUV2RGB_NV12}, {normalize_enum("YUV2BGR_NV12"), CvtColor::Mode::YUV2BGR_NV12}, {normalize_enum("YUV2GRAY_YV12"), CvtColor::Mode::YUV2GRAY_YV12}, {normalize_enum("YUV2RGB_YV12"), CvtColor::Mode::YUV2RGB_YV12}, {normalize_enum("YUV2BGR_YV12"), CvtColor::Mode::YUV2BGR_YV12}, {normalize_enum("YUV2GRAY_YU12"), CvtColor::Mode::YUV2GRAY_YU12}, {normalize_enum("YUV2RGB_YU12"), CvtColor::Mode::YUV2RGB_YU12}, {normalize_enum("YUV2BGR_YU12"), CvtColor::Mode::YUV2BGR_YU12}, {normalize_enum("YCrCb2RGB"), CvtColor::Mode::YCrCb2RGB}, {normalize_enum("YCrCb2BGR"), CvtColor::Mode::YCrCb2BGR}, {normalize_enum("BT601_YUV2RGB_NV21"), CvtColor::Mode::BT601_YUV2RGB_NV21}, {normalize_enum("BT601_YUV2BGR_NV21"), CvtColor::Mode::BT601_YUV2BGR_NV21}, {normalize_enum("BT601_YUV2RGB_NV12"), CvtColor::Mode::BT601_YUV2RGB_NV12}, {normalize_enum("BT601_YUV2BGR_NV12"), CvtColor::Mode::BT601_YUV2BGR_NV12}, {normalize_enum("BT601_YUV2RGB_YV12"), CvtColor::Mode::BT601_YUV2RGB_YV12}, {normalize_enum("BT601_YUV2BGR_YV12"), CvtColor::Mode::BT601_YUV2BGR_YV12}, {normalize_enum("BT601_YUV2RGB_YU12"), CvtColor::Mode::BT601_YUV2RGB_YU12}, {normalize_enum("BT601_YUV2BGR_YU12"), CvtColor::Mode::BT601_YUV2BGR_YU12}}; template<> PyObject* EnumWrapper::pyobj_insts[32] = {nullptr}; void _init_py_CvtColor_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.CvtColor.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("CvtColor.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::RGB2GRAY; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RGB2GRAY", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::RGB2YUV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RGB2YUV", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2RGB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2RGB", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::GRAY2RGB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "GRAY2RGB", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::RGBA2RGB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RGBA2RGB", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::RGBA2BGR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RGBA2BGR", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::RGBA2GRAY; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RGBA2GRAY", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::RGB2BGR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RGB2BGR", inst) >= 0); EnumWrapper::pyobj_insts[7] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BGR2GRAY; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BGR2GRAY", inst) >= 0); EnumWrapper::pyobj_insts[8] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BGR2RGB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BGR2RGB", inst) >= 0); EnumWrapper::pyobj_insts[9] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2GRAY_NV21; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2GRAY_NV21", inst) >= 0); EnumWrapper::pyobj_insts[10] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2RGB_NV21; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2RGB_NV21", inst) >= 0); EnumWrapper::pyobj_insts[11] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2BGR_NV21; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2BGR_NV21", inst) >= 0); EnumWrapper::pyobj_insts[12] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2GRAY_NV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2GRAY_NV12", inst) >= 0); EnumWrapper::pyobj_insts[13] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2RGB_NV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2RGB_NV12", inst) >= 0); EnumWrapper::pyobj_insts[14] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2BGR_NV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2BGR_NV12", inst) >= 0); EnumWrapper::pyobj_insts[15] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2GRAY_YV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2GRAY_YV12", inst) >= 0); EnumWrapper::pyobj_insts[16] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2RGB_YV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2RGB_YV12", inst) >= 0); EnumWrapper::pyobj_insts[17] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2BGR_YV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2BGR_YV12", inst) >= 0); EnumWrapper::pyobj_insts[18] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2GRAY_YU12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2GRAY_YU12", inst) >= 0); EnumWrapper::pyobj_insts[19] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2RGB_YU12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2RGB_YU12", inst) >= 0); EnumWrapper::pyobj_insts[20] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YUV2BGR_YU12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YUV2BGR_YU12", inst) >= 0); EnumWrapper::pyobj_insts[21] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YCrCb2RGB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YCrCb2RGB", inst) >= 0); EnumWrapper::pyobj_insts[22] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::YCrCb2BGR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "YCrCb2BGR", inst) >= 0); EnumWrapper::pyobj_insts[23] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2RGB_NV21; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2RGB_NV21", inst) >= 0); EnumWrapper::pyobj_insts[24] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2BGR_NV21; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2BGR_NV21", inst) >= 0); EnumWrapper::pyobj_insts[25] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2RGB_NV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2RGB_NV12", inst) >= 0); EnumWrapper::pyobj_insts[26] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2BGR_NV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2BGR_NV12", inst) >= 0); EnumWrapper::pyobj_insts[27] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2RGB_YV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2RGB_YV12", inst) >= 0); EnumWrapper::pyobj_insts[28] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2BGR_YV12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2BGR_YV12", inst) >= 0); EnumWrapper::pyobj_insts[29] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2RGB_YU12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2RGB_YU12", inst) >= 0); EnumWrapper::pyobj_insts[30] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = CvtColor::Mode::BT601_YUV2BGR_YU12; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "BT601_YUV2BGR_YU12", inst) >= 0); EnumWrapper::pyobj_insts[31] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(CvtColor) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(CvtColor) int PyOp(CvtColor)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "scope", NULL}; PyObject *mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &mode, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(CvtColor)::py_getsetters[] = { {const_cast("mode"), py_get_generic(CvtColor, mode), py_set_generic(CvtColor, mode), const_cast("mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(CvtColor)::tp_methods[] = { {const_cast("__getstate__"), PyOp(CvtColor)::getstate, METH_NOARGS, "CvtColor getstate"}, {const_cast("__setstate__"), PyOp(CvtColor)::setstate, METH_VARARGS, "CvtColor setstate"}, {NULL} /* Sentinel */ }; void _init_py_CvtColor(py::module m) { using py_op = PyOp(CvtColor); auto& py_type = PyOpType(CvtColor); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.CvtColor"; py_type.tp_basicsize = sizeof(PyOp(CvtColor)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "CvtColor"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_CvtColor_Mode(py_type); PyType_Modified(&py_type); m.add_object("CvtColor", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(CvtColor::typeinfo(), &py_type).second); } void _init_py_DeformableConv_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_DeformableConv_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_DeformableConv_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_DeformableConv_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } void _init_py_DeformableConv_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(DeformableConv) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"format", serialization::dump(opdef.format)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(DeformableConv) int PyOp(DeformableConv)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "sparse", "format", "compute_mode", "strategy", "workspace_limit", "scope", NULL}; PyObject *mode = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *format = NULL, *compute_mode = NULL, *strategy = NULL, *workspace_limit = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOO", const_cast(kwlist), &mode, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &sparse, &format, &compute_mode, &strategy, &workspace_limit, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(DeformableConv)::py_getsetters[] = { {const_cast("mode"), py_get_generic(DeformableConv, mode), py_set_generic(DeformableConv, mode), const_cast("mode"), NULL}, {const_cast("pad_h"), py_get_generic(DeformableConv, pad_h), py_set_generic(DeformableConv, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(DeformableConv, pad_w), py_set_generic(DeformableConv, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(DeformableConv, stride_h), py_set_generic(DeformableConv, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(DeformableConv, stride_w), py_set_generic(DeformableConv, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(DeformableConv, dilate_h), py_set_generic(DeformableConv, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(DeformableConv, dilate_w), py_set_generic(DeformableConv, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(DeformableConv, sparse), py_set_generic(DeformableConv, sparse), const_cast("sparse"), NULL}, {const_cast("format"), py_get_generic(DeformableConv, format), py_set_generic(DeformableConv, format), const_cast("format"), NULL}, {const_cast("compute_mode"), py_get_generic(DeformableConv, compute_mode), py_set_generic(DeformableConv, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("strategy"), py_get_generic(DeformableConv, strategy), py_set_generic(DeformableConv, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(DeformableConv, workspace_limit), py_set_generic(DeformableConv, workspace_limit), const_cast("workspace_limit"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(DeformableConv)::tp_methods[] = { {const_cast("__getstate__"), PyOp(DeformableConv)::getstate, METH_NOARGS, "DeformableConv getstate"}, {const_cast("__setstate__"), PyOp(DeformableConv)::setstate, METH_VARARGS, "DeformableConv setstate"}, {NULL} /* Sentinel */ }; void _init_py_DeformableConv(py::module m) { using py_op = PyOp(DeformableConv); auto& py_type = PyOpType(DeformableConv); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.DeformableConv"; py_type.tp_basicsize = sizeof(PyOp(DeformableConv)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "DeformableConv"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_DeformableConv_Mode(py_type); _init_py_DeformableConv_Sparse(py_type); _init_py_DeformableConv_Format(py_type); _init_py_DeformableConv_ComputeMode(py_type); _init_py_DeformableConv_Strategy(py_type); PyType_Modified(&py_type); m.add_object("DeformableConv", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(DeformableConv::typeinfo(), &py_type).second); } PyOpDefBegin(DeformablePSROIPooling) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"no_trans", serialization::dump(opdef.no_trans)}, {"spatial_scale", serialization::dump(opdef.spatial_scale)}, {"trans_std", serialization::dump(opdef.trans_std)}, {"pooled_h", serialization::dump(opdef.pooled_h)}, {"pooled_w", serialization::dump(opdef.pooled_w)}, {"part_size", serialization::dump(opdef.part_size)}, {"sample_per_part", serialization::dump(opdef.sample_per_part)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("no_trans"); if (iter != state.end()) { opdef.no_trans = serialization::load(iter->second); } } { auto&& iter = state.find("spatial_scale"); if (iter != state.end()) { opdef.spatial_scale = serialization::load(iter->second); } } { auto&& iter = state.find("trans_std"); if (iter != state.end()) { opdef.trans_std = serialization::load(iter->second); } } { auto&& iter = state.find("pooled_h"); if (iter != state.end()) { opdef.pooled_h = serialization::load(iter->second); } } { auto&& iter = state.find("pooled_w"); if (iter != state.end()) { opdef.pooled_w = serialization::load(iter->second); } } { auto&& iter = state.find("part_size"); if (iter != state.end()) { opdef.part_size = serialization::load(iter->second); } } { auto&& iter = state.find("sample_per_part"); if (iter != state.end()) { opdef.sample_per_part = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(DeformablePSROIPooling) int PyOp(DeformablePSROIPooling)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"no_trans", "spatial_scale", "trans_std", "pooled_h", "pooled_w", "part_size", "sample_per_part", "scope", NULL}; PyObject *no_trans = NULL, *spatial_scale = NULL, *trans_std = NULL, *pooled_h = NULL, *pooled_w = NULL, *part_size = NULL, *sample_per_part = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOO", const_cast(kwlist), &no_trans, &spatial_scale, &trans_std, &pooled_h, &pooled_w, &part_size, &sample_per_part, &scope)) return -1; if (no_trans) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().no_trans = py::cast(py::handle(no_trans)); } CATCH_ALL(-1) } if (spatial_scale) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().spatial_scale = py::cast(py::handle(spatial_scale)); } CATCH_ALL(-1) } if (trans_std) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().trans_std = py::cast(py::handle(trans_std)); } CATCH_ALL(-1) } if (pooled_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pooled_h = py::cast(py::handle(pooled_h)); } CATCH_ALL(-1) } if (pooled_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pooled_w = py::cast(py::handle(pooled_w)); } CATCH_ALL(-1) } if (part_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().part_size = py::cast(py::handle(part_size)); } CATCH_ALL(-1) } if (sample_per_part) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sample_per_part = py::cast(py::handle(sample_per_part)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(DeformablePSROIPooling)::py_getsetters[] = { {const_cast("no_trans"), py_get_generic(DeformablePSROIPooling, no_trans), py_set_generic(DeformablePSROIPooling, no_trans), const_cast("no_trans"), NULL}, {const_cast("spatial_scale"), py_get_generic(DeformablePSROIPooling, spatial_scale), py_set_generic(DeformablePSROIPooling, spatial_scale), const_cast("spatial_scale"), NULL}, {const_cast("trans_std"), py_get_generic(DeformablePSROIPooling, trans_std), py_set_generic(DeformablePSROIPooling, trans_std), const_cast("trans_std"), NULL}, {const_cast("pooled_h"), py_get_generic(DeformablePSROIPooling, pooled_h), py_set_generic(DeformablePSROIPooling, pooled_h), const_cast("pooled_h"), NULL}, {const_cast("pooled_w"), py_get_generic(DeformablePSROIPooling, pooled_w), py_set_generic(DeformablePSROIPooling, pooled_w), const_cast("pooled_w"), NULL}, {const_cast("part_size"), py_get_generic(DeformablePSROIPooling, part_size), py_set_generic(DeformablePSROIPooling, part_size), const_cast("part_size"), NULL}, {const_cast("sample_per_part"), py_get_generic(DeformablePSROIPooling, sample_per_part), py_set_generic(DeformablePSROIPooling, sample_per_part), const_cast("sample_per_part"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(DeformablePSROIPooling)::tp_methods[] = { {const_cast("__getstate__"), PyOp(DeformablePSROIPooling)::getstate, METH_NOARGS, "DeformablePSROIPooling getstate"}, {const_cast("__setstate__"), PyOp(DeformablePSROIPooling)::setstate, METH_VARARGS, "DeformablePSROIPooling setstate"}, {NULL} /* Sentinel */ }; void _init_py_DeformablePSROIPooling(py::module m) { using py_op = PyOp(DeformablePSROIPooling); auto& py_type = PyOpType(DeformablePSROIPooling); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.DeformablePSROIPooling"; py_type.tp_basicsize = sizeof(PyOp(DeformablePSROIPooling)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "DeformablePSROIPooling"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("DeformablePSROIPooling", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(DeformablePSROIPooling::typeinfo(), &py_type).second); } PyOpDefBegin(Diag) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"k", serialization::dump(opdef.k)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("k"); if (iter != state.end()) { opdef.k = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Diag) int PyOp(Diag)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"k", "scope", NULL}; PyObject *k = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &k, &scope)) return -1; if (k) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().k = py::cast(py::handle(k)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Diag)::py_getsetters[] = { {const_cast("k"), py_get_generic(Diag, k), py_set_generic(Diag, k), const_cast("k"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Diag)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Diag)::getstate, METH_NOARGS, "Diag getstate"}, {const_cast("__setstate__"), PyOp(Diag)::setstate, METH_VARARGS, "Diag setstate"}, {NULL} /* Sentinel */ }; void _init_py_Diag(py::module m) { using py_op = PyOp(Diag); auto& py_type = PyOpType(Diag); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Diag"; py_type.tp_basicsize = sizeof(PyOp(Diag)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Diag"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Diag", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Diag::typeinfo(), &py_type).second); } PyOpDefBegin(Dimshuffle) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"pattern", serialization::dump(opdef.pattern)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("pattern"); if (iter != state.end()) { opdef.pattern = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Dimshuffle) int PyOp(Dimshuffle)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"pattern", "scope", NULL}; PyObject *pattern = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &pattern, &scope)) return -1; if (pattern) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pattern = py::cast(py::handle(pattern)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Dimshuffle)::py_getsetters[] = { {const_cast("pattern"), py_get_generic(Dimshuffle, pattern), py_set_generic(Dimshuffle, pattern), const_cast("pattern"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Dimshuffle)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Dimshuffle)::getstate, METH_NOARGS, "Dimshuffle getstate"}, {const_cast("__setstate__"), PyOp(Dimshuffle)::setstate, METH_VARARGS, "Dimshuffle setstate"}, {NULL} /* Sentinel */ }; void _init_py_Dimshuffle(py::module m) { using py_op = PyOp(Dimshuffle); auto& py_type = PyOpType(Dimshuffle); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Dimshuffle"; py_type.tp_basicsize = sizeof(PyOp(Dimshuffle)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Dimshuffle"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Dimshuffle", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Dimshuffle::typeinfo(), &py_type).second); } PyOpDefBegin(Dot) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Dot) int PyOp(Dot)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(Dot)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(Dot)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Dot)::getstate, METH_NOARGS, "Dot getstate"}, {const_cast("__setstate__"), PyOp(Dot)::setstate, METH_VARARGS, "Dot setstate"}, {NULL} /* Sentinel */ }; void _init_py_Dot(py::module m) { using py_op = PyOp(Dot); auto& py_type = PyOpType(Dot); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Dot"; py_type.tp_basicsize = sizeof(PyOp(Dot)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Dot"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Dot", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Dot::typeinfo(), &py_type).second); } PyOpDefBegin(Dropout) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"drop_prob", serialization::dump(opdef.drop_prob)}, {"seed", serialization::dump(opdef.seed)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("drop_prob"); if (iter != state.end()) { opdef.drop_prob = serialization::load(iter->second); } } { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Dropout) int PyOp(Dropout)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"drop_prob", "seed", "handle", "scope", NULL}; PyObject *drop_prob = NULL, *seed = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOO", const_cast(kwlist), &drop_prob, &seed, &handle, &scope)) return -1; if (drop_prob) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().drop_prob = py::cast(py::handle(drop_prob)); } CATCH_ALL(-1) } if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Dropout)::py_getsetters[] = { {const_cast("drop_prob"), py_get_generic(Dropout, drop_prob), py_set_generic(Dropout, drop_prob), const_cast("drop_prob"), NULL}, {const_cast("seed"), py_get_generic(Dropout, seed), py_set_generic(Dropout, seed), const_cast("seed"), NULL}, {const_cast("handle"), py_get_generic(Dropout, handle), py_set_generic(Dropout, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Dropout)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Dropout)::getstate, METH_NOARGS, "Dropout getstate"}, {const_cast("__setstate__"), PyOp(Dropout)::setstate, METH_VARARGS, "Dropout setstate"}, {NULL} /* Sentinel */ }; void _init_py_Dropout(py::module m) { using py_op = PyOp(Dropout); auto& py_type = PyOpType(Dropout); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Dropout"; py_type.tp_basicsize = sizeof(PyOp(Dropout)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Dropout"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Dropout", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Dropout::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Elemwise.Mode"; static constexpr std::underlying_type_t max = 64 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"RELU", "ABS", "ACOS", "ASIN", "CEIL", "COS", "EXP", "EXPM1", "FLOOR", "LOG", "LOG1P", "NEGATE", "SIGMOID", "SIN", "TANH", "ABS_GRAD", "ADD", "FLOOR_DIV", "MAX", "MIN", "MOD", "MUL", "POW", "SIGMOID_GRAD", "SUB", "SWITCH_GT0", "TANH_GRAD", "TRUE_DIV", "LOG_SUM_EXP", "LT", "LEQ", "EQ", "SHL", "SHR", "COND_LEQ_MOV", "FUSE_MUL_ADD3", "FUSE_MUL_ADD4", "FUSE_ADD_RELU", "FUSE_ADD_SIGMOID", "FUSE_ADD_TANH", "FAST_TANH", "FAST_TANH_GRAD", "ROUND", "RMULH", "ATAN2", "ERF", "ERFINV", "ERFC", "ERFCINV", "H_SWISH", "H_SWISH_GRAD", "FUSE_ADD_H_SWISH", "NOT", "AND", "OR", "XOR", "SILU", "SILU_GRAD", "GELU", "GELU_GRAD", "COND_LT_MOV", "NEQ", "ISNAN", "ISINF"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("RELU"), Elemwise::Mode::RELU}, {normalize_enum("ABS"), Elemwise::Mode::ABS}, {normalize_enum("ACOS"), Elemwise::Mode::ACOS}, {normalize_enum("ASIN"), Elemwise::Mode::ASIN}, {normalize_enum("CEIL"), Elemwise::Mode::CEIL}, {normalize_enum("COS"), Elemwise::Mode::COS}, {normalize_enum("EXP"), Elemwise::Mode::EXP}, {normalize_enum("EXPM1"), Elemwise::Mode::EXPM1}, {normalize_enum("FLOOR"), Elemwise::Mode::FLOOR}, {normalize_enum("LOG"), Elemwise::Mode::LOG}, {normalize_enum("LOG1P"), Elemwise::Mode::LOG1P}, {normalize_enum("NEGATE"), Elemwise::Mode::NEGATE}, {normalize_enum("SIGMOID"), Elemwise::Mode::SIGMOID}, {normalize_enum("SIN"), Elemwise::Mode::SIN}, {normalize_enum("TANH"), Elemwise::Mode::TANH}, {normalize_enum("ABS_GRAD"), Elemwise::Mode::ABS_GRAD}, {normalize_enum("ADD"), Elemwise::Mode::ADD}, {normalize_enum("FLOOR_DIV"), Elemwise::Mode::FLOOR_DIV}, {normalize_enum("MAX"), Elemwise::Mode::MAX}, {normalize_enum("MIN"), Elemwise::Mode::MIN}, {normalize_enum("MOD"), Elemwise::Mode::MOD}, {normalize_enum("MUL"), Elemwise::Mode::MUL}, {normalize_enum("POW"), Elemwise::Mode::POW}, {normalize_enum("SIGMOID_GRAD"), Elemwise::Mode::SIGMOID_GRAD}, {normalize_enum("SUB"), Elemwise::Mode::SUB}, {normalize_enum("SWITCH_GT0"), Elemwise::Mode::SWITCH_GT0}, {normalize_enum("TANH_GRAD"), Elemwise::Mode::TANH_GRAD}, {normalize_enum("TRUE_DIV"), Elemwise::Mode::TRUE_DIV}, {normalize_enum("LOG_SUM_EXP"), Elemwise::Mode::LOG_SUM_EXP}, {normalize_enum("LT"), Elemwise::Mode::LT}, {normalize_enum("LEQ"), Elemwise::Mode::LEQ}, {normalize_enum("EQ"), Elemwise::Mode::EQ}, {normalize_enum("SHL"), Elemwise::Mode::SHL}, {normalize_enum("SHR"), Elemwise::Mode::SHR}, {normalize_enum("COND_LEQ_MOV"), Elemwise::Mode::COND_LEQ_MOV}, {normalize_enum("FUSE_MUL_ADD3"), Elemwise::Mode::FUSE_MUL_ADD3}, {normalize_enum("FUSE_MUL_ADD4"), Elemwise::Mode::FUSE_MUL_ADD4}, {normalize_enum("FUSE_ADD_RELU"), Elemwise::Mode::FUSE_ADD_RELU}, {normalize_enum("FUSE_ADD_SIGMOID"), Elemwise::Mode::FUSE_ADD_SIGMOID}, {normalize_enum("FUSE_ADD_TANH"), Elemwise::Mode::FUSE_ADD_TANH}, {normalize_enum("FAST_TANH"), Elemwise::Mode::FAST_TANH}, {normalize_enum("FAST_TANH_GRAD"), Elemwise::Mode::FAST_TANH_GRAD}, {normalize_enum("ROUND"), Elemwise::Mode::ROUND}, {normalize_enum("RMULH"), Elemwise::Mode::RMULH}, {normalize_enum("ATAN2"), Elemwise::Mode::ATAN2}, {normalize_enum("ERF"), Elemwise::Mode::ERF}, {normalize_enum("ERFINV"), Elemwise::Mode::ERFINV}, {normalize_enum("ERFC"), Elemwise::Mode::ERFC}, {normalize_enum("ERFCINV"), Elemwise::Mode::ERFCINV}, {normalize_enum("H_SWISH"), Elemwise::Mode::H_SWISH}, {normalize_enum("H_SWISH_GRAD"), Elemwise::Mode::H_SWISH_GRAD}, {normalize_enum("FUSE_ADD_H_SWISH"), Elemwise::Mode::FUSE_ADD_H_SWISH}, {normalize_enum("NOT"), Elemwise::Mode::NOT}, {normalize_enum("AND"), Elemwise::Mode::AND}, {normalize_enum("OR"), Elemwise::Mode::OR}, {normalize_enum("XOR"), Elemwise::Mode::XOR}, {normalize_enum("SILU"), Elemwise::Mode::SILU}, {normalize_enum("SILU_GRAD"), Elemwise::Mode::SILU_GRAD}, {normalize_enum("GELU"), Elemwise::Mode::GELU}, {normalize_enum("GELU_GRAD"), Elemwise::Mode::GELU_GRAD}, {normalize_enum("COND_LT_MOV"), Elemwise::Mode::COND_LT_MOV}, {normalize_enum("NEQ"), Elemwise::Mode::NEQ}, {normalize_enum("ISNAN"), Elemwise::Mode::ISNAN}, {normalize_enum("ISINF"), Elemwise::Mode::ISINF}}; template<> PyObject* EnumWrapper::pyobj_insts[64] = {nullptr}; void _init_py_Elemwise_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Elemwise.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Elemwise.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::RELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RELU", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ABS; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ABS", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ACOS; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ACOS", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ASIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ASIN", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::CEIL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CEIL", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::COS; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "COS", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::EXP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "EXP", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::EXPM1; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "EXPM1", inst) >= 0); EnumWrapper::pyobj_insts[7] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FLOOR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOOR", inst) >= 0); EnumWrapper::pyobj_insts[8] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::LOG; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LOG", inst) >= 0); EnumWrapper::pyobj_insts[9] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::LOG1P; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LOG1P", inst) >= 0); EnumWrapper::pyobj_insts[10] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::NEGATE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NEGATE", inst) >= 0); EnumWrapper::pyobj_insts[11] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SIGMOID; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SIGMOID", inst) >= 0); EnumWrapper::pyobj_insts[12] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SIN", inst) >= 0); EnumWrapper::pyobj_insts[13] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::TANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "TANH", inst) >= 0); EnumWrapper::pyobj_insts[14] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ABS_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ABS_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[15] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ADD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ADD", inst) >= 0); EnumWrapper::pyobj_insts[16] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FLOOR_DIV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOOR_DIV", inst) >= 0); EnumWrapper::pyobj_insts[17] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::MAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MAX", inst) >= 0); EnumWrapper::pyobj_insts[18] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::MIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MIN", inst) >= 0); EnumWrapper::pyobj_insts[19] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::MOD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MOD", inst) >= 0); EnumWrapper::pyobj_insts[20] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::MUL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MUL", inst) >= 0); EnumWrapper::pyobj_insts[21] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::POW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "POW", inst) >= 0); EnumWrapper::pyobj_insts[22] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SIGMOID_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SIGMOID_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[23] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SUB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SUB", inst) >= 0); EnumWrapper::pyobj_insts[24] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SWITCH_GT0; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SWITCH_GT0", inst) >= 0); EnumWrapper::pyobj_insts[25] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::TANH_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "TANH_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[26] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::TRUE_DIV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "TRUE_DIV", inst) >= 0); EnumWrapper::pyobj_insts[27] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::LOG_SUM_EXP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LOG_SUM_EXP", inst) >= 0); EnumWrapper::pyobj_insts[28] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::LT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LT", inst) >= 0); EnumWrapper::pyobj_insts[29] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::LEQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LEQ", inst) >= 0); EnumWrapper::pyobj_insts[30] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::EQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "EQ", inst) >= 0); EnumWrapper::pyobj_insts[31] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SHL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SHL", inst) >= 0); EnumWrapper::pyobj_insts[32] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SHR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SHR", inst) >= 0); EnumWrapper::pyobj_insts[33] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::COND_LEQ_MOV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "COND_LEQ_MOV", inst) >= 0); EnumWrapper::pyobj_insts[34] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FUSE_MUL_ADD3; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_MUL_ADD3", inst) >= 0); EnumWrapper::pyobj_insts[35] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FUSE_MUL_ADD4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_MUL_ADD4", inst) >= 0); EnumWrapper::pyobj_insts[36] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FUSE_ADD_RELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_ADD_RELU", inst) >= 0); EnumWrapper::pyobj_insts[37] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FUSE_ADD_SIGMOID; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_ADD_SIGMOID", inst) >= 0); EnumWrapper::pyobj_insts[38] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FUSE_ADD_TANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_ADD_TANH", inst) >= 0); EnumWrapper::pyobj_insts[39] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FAST_TANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FAST_TANH", inst) >= 0); EnumWrapper::pyobj_insts[40] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FAST_TANH_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FAST_TANH_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[41] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ROUND; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ROUND", inst) >= 0); EnumWrapper::pyobj_insts[42] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::RMULH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RMULH", inst) >= 0); EnumWrapper::pyobj_insts[43] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ATAN2; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ATAN2", inst) >= 0); EnumWrapper::pyobj_insts[44] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ERF; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ERF", inst) >= 0); EnumWrapper::pyobj_insts[45] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ERFINV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ERFINV", inst) >= 0); EnumWrapper::pyobj_insts[46] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ERFC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ERFC", inst) >= 0); EnumWrapper::pyobj_insts[47] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ERFCINV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ERFCINV", inst) >= 0); EnumWrapper::pyobj_insts[48] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::H_SWISH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "H_SWISH", inst) >= 0); EnumWrapper::pyobj_insts[49] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::H_SWISH_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "H_SWISH_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[50] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::FUSE_ADD_H_SWISH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_ADD_H_SWISH", inst) >= 0); EnumWrapper::pyobj_insts[51] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::NOT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NOT", inst) >= 0); EnumWrapper::pyobj_insts[52] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::AND; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "AND", inst) >= 0); EnumWrapper::pyobj_insts[53] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::OR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "OR", inst) >= 0); EnumWrapper::pyobj_insts[54] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::XOR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "XOR", inst) >= 0); EnumWrapper::pyobj_insts[55] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SILU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SILU", inst) >= 0); EnumWrapper::pyobj_insts[56] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::SILU_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SILU_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[57] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::GELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "GELU", inst) >= 0); EnumWrapper::pyobj_insts[58] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::GELU_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "GELU_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[59] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::COND_LT_MOV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "COND_LT_MOV", inst) >= 0); EnumWrapper::pyobj_insts[60] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::NEQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NEQ", inst) >= 0); EnumWrapper::pyobj_insts[61] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ISNAN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ISNAN", inst) >= 0); EnumWrapper::pyobj_insts[62] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Elemwise::Mode::ISINF; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ISINF", inst) >= 0); EnumWrapper::pyobj_insts[63] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Elemwise) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Elemwise) int PyOp(Elemwise)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "scope", NULL}; PyObject *mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &mode, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Elemwise)::py_getsetters[] = { {const_cast("mode"), py_get_generic(Elemwise, mode), py_set_generic(Elemwise, mode), const_cast("mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Elemwise)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Elemwise)::getstate, METH_NOARGS, "Elemwise getstate"}, {const_cast("__setstate__"), PyOp(Elemwise)::setstate, METH_VARARGS, "Elemwise setstate"}, {NULL} /* Sentinel */ }; void _init_py_Elemwise(py::module m) { using py_op = PyOp(Elemwise); auto& py_type = PyOpType(Elemwise); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Elemwise"; py_type.tp_basicsize = sizeof(PyOp(Elemwise)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Elemwise"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Elemwise_Mode(py_type); PyType_Modified(&py_type); m.add_object("Elemwise", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Elemwise::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "ElemwiseMultiType.Mode"; static constexpr std::underlying_type_t max = 64 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"FUSE_MUL_ADD3_INT16x32x32x32", "FUSE_MUL_ADD3_IXxF32xF32xI8", "ROUND_SHR_SATURATE_IXxI8xI8", "FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT16x16x16x8", "FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT32x32x32x8", "ROUND_SHR_SATURATE_IXxI8xI16", "QADD", "QFUSE_ADD_RELU", "QMUL", "QMIN", "QMAX", "QSUB", "QTRUE_DIV", "QFUSE_ADD_SIGMOID", "QFUSE_ADD_TANH", "QRELU", "QABS", "QSIGMOID", "QEXP", "QTANH", "QFUSE_MUL_ADD3", "QFAST_TANH", "QNEGATE", "QACOS", "QASIN", "QCEIL", "QCOS", "QEXPM1", "QFLOOR", "QLOG", "QLOG1P", "QSIN", "QROUND", "QERF", "QERFINV", "QERFC", "QERFCINV", "QABS_GRAD", "QFLOOR_DIV", "QMOD", "QSIGMOID_GRAD", "QSWITCH_GT0", "QTANH_GRAD", "QLT", "QLEQ", "QEQ", "QPOW", "QLOG_SUM_EXP", "QFAST_TANH_GRAD", "QATAN2", "QCOND_LEQ_MOV", "QH_SWISH", "QFUSE_ADD_H_SWISH", "QH_SWISH_GRAD", "FUSE_MUL_ADD3_INT16xF32xF32xF32", "MUL_INT16xF32xF32", "FUSE_MUL_ADD3_UINT8xF32xF32xF32", "QCOND_LT_MOV", "EQ", "NEQ", "LT", "LEQ", "ISNAN", "ISINF"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("FUSE_MUL_ADD3_INT16x32x32x32"), ElemwiseMultiType::Mode::FUSE_MUL_ADD3_INT16x32x32x32}, {normalize_enum("FUSE_MUL_ADD3_IXxF32xF32xI8"), ElemwiseMultiType::Mode::FUSE_MUL_ADD3_IXxF32xF32xI8}, {normalize_enum("ROUND_SHR_SATURATE_IXxI8xI8"), ElemwiseMultiType::Mode::ROUND_SHR_SATURATE_IXxI8xI8}, {normalize_enum("FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT16x16x16x8"), ElemwiseMultiType::Mode::FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT16x16x16x8}, {normalize_enum("FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT32x32x32x8"), ElemwiseMultiType::Mode::FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT32x32x32x8}, {normalize_enum("ROUND_SHR_SATURATE_IXxI8xI16"), ElemwiseMultiType::Mode::ROUND_SHR_SATURATE_IXxI8xI16}, {normalize_enum("QADD"), ElemwiseMultiType::Mode::QADD}, {normalize_enum("QFUSE_ADD_RELU"), ElemwiseMultiType::Mode::QFUSE_ADD_RELU}, {normalize_enum("QMUL"), ElemwiseMultiType::Mode::QMUL}, {normalize_enum("QMIN"), ElemwiseMultiType::Mode::QMIN}, {normalize_enum("QMAX"), ElemwiseMultiType::Mode::QMAX}, {normalize_enum("QSUB"), ElemwiseMultiType::Mode::QSUB}, {normalize_enum("QTRUE_DIV"), ElemwiseMultiType::Mode::QTRUE_DIV}, {normalize_enum("QFUSE_ADD_SIGMOID"), ElemwiseMultiType::Mode::QFUSE_ADD_SIGMOID}, {normalize_enum("QFUSE_ADD_TANH"), ElemwiseMultiType::Mode::QFUSE_ADD_TANH}, {normalize_enum("QRELU"), ElemwiseMultiType::Mode::QRELU}, {normalize_enum("QABS"), ElemwiseMultiType::Mode::QABS}, {normalize_enum("QSIGMOID"), ElemwiseMultiType::Mode::QSIGMOID}, {normalize_enum("QEXP"), ElemwiseMultiType::Mode::QEXP}, {normalize_enum("QTANH"), ElemwiseMultiType::Mode::QTANH}, {normalize_enum("QFUSE_MUL_ADD3"), ElemwiseMultiType::Mode::QFUSE_MUL_ADD3}, {normalize_enum("QFAST_TANH"), ElemwiseMultiType::Mode::QFAST_TANH}, {normalize_enum("QNEGATE"), ElemwiseMultiType::Mode::QNEGATE}, {normalize_enum("QACOS"), ElemwiseMultiType::Mode::QACOS}, {normalize_enum("QASIN"), ElemwiseMultiType::Mode::QASIN}, {normalize_enum("QCEIL"), ElemwiseMultiType::Mode::QCEIL}, {normalize_enum("QCOS"), ElemwiseMultiType::Mode::QCOS}, {normalize_enum("QEXPM1"), ElemwiseMultiType::Mode::QEXPM1}, {normalize_enum("QFLOOR"), ElemwiseMultiType::Mode::QFLOOR}, {normalize_enum("QLOG"), ElemwiseMultiType::Mode::QLOG}, {normalize_enum("QLOG1P"), ElemwiseMultiType::Mode::QLOG1P}, {normalize_enum("QSIN"), ElemwiseMultiType::Mode::QSIN}, {normalize_enum("QROUND"), ElemwiseMultiType::Mode::QROUND}, {normalize_enum("QERF"), ElemwiseMultiType::Mode::QERF}, {normalize_enum("QERFINV"), ElemwiseMultiType::Mode::QERFINV}, {normalize_enum("QERFC"), ElemwiseMultiType::Mode::QERFC}, {normalize_enum("QERFCINV"), ElemwiseMultiType::Mode::QERFCINV}, {normalize_enum("QABS_GRAD"), ElemwiseMultiType::Mode::QABS_GRAD}, {normalize_enum("QFLOOR_DIV"), ElemwiseMultiType::Mode::QFLOOR_DIV}, {normalize_enum("QMOD"), ElemwiseMultiType::Mode::QMOD}, {normalize_enum("QSIGMOID_GRAD"), ElemwiseMultiType::Mode::QSIGMOID_GRAD}, {normalize_enum("QSWITCH_GT0"), ElemwiseMultiType::Mode::QSWITCH_GT0}, {normalize_enum("QTANH_GRAD"), ElemwiseMultiType::Mode::QTANH_GRAD}, {normalize_enum("QLT"), ElemwiseMultiType::Mode::QLT}, {normalize_enum("QLEQ"), ElemwiseMultiType::Mode::QLEQ}, {normalize_enum("QEQ"), ElemwiseMultiType::Mode::QEQ}, {normalize_enum("QPOW"), ElemwiseMultiType::Mode::QPOW}, {normalize_enum("QLOG_SUM_EXP"), ElemwiseMultiType::Mode::QLOG_SUM_EXP}, {normalize_enum("QFAST_TANH_GRAD"), ElemwiseMultiType::Mode::QFAST_TANH_GRAD}, {normalize_enum("QATAN2"), ElemwiseMultiType::Mode::QATAN2}, {normalize_enum("QCOND_LEQ_MOV"), ElemwiseMultiType::Mode::QCOND_LEQ_MOV}, {normalize_enum("QH_SWISH"), ElemwiseMultiType::Mode::QH_SWISH}, {normalize_enum("QFUSE_ADD_H_SWISH"), ElemwiseMultiType::Mode::QFUSE_ADD_H_SWISH}, {normalize_enum("QH_SWISH_GRAD"), ElemwiseMultiType::Mode::QH_SWISH_GRAD}, {normalize_enum("FUSE_MUL_ADD3_INT16xF32xF32xF32"), ElemwiseMultiType::Mode::FUSE_MUL_ADD3_INT16xF32xF32xF32}, {normalize_enum("MUL_INT16xF32xF32"), ElemwiseMultiType::Mode::MUL_INT16xF32xF32}, {normalize_enum("FUSE_MUL_ADD3_UINT8xF32xF32xF32"), ElemwiseMultiType::Mode::FUSE_MUL_ADD3_UINT8xF32xF32xF32}, {normalize_enum("QCOND_LT_MOV"), ElemwiseMultiType::Mode::QCOND_LT_MOV}, {normalize_enum("EQ"), ElemwiseMultiType::Mode::EQ}, {normalize_enum("NEQ"), ElemwiseMultiType::Mode::NEQ}, {normalize_enum("LT"), ElemwiseMultiType::Mode::LT}, {normalize_enum("LEQ"), ElemwiseMultiType::Mode::LEQ}, {normalize_enum("ISNAN"), ElemwiseMultiType::Mode::ISNAN}, {normalize_enum("ISINF"), ElemwiseMultiType::Mode::ISINF}}; template<> PyObject* EnumWrapper::pyobj_insts[64] = {nullptr}; void _init_py_ElemwiseMultiType_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.ElemwiseMultiType.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("ElemwiseMultiType.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::FUSE_MUL_ADD3_INT16x32x32x32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_MUL_ADD3_INT16x32x32x32", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::FUSE_MUL_ADD3_IXxF32xF32xI8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_MUL_ADD3_IXxF32xF32xI8", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::ROUND_SHR_SATURATE_IXxI8xI8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ROUND_SHR_SATURATE_IXxI8xI8", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT16x16x16x8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT16x16x16x8", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT32x32x32x8; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT32x32x32x8", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::ROUND_SHR_SATURATE_IXxI8xI16; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ROUND_SHR_SATURATE_IXxI8xI16", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QADD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QADD", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFUSE_ADD_RELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFUSE_ADD_RELU", inst) >= 0); EnumWrapper::pyobj_insts[7] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QMUL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QMUL", inst) >= 0); EnumWrapper::pyobj_insts[8] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QMIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QMIN", inst) >= 0); EnumWrapper::pyobj_insts[9] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QMAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QMAX", inst) >= 0); EnumWrapper::pyobj_insts[10] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QSUB; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QSUB", inst) >= 0); EnumWrapper::pyobj_insts[11] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QTRUE_DIV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QTRUE_DIV", inst) >= 0); EnumWrapper::pyobj_insts[12] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFUSE_ADD_SIGMOID; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFUSE_ADD_SIGMOID", inst) >= 0); EnumWrapper::pyobj_insts[13] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFUSE_ADD_TANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFUSE_ADD_TANH", inst) >= 0); EnumWrapper::pyobj_insts[14] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QRELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QRELU", inst) >= 0); EnumWrapper::pyobj_insts[15] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QABS; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QABS", inst) >= 0); EnumWrapper::pyobj_insts[16] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QSIGMOID; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QSIGMOID", inst) >= 0); EnumWrapper::pyobj_insts[17] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QEXP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QEXP", inst) >= 0); EnumWrapper::pyobj_insts[18] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QTANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QTANH", inst) >= 0); EnumWrapper::pyobj_insts[19] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFUSE_MUL_ADD3; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFUSE_MUL_ADD3", inst) >= 0); EnumWrapper::pyobj_insts[20] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFAST_TANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFAST_TANH", inst) >= 0); EnumWrapper::pyobj_insts[21] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QNEGATE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QNEGATE", inst) >= 0); EnumWrapper::pyobj_insts[22] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QACOS; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QACOS", inst) >= 0); EnumWrapper::pyobj_insts[23] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QASIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QASIN", inst) >= 0); EnumWrapper::pyobj_insts[24] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QCEIL; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QCEIL", inst) >= 0); EnumWrapper::pyobj_insts[25] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QCOS; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QCOS", inst) >= 0); EnumWrapper::pyobj_insts[26] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QEXPM1; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QEXPM1", inst) >= 0); EnumWrapper::pyobj_insts[27] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFLOOR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFLOOR", inst) >= 0); EnumWrapper::pyobj_insts[28] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QLOG; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QLOG", inst) >= 0); EnumWrapper::pyobj_insts[29] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QLOG1P; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QLOG1P", inst) >= 0); EnumWrapper::pyobj_insts[30] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QSIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QSIN", inst) >= 0); EnumWrapper::pyobj_insts[31] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QROUND; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QROUND", inst) >= 0); EnumWrapper::pyobj_insts[32] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QERF; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QERF", inst) >= 0); EnumWrapper::pyobj_insts[33] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QERFINV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QERFINV", inst) >= 0); EnumWrapper::pyobj_insts[34] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QERFC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QERFC", inst) >= 0); EnumWrapper::pyobj_insts[35] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QERFCINV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QERFCINV", inst) >= 0); EnumWrapper::pyobj_insts[36] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QABS_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QABS_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[37] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFLOOR_DIV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFLOOR_DIV", inst) >= 0); EnumWrapper::pyobj_insts[38] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QMOD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QMOD", inst) >= 0); EnumWrapper::pyobj_insts[39] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QSIGMOID_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QSIGMOID_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[40] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QSWITCH_GT0; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QSWITCH_GT0", inst) >= 0); EnumWrapper::pyobj_insts[41] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QTANH_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QTANH_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[42] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QLT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QLT", inst) >= 0); EnumWrapper::pyobj_insts[43] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QLEQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QLEQ", inst) >= 0); EnumWrapper::pyobj_insts[44] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QEQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QEQ", inst) >= 0); EnumWrapper::pyobj_insts[45] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QPOW; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QPOW", inst) >= 0); EnumWrapper::pyobj_insts[46] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QLOG_SUM_EXP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QLOG_SUM_EXP", inst) >= 0); EnumWrapper::pyobj_insts[47] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFAST_TANH_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFAST_TANH_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[48] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QATAN2; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QATAN2", inst) >= 0); EnumWrapper::pyobj_insts[49] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QCOND_LEQ_MOV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QCOND_LEQ_MOV", inst) >= 0); EnumWrapper::pyobj_insts[50] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QH_SWISH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QH_SWISH", inst) >= 0); EnumWrapper::pyobj_insts[51] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QFUSE_ADD_H_SWISH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QFUSE_ADD_H_SWISH", inst) >= 0); EnumWrapper::pyobj_insts[52] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QH_SWISH_GRAD; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QH_SWISH_GRAD", inst) >= 0); EnumWrapper::pyobj_insts[53] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::FUSE_MUL_ADD3_INT16xF32xF32xF32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_MUL_ADD3_INT16xF32xF32xF32", inst) >= 0); EnumWrapper::pyobj_insts[54] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::MUL_INT16xF32xF32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MUL_INT16xF32xF32", inst) >= 0); EnumWrapper::pyobj_insts[55] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::FUSE_MUL_ADD3_UINT8xF32xF32xF32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FUSE_MUL_ADD3_UINT8xF32xF32xF32", inst) >= 0); EnumWrapper::pyobj_insts[56] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::QCOND_LT_MOV; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QCOND_LT_MOV", inst) >= 0); EnumWrapper::pyobj_insts[57] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::EQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "EQ", inst) >= 0); EnumWrapper::pyobj_insts[58] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::NEQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NEQ", inst) >= 0); EnumWrapper::pyobj_insts[59] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::LT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LT", inst) >= 0); EnumWrapper::pyobj_insts[60] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::LEQ; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LEQ", inst) >= 0); EnumWrapper::pyobj_insts[61] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::ISNAN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ISNAN", inst) >= 0); EnumWrapper::pyobj_insts[62] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ElemwiseMultiType::Mode::ISINF; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ISINF", inst) >= 0); EnumWrapper::pyobj_insts[63] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(ElemwiseMultiType) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"dtype", serialization::dump(opdef.dtype)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ElemwiseMultiType) int PyOp(ElemwiseMultiType)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "dtype", "scope", NULL}; PyObject *mode = NULL, *dtype = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &mode, &dtype, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ElemwiseMultiType)::py_getsetters[] = { {const_cast("mode"), py_get_generic(ElemwiseMultiType, mode), py_set_generic(ElemwiseMultiType, mode), const_cast("mode"), NULL}, {const_cast("dtype"), py_get_generic(ElemwiseMultiType, dtype), py_set_generic(ElemwiseMultiType, dtype), const_cast("dtype"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ElemwiseMultiType)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ElemwiseMultiType)::getstate, METH_NOARGS, "ElemwiseMultiType getstate"}, {const_cast("__setstate__"), PyOp(ElemwiseMultiType)::setstate, METH_VARARGS, "ElemwiseMultiType setstate"}, {NULL} /* Sentinel */ }; void _init_py_ElemwiseMultiType(py::module m) { using py_op = PyOp(ElemwiseMultiType); auto& py_type = PyOpType(ElemwiseMultiType); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ElemwiseMultiType"; py_type.tp_basicsize = sizeof(PyOp(ElemwiseMultiType)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ElemwiseMultiType"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_ElemwiseMultiType_Mode(py_type); PyType_Modified(&py_type); m.add_object("ElemwiseMultiType", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ElemwiseMultiType::typeinfo(), &py_type).second); } PyOpDefBegin(ExternOpr) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"output_shapes", serialization::dump(opdef.output_shapes)}, {"name", serialization::dump(opdef.name)}, {"data", serialization::dump(opdef.data)}, {"data_len", serialization::dump(opdef.data_len)}, {"output_dtypes", serialization::dump(opdef.output_dtypes)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("output_shapes"); if (iter != state.end()) { opdef.output_shapes = serialization::load(iter->second); } } { auto&& iter = state.find("name"); if (iter != state.end()) { opdef.name = serialization::load(iter->second); } } { auto&& iter = state.find("data"); if (iter != state.end()) { opdef.data = serialization::load(iter->second); } } { auto&& iter = state.find("data_len"); if (iter != state.end()) { opdef.data_len = serialization::load(iter->second); } } { auto&& iter = state.find("output_dtypes"); if (iter != state.end()) { opdef.output_dtypes = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ExternOpr) int PyOp(ExternOpr)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"output_shapes", "name", "data", "data_len", "output_dtypes", "scope", NULL}; PyObject *output_shapes = NULL, *name = NULL, *data = NULL, *data_len = NULL, *output_dtypes = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOO", const_cast(kwlist), &output_shapes, &name, &data, &data_len, &output_dtypes, &scope)) return -1; if (output_shapes) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().output_shapes = py::cast(py::handle(output_shapes)); } CATCH_ALL(-1) } if (name) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().name = py::cast(py::handle(name)); } CATCH_ALL(-1) } if (data) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().data = py::cast(py::handle(data)); } CATCH_ALL(-1) } if (data_len) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().data_len = py::cast(py::handle(data_len)); } CATCH_ALL(-1) } if (output_dtypes) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().output_dtypes = py::cast(py::handle(output_dtypes)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ExternOpr)::py_getsetters[] = { {const_cast("output_shapes"), py_get_generic(ExternOpr, output_shapes), py_set_generic(ExternOpr, output_shapes), const_cast("output_shapes"), NULL}, {const_cast("name"), py_get_generic(ExternOpr, name), py_set_generic(ExternOpr, name), const_cast("name"), NULL}, {const_cast("data"), py_get_generic(ExternOpr, data), py_set_generic(ExternOpr, data), const_cast("data"), NULL}, {const_cast("data_len"), py_get_generic(ExternOpr, data_len), py_set_generic(ExternOpr, data_len), const_cast("data_len"), NULL}, {const_cast("output_dtypes"), py_get_generic(ExternOpr, output_dtypes), py_set_generic(ExternOpr, output_dtypes), const_cast("output_dtypes"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ExternOpr)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ExternOpr)::getstate, METH_NOARGS, "ExternOpr getstate"}, {const_cast("__setstate__"), PyOp(ExternOpr)::setstate, METH_VARARGS, "ExternOpr setstate"}, {NULL} /* Sentinel */ }; void _init_py_ExternOpr(py::module m) { using py_op = PyOp(ExternOpr); auto& py_type = PyOpType(ExternOpr); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ExternOpr"; py_type.tp_basicsize = sizeof(PyOp(ExternOpr)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ExternOpr"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("ExternOpr", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ExternOpr::typeinfo(), &py_type).second); } PyOpDefBegin(Eye) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"k", serialization::dump(opdef.k)}, {"dtype", serialization::dump(opdef.dtype)}, {"comp_node", serialization::dump(opdef.comp_node)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("k"); if (iter != state.end()) { opdef.k = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Eye) int PyOp(Eye)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"k", "dtype", "comp_node", "scope", NULL}; PyObject *k = NULL, *dtype = NULL, *comp_node = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOO", const_cast(kwlist), &k, &dtype, &comp_node, &scope)) return -1; if (k) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().k = py::cast(py::handle(k)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Eye)::py_getsetters[] = { {const_cast("k"), py_get_generic(Eye, k), py_set_generic(Eye, k), const_cast("k"), NULL}, {const_cast("dtype"), py_get_generic(Eye, dtype), py_set_generic(Eye, dtype), const_cast("dtype"), NULL}, {const_cast("comp_node"), py_get_generic(Eye, comp_node), py_set_generic(Eye, comp_node), const_cast("comp_node"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Eye)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Eye)::getstate, METH_NOARGS, "Eye getstate"}, {const_cast("__setstate__"), PyOp(Eye)::setstate, METH_VARARGS, "Eye setstate"}, {NULL} /* Sentinel */ }; void _init_py_Eye(py::module m) { using py_op = PyOp(Eye); auto& py_type = PyOpType(Eye); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Eye"; py_type.tp_basicsize = sizeof(PyOp(Eye)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Eye"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Eye", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Eye::typeinfo(), &py_type).second); } PyOpDefBegin(FakeQuant) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"qmin", serialization::dump(opdef.qmin)}, {"qmax", serialization::dump(opdef.qmax)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("qmin"); if (iter != state.end()) { opdef.qmin = serialization::load(iter->second); } } { auto&& iter = state.find("qmax"); if (iter != state.end()) { opdef.qmax = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(FakeQuant) int PyOp(FakeQuant)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"qmin", "qmax", "scope", NULL}; PyObject *qmin = NULL, *qmax = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &qmin, &qmax, &scope)) return -1; if (qmin) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().qmin = py::cast(py::handle(qmin)); } CATCH_ALL(-1) } if (qmax) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().qmax = py::cast(py::handle(qmax)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(FakeQuant)::py_getsetters[] = { {const_cast("qmin"), py_get_generic(FakeQuant, qmin), py_set_generic(FakeQuant, qmin), const_cast("qmin"), NULL}, {const_cast("qmax"), py_get_generic(FakeQuant, qmax), py_set_generic(FakeQuant, qmax), const_cast("qmax"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(FakeQuant)::tp_methods[] = { {const_cast("__getstate__"), PyOp(FakeQuant)::getstate, METH_NOARGS, "FakeQuant getstate"}, {const_cast("__setstate__"), PyOp(FakeQuant)::setstate, METH_VARARGS, "FakeQuant setstate"}, {NULL} /* Sentinel */ }; void _init_py_FakeQuant(py::module m) { using py_op = PyOp(FakeQuant); auto& py_type = PyOpType(FakeQuant); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.FakeQuant"; py_type.tp_basicsize = sizeof(PyOp(FakeQuant)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "FakeQuant"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("FakeQuant", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(FakeQuant::typeinfo(), &py_type).second); } PyOpDefBegin(FastpathCopy) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(FastpathCopy) int PyOp(FastpathCopy)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(FastpathCopy)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(FastpathCopy)::tp_methods[] = { {const_cast("__getstate__"), PyOp(FastpathCopy)::getstate, METH_NOARGS, "FastpathCopy getstate"}, {const_cast("__setstate__"), PyOp(FastpathCopy)::setstate, METH_VARARGS, "FastpathCopy setstate"}, {NULL} /* Sentinel */ }; void _init_py_FastpathCopy(py::module m) { using py_op = PyOp(FastpathCopy); auto& py_type = PyOpType(FastpathCopy); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.FastpathCopy"; py_type.tp_basicsize = sizeof(PyOp(FastpathCopy)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "FastpathCopy"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("FastpathCopy", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(FastpathCopy::typeinfo(), &py_type).second); } PyOpDefBegin(GammaRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(GammaRNG) int PyOp(GammaRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "handle", "scope", NULL}; PyObject *seed = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &seed, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(GammaRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(GammaRNG, seed), py_set_generic(GammaRNG, seed), const_cast("seed"), NULL}, {const_cast("handle"), py_get_generic(GammaRNG, handle), py_set_generic(GammaRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(GammaRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(GammaRNG)::getstate, METH_NOARGS, "GammaRNG getstate"}, {const_cast("__setstate__"), PyOp(GammaRNG)::setstate, METH_VARARGS, "GammaRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_GammaRNG(py::module m) { using py_op = PyOp(GammaRNG); auto& py_type = PyOpType(GammaRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.GammaRNG"; py_type.tp_basicsize = sizeof(PyOp(GammaRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "GammaRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("GammaRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(GammaRNG::typeinfo(), &py_type).second); } PyOpDefBegin(GaussianRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"mean", serialization::dump(opdef.mean)}, {"std", serialization::dump(opdef.std)}, {"dtype", serialization::dump(opdef.dtype)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("mean"); if (iter != state.end()) { opdef.mean = serialization::load(iter->second); } } { auto&& iter = state.find("std"); if (iter != state.end()) { opdef.std = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(GaussianRNG) int PyOp(GaussianRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "mean", "std", "dtype", "handle", "scope", NULL}; PyObject *seed = NULL, *mean = NULL, *std = NULL, *dtype = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOO", const_cast(kwlist), &seed, &mean, &std, &dtype, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (mean) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mean = py::cast(py::handle(mean)); } CATCH_ALL(-1) } if (std) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().std = py::cast(py::handle(std)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(GaussianRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(GaussianRNG, seed), py_set_generic(GaussianRNG, seed), const_cast("seed"), NULL}, {const_cast("mean"), py_get_generic(GaussianRNG, mean), py_set_generic(GaussianRNG, mean), const_cast("mean"), NULL}, {const_cast("std"), py_get_generic(GaussianRNG, std), py_set_generic(GaussianRNG, std), const_cast("std"), NULL}, {const_cast("dtype"), py_get_generic(GaussianRNG, dtype), py_set_generic(GaussianRNG, dtype), const_cast("dtype"), NULL}, {const_cast("handle"), py_get_generic(GaussianRNG, handle), py_set_generic(GaussianRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(GaussianRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(GaussianRNG)::getstate, METH_NOARGS, "GaussianRNG getstate"}, {const_cast("__setstate__"), PyOp(GaussianRNG)::setstate, METH_VARARGS, "GaussianRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_GaussianRNG(py::module m) { using py_op = PyOp(GaussianRNG); auto& py_type = PyOpType(GaussianRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.GaussianRNG"; py_type.tp_basicsize = sizeof(PyOp(GaussianRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "GaussianRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("GaussianRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(GaussianRNG::typeinfo(), &py_type).second); } PyOpDefBegin(GetVarShape) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(GetVarShape) int PyOp(GetVarShape)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "scope", NULL}; PyObject *axis = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &axis, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(GetVarShape)::py_getsetters[] = { {const_cast("axis"), py_get_generic(GetVarShape, axis), py_set_generic(GetVarShape, axis), const_cast("axis"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(GetVarShape)::tp_methods[] = { {const_cast("__getstate__"), PyOp(GetVarShape)::getstate, METH_NOARGS, "GetVarShape getstate"}, {const_cast("__setstate__"), PyOp(GetVarShape)::setstate, METH_VARARGS, "GetVarShape setstate"}, {NULL} /* Sentinel */ }; void _init_py_GetVarShape(py::module m) { using py_op = PyOp(GetVarShape); auto& py_type = PyOpType(GetVarShape); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.GetVarShape"; py_type.tp_basicsize = sizeof(PyOp(GetVarShape)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "GetVarShape"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("GetVarShape", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(GetVarShape::typeinfo(), &py_type).second); } void _init_py_GroupLocal_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_GroupLocal_Sparse(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Sparse", reinterpret_cast(e_type)) >= 0); } void _init_py_GroupLocal_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_GroupLocal_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(GroupLocal) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"sparse", serialization::dump(opdef.sparse)}, {"format", serialization::dump(opdef.format)}, {"compute_mode", serialization::dump(opdef.compute_mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("sparse"); if (iter != state.end()) { opdef.sparse = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(GroupLocal) int PyOp(GroupLocal)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "sparse", "format", "compute_mode", "scope", NULL}; PyObject *mode = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *sparse = NULL, *format = NULL, *compute_mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOO", const_cast(kwlist), &mode, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &sparse, &format, &compute_mode, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (sparse) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sparse = py::cast(py::handle(sparse)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(GroupLocal)::py_getsetters[] = { {const_cast("mode"), py_get_generic(GroupLocal, mode), py_set_generic(GroupLocal, mode), const_cast("mode"), NULL}, {const_cast("pad_h"), py_get_generic(GroupLocal, pad_h), py_set_generic(GroupLocal, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(GroupLocal, pad_w), py_set_generic(GroupLocal, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(GroupLocal, stride_h), py_set_generic(GroupLocal, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(GroupLocal, stride_w), py_set_generic(GroupLocal, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(GroupLocal, dilate_h), py_set_generic(GroupLocal, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(GroupLocal, dilate_w), py_set_generic(GroupLocal, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("sparse"), py_get_generic(GroupLocal, sparse), py_set_generic(GroupLocal, sparse), const_cast("sparse"), NULL}, {const_cast("format"), py_get_generic(GroupLocal, format), py_set_generic(GroupLocal, format), const_cast("format"), NULL}, {const_cast("compute_mode"), py_get_generic(GroupLocal, compute_mode), py_set_generic(GroupLocal, compute_mode), const_cast("compute_mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(GroupLocal)::tp_methods[] = { {const_cast("__getstate__"), PyOp(GroupLocal)::getstate, METH_NOARGS, "GroupLocal getstate"}, {const_cast("__setstate__"), PyOp(GroupLocal)::setstate, METH_VARARGS, "GroupLocal setstate"}, {NULL} /* Sentinel */ }; void _init_py_GroupLocal(py::module m) { using py_op = PyOp(GroupLocal); auto& py_type = PyOpType(GroupLocal); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.GroupLocal"; py_type.tp_basicsize = sizeof(PyOp(GroupLocal)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "GroupLocal"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_GroupLocal_Mode(py_type); _init_py_GroupLocal_Sparse(py_type); _init_py_GroupLocal_Format(py_type); _init_py_GroupLocal_ComputeMode(py_type); PyType_Modified(&py_type); m.add_object("GroupLocal", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(GroupLocal::typeinfo(), &py_type).second); } PyOpDefBegin(Identity) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Identity) int PyOp(Identity)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(Identity)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(Identity)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Identity)::getstate, METH_NOARGS, "Identity getstate"}, {const_cast("__setstate__"), PyOp(Identity)::setstate, METH_VARARGS, "Identity setstate"}, {NULL} /* Sentinel */ }; void _init_py_Identity(py::module m) { using py_op = PyOp(Identity); auto& py_type = PyOpType(Identity); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Identity"; py_type.tp_basicsize = sizeof(PyOp(Identity)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Identity"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Identity", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Identity::typeinfo(), &py_type).second); } PyOpDefBegin(Images2Neibs) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"window_h", serialization::dump(opdef.window_h)}, {"window_w", serialization::dump(opdef.window_w)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("window_h"); if (iter != state.end()) { opdef.window_h = serialization::load(iter->second); } } { auto&& iter = state.find("window_w"); if (iter != state.end()) { opdef.window_w = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Images2Neibs) int PyOp(Images2Neibs)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "window_h", "window_w", "scope", NULL}; PyObject *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *window_h = NULL, *window_w = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOO", const_cast(kwlist), &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &window_h, &window_w, &scope)) return -1; if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (window_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().window_h = py::cast(py::handle(window_h)); } CATCH_ALL(-1) } if (window_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().window_w = py::cast(py::handle(window_w)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Images2Neibs)::py_getsetters[] = { {const_cast("pad_h"), py_get_generic(Images2Neibs, pad_h), py_set_generic(Images2Neibs, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(Images2Neibs, pad_w), py_set_generic(Images2Neibs, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(Images2Neibs, stride_h), py_set_generic(Images2Neibs, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(Images2Neibs, stride_w), py_set_generic(Images2Neibs, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(Images2Neibs, dilate_h), py_set_generic(Images2Neibs, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(Images2Neibs, dilate_w), py_set_generic(Images2Neibs, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("window_h"), py_get_generic(Images2Neibs, window_h), py_set_generic(Images2Neibs, window_h), const_cast("window_h"), NULL}, {const_cast("window_w"), py_get_generic(Images2Neibs, window_w), py_set_generic(Images2Neibs, window_w), const_cast("window_w"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Images2Neibs)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Images2Neibs)::getstate, METH_NOARGS, "Images2Neibs getstate"}, {const_cast("__setstate__"), PyOp(Images2Neibs)::setstate, METH_VARARGS, "Images2Neibs setstate"}, {NULL} /* Sentinel */ }; void _init_py_Images2Neibs(py::module m) { using py_op = PyOp(Images2Neibs); auto& py_type = PyOpType(Images2Neibs); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Images2Neibs"; py_type.tp_basicsize = sizeof(PyOp(Images2Neibs)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Images2Neibs"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Images2Neibs", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Images2Neibs::typeinfo(), &py_type).second); } PyOpDefBegin(IncrMeshIndexing) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IncrMeshIndexing) int PyOp(IncrMeshIndexing)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IncrMeshIndexing)::py_getsetters[] = { {const_cast("items"), py_get_generic(IncrMeshIndexing, items), py_set_generic(IncrMeshIndexing, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IncrMeshIndexing)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IncrMeshIndexing)::getstate, METH_NOARGS, "IncrMeshIndexing getstate"}, {const_cast("__setstate__"), PyOp(IncrMeshIndexing)::setstate, METH_VARARGS, "IncrMeshIndexing setstate"}, {NULL} /* Sentinel */ }; void _init_py_IncrMeshIndexing(py::module m) { using py_op = PyOp(IncrMeshIndexing); auto& py_type = PyOpType(IncrMeshIndexing); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IncrMeshIndexing"; py_type.tp_basicsize = sizeof(PyOp(IncrMeshIndexing)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IncrMeshIndexing"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IncrMeshIndexing", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IncrMeshIndexing::typeinfo(), &py_type).second); } PyOpDefBegin(IncrSubtensor) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IncrSubtensor) int PyOp(IncrSubtensor)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IncrSubtensor)::py_getsetters[] = { {const_cast("items"), py_get_generic(IncrSubtensor, items), py_set_generic(IncrSubtensor, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IncrSubtensor)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IncrSubtensor)::getstate, METH_NOARGS, "IncrSubtensor getstate"}, {const_cast("__setstate__"), PyOp(IncrSubtensor)::setstate, METH_VARARGS, "IncrSubtensor setstate"}, {NULL} /* Sentinel */ }; void _init_py_IncrSubtensor(py::module m) { using py_op = PyOp(IncrSubtensor); auto& py_type = PyOpType(IncrSubtensor); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IncrSubtensor"; py_type.tp_basicsize = sizeof(PyOp(IncrSubtensor)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IncrSubtensor"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IncrSubtensor", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IncrSubtensor::typeinfo(), &py_type).second); } PyOpDefBegin(IndexingIncrMultiAxisVec) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IndexingIncrMultiAxisVec) int PyOp(IndexingIncrMultiAxisVec)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IndexingIncrMultiAxisVec)::py_getsetters[] = { {const_cast("items"), py_get_generic(IndexingIncrMultiAxisVec, items), py_set_generic(IndexingIncrMultiAxisVec, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IndexingIncrMultiAxisVec)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IndexingIncrMultiAxisVec)::getstate, METH_NOARGS, "IndexingIncrMultiAxisVec getstate"}, {const_cast("__setstate__"), PyOp(IndexingIncrMultiAxisVec)::setstate, METH_VARARGS, "IndexingIncrMultiAxisVec setstate"}, {NULL} /* Sentinel */ }; void _init_py_IndexingIncrMultiAxisVec(py::module m) { using py_op = PyOp(IndexingIncrMultiAxisVec); auto& py_type = PyOpType(IndexingIncrMultiAxisVec); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IndexingIncrMultiAxisVec"; py_type.tp_basicsize = sizeof(PyOp(IndexingIncrMultiAxisVec)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IndexingIncrMultiAxisVec"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IndexingIncrMultiAxisVec", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IndexingIncrMultiAxisVec::typeinfo(), &py_type).second); } PyOpDefBegin(IndexingMultiAxisVec) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IndexingMultiAxisVec) int PyOp(IndexingMultiAxisVec)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IndexingMultiAxisVec)::py_getsetters[] = { {const_cast("items"), py_get_generic(IndexingMultiAxisVec, items), py_set_generic(IndexingMultiAxisVec, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IndexingMultiAxisVec)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IndexingMultiAxisVec)::getstate, METH_NOARGS, "IndexingMultiAxisVec getstate"}, {const_cast("__setstate__"), PyOp(IndexingMultiAxisVec)::setstate, METH_VARARGS, "IndexingMultiAxisVec setstate"}, {NULL} /* Sentinel */ }; void _init_py_IndexingMultiAxisVec(py::module m) { using py_op = PyOp(IndexingMultiAxisVec); auto& py_type = PyOpType(IndexingMultiAxisVec); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IndexingMultiAxisVec"; py_type.tp_basicsize = sizeof(PyOp(IndexingMultiAxisVec)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IndexingMultiAxisVec"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IndexingMultiAxisVec", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IndexingMultiAxisVec::typeinfo(), &py_type).second); } PyOpDefBegin(IndexingOneHot) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)}, {"ndim", serialization::dump(opdef.ndim)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("ndim"); if (iter != state.end()) { opdef.ndim = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IndexingOneHot) int PyOp(IndexingOneHot)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "ndim", "scope", NULL}; PyObject *axis = NULL, *ndim = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &axis, &ndim, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (ndim) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().ndim = py::cast(py::handle(ndim)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IndexingOneHot)::py_getsetters[] = { {const_cast("axis"), py_get_generic(IndexingOneHot, axis), py_set_generic(IndexingOneHot, axis), const_cast("axis"), NULL}, {const_cast("ndim"), py_get_generic(IndexingOneHot, ndim), py_set_generic(IndexingOneHot, ndim), const_cast("ndim"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IndexingOneHot)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IndexingOneHot)::getstate, METH_NOARGS, "IndexingOneHot getstate"}, {const_cast("__setstate__"), PyOp(IndexingOneHot)::setstate, METH_VARARGS, "IndexingOneHot setstate"}, {NULL} /* Sentinel */ }; void _init_py_IndexingOneHot(py::module m) { using py_op = PyOp(IndexingOneHot); auto& py_type = PyOpType(IndexingOneHot); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IndexingOneHot"; py_type.tp_basicsize = sizeof(PyOp(IndexingOneHot)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IndexingOneHot"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IndexingOneHot", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IndexingOneHot::typeinfo(), &py_type).second); } PyOpDefBegin(IndexingSetMultiAxisVec) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IndexingSetMultiAxisVec) int PyOp(IndexingSetMultiAxisVec)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IndexingSetMultiAxisVec)::py_getsetters[] = { {const_cast("items"), py_get_generic(IndexingSetMultiAxisVec, items), py_set_generic(IndexingSetMultiAxisVec, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IndexingSetMultiAxisVec)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IndexingSetMultiAxisVec)::getstate, METH_NOARGS, "IndexingSetMultiAxisVec getstate"}, {const_cast("__setstate__"), PyOp(IndexingSetMultiAxisVec)::setstate, METH_VARARGS, "IndexingSetMultiAxisVec setstate"}, {NULL} /* Sentinel */ }; void _init_py_IndexingSetMultiAxisVec(py::module m) { using py_op = PyOp(IndexingSetMultiAxisVec); auto& py_type = PyOpType(IndexingSetMultiAxisVec); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IndexingSetMultiAxisVec"; py_type.tp_basicsize = sizeof(PyOp(IndexingSetMultiAxisVec)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IndexingSetMultiAxisVec"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IndexingSetMultiAxisVec", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IndexingSetMultiAxisVec::typeinfo(), &py_type).second); } PyOpDefBegin(IndexingSetOneHot) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)}, {"ndim", serialization::dump(opdef.ndim)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("ndim"); if (iter != state.end()) { opdef.ndim = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(IndexingSetOneHot) int PyOp(IndexingSetOneHot)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "ndim", "scope", NULL}; PyObject *axis = NULL, *ndim = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &axis, &ndim, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (ndim) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().ndim = py::cast(py::handle(ndim)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(IndexingSetOneHot)::py_getsetters[] = { {const_cast("axis"), py_get_generic(IndexingSetOneHot, axis), py_set_generic(IndexingSetOneHot, axis), const_cast("axis"), NULL}, {const_cast("ndim"), py_get_generic(IndexingSetOneHot, ndim), py_set_generic(IndexingSetOneHot, ndim), const_cast("ndim"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(IndexingSetOneHot)::tp_methods[] = { {const_cast("__getstate__"), PyOp(IndexingSetOneHot)::getstate, METH_NOARGS, "IndexingSetOneHot getstate"}, {const_cast("__setstate__"), PyOp(IndexingSetOneHot)::setstate, METH_VARARGS, "IndexingSetOneHot setstate"}, {NULL} /* Sentinel */ }; void _init_py_IndexingSetOneHot(py::module m) { using py_op = PyOp(IndexingSetOneHot); auto& py_type = PyOpType(IndexingSetOneHot); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.IndexingSetOneHot"; py_type.tp_basicsize = sizeof(PyOp(IndexingSetOneHot)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "IndexingSetOneHot"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("IndexingSetOneHot", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(IndexingSetOneHot::typeinfo(), &py_type).second); } PyOpDefBegin(InplaceAdd) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(InplaceAdd) int PyOp(InplaceAdd)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(InplaceAdd)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(InplaceAdd)::tp_methods[] = { {const_cast("__getstate__"), PyOp(InplaceAdd)::getstate, METH_NOARGS, "InplaceAdd getstate"}, {const_cast("__setstate__"), PyOp(InplaceAdd)::setstate, METH_VARARGS, "InplaceAdd setstate"}, {NULL} /* Sentinel */ }; void _init_py_InplaceAdd(py::module m) { using py_op = PyOp(InplaceAdd); auto& py_type = PyOpType(InplaceAdd); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.InplaceAdd"; py_type.tp_basicsize = sizeof(PyOp(InplaceAdd)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "InplaceAdd"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("InplaceAdd", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(InplaceAdd::typeinfo(), &py_type).second); } PyOpDefBegin(LAMBUpdate) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"beta_1", serialization::dump(opdef.beta_1)}, {"beta_2", serialization::dump(opdef.beta_2)}, {"step", serialization::dump(opdef.step)}, {"lr", serialization::dump(opdef.lr)}, {"weight_decay", serialization::dump(opdef.weight_decay)}, {"eps", serialization::dump(opdef.eps)}, {"bias_correction", serialization::dump(opdef.bias_correction)}, {"always_adapt", serialization::dump(opdef.always_adapt)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("beta_1"); if (iter != state.end()) { opdef.beta_1 = serialization::load(iter->second); } } { auto&& iter = state.find("beta_2"); if (iter != state.end()) { opdef.beta_2 = serialization::load(iter->second); } } { auto&& iter = state.find("step"); if (iter != state.end()) { opdef.step = serialization::load(iter->second); } } { auto&& iter = state.find("lr"); if (iter != state.end()) { opdef.lr = serialization::load(iter->second); } } { auto&& iter = state.find("weight_decay"); if (iter != state.end()) { opdef.weight_decay = serialization::load(iter->second); } } { auto&& iter = state.find("eps"); if (iter != state.end()) { opdef.eps = serialization::load(iter->second); } } { auto&& iter = state.find("bias_correction"); if (iter != state.end()) { opdef.bias_correction = serialization::load(iter->second); } } { auto&& iter = state.find("always_adapt"); if (iter != state.end()) { opdef.always_adapt = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(LAMBUpdate) int PyOp(LAMBUpdate)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"beta_1", "beta_2", "step", "lr", "weight_decay", "eps", "bias_correction", "always_adapt", "scope", NULL}; PyObject *beta_1 = NULL, *beta_2 = NULL, *step = NULL, *lr = NULL, *weight_decay = NULL, *eps = NULL, *bias_correction = NULL, *always_adapt = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOO", const_cast(kwlist), &beta_1, &beta_2, &step, &lr, &weight_decay, &eps, &bias_correction, &always_adapt, &scope)) return -1; if (beta_1) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().beta_1 = py::cast(py::handle(beta_1)); } CATCH_ALL(-1) } if (beta_2) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().beta_2 = py::cast(py::handle(beta_2)); } CATCH_ALL(-1) } if (step) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().step = py::cast(py::handle(step)); } CATCH_ALL(-1) } if (lr) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().lr = py::cast(py::handle(lr)); } CATCH_ALL(-1) } if (weight_decay) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().weight_decay = py::cast(py::handle(weight_decay)); } CATCH_ALL(-1) } if (eps) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().eps = py::cast(py::handle(eps)); } CATCH_ALL(-1) } if (bias_correction) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bias_correction = py::cast(py::handle(bias_correction)); } CATCH_ALL(-1) } if (always_adapt) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().always_adapt = py::cast(py::handle(always_adapt)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(LAMBUpdate)::py_getsetters[] = { {const_cast("beta_1"), py_get_generic(LAMBUpdate, beta_1), py_set_generic(LAMBUpdate, beta_1), const_cast("beta_1"), NULL}, {const_cast("beta_2"), py_get_generic(LAMBUpdate, beta_2), py_set_generic(LAMBUpdate, beta_2), const_cast("beta_2"), NULL}, {const_cast("step"), py_get_generic(LAMBUpdate, step), py_set_generic(LAMBUpdate, step), const_cast("step"), NULL}, {const_cast("lr"), py_get_generic(LAMBUpdate, lr), py_set_generic(LAMBUpdate, lr), const_cast("lr"), NULL}, {const_cast("weight_decay"), py_get_generic(LAMBUpdate, weight_decay), py_set_generic(LAMBUpdate, weight_decay), const_cast("weight_decay"), NULL}, {const_cast("eps"), py_get_generic(LAMBUpdate, eps), py_set_generic(LAMBUpdate, eps), const_cast("eps"), NULL}, {const_cast("bias_correction"), py_get_generic(LAMBUpdate, bias_correction), py_set_generic(LAMBUpdate, bias_correction), const_cast("bias_correction"), NULL}, {const_cast("always_adapt"), py_get_generic(LAMBUpdate, always_adapt), py_set_generic(LAMBUpdate, always_adapt), const_cast("always_adapt"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(LAMBUpdate)::tp_methods[] = { {const_cast("__getstate__"), PyOp(LAMBUpdate)::getstate, METH_NOARGS, "LAMBUpdate getstate"}, {const_cast("__setstate__"), PyOp(LAMBUpdate)::setstate, METH_VARARGS, "LAMBUpdate setstate"}, {NULL} /* Sentinel */ }; void _init_py_LAMBUpdate(py::module m) { using py_op = PyOp(LAMBUpdate); auto& py_type = PyOpType(LAMBUpdate); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.LAMBUpdate"; py_type.tp_basicsize = sizeof(PyOp(LAMBUpdate)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "LAMBUpdate"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("LAMBUpdate", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(LAMBUpdate::typeinfo(), &py_type).second); } PyOpDefBegin(LRN) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"n", serialization::dump(opdef.n)}, {"k", serialization::dump(opdef.k)}, {"alpha", serialization::dump(opdef.alpha)}, {"beta", serialization::dump(opdef.beta)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("n"); if (iter != state.end()) { opdef.n = serialization::load(iter->second); } } { auto&& iter = state.find("k"); if (iter != state.end()) { opdef.k = serialization::load(iter->second); } } { auto&& iter = state.find("alpha"); if (iter != state.end()) { opdef.alpha = serialization::load(iter->second); } } { auto&& iter = state.find("beta"); if (iter != state.end()) { opdef.beta = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(LRN) int PyOp(LRN)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"n", "k", "alpha", "beta", "scope", NULL}; PyObject *n = NULL, *k = NULL, *alpha = NULL, *beta = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &n, &k, &alpha, &beta, &scope)) return -1; if (n) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().n = py::cast(py::handle(n)); } CATCH_ALL(-1) } if (k) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().k = py::cast(py::handle(k)); } CATCH_ALL(-1) } if (alpha) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().alpha = py::cast(py::handle(alpha)); } CATCH_ALL(-1) } if (beta) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().beta = py::cast(py::handle(beta)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(LRN)::py_getsetters[] = { {const_cast("n"), py_get_generic(LRN, n), py_set_generic(LRN, n), const_cast("n"), NULL}, {const_cast("k"), py_get_generic(LRN, k), py_set_generic(LRN, k), const_cast("k"), NULL}, {const_cast("alpha"), py_get_generic(LRN, alpha), py_set_generic(LRN, alpha), const_cast("alpha"), NULL}, {const_cast("beta"), py_get_generic(LRN, beta), py_set_generic(LRN, beta), const_cast("beta"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(LRN)::tp_methods[] = { {const_cast("__getstate__"), PyOp(LRN)::getstate, METH_NOARGS, "LRN getstate"}, {const_cast("__setstate__"), PyOp(LRN)::setstate, METH_VARARGS, "LRN setstate"}, {NULL} /* Sentinel */ }; void _init_py_LRN(py::module m) { using py_op = PyOp(LRN); auto& py_type = PyOpType(LRN); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.LRN"; py_type.tp_basicsize = sizeof(PyOp(LRN)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "LRN"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("LRN", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(LRN::typeinfo(), &py_type).second); } PyOpDefBegin(LSQ) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"qmin", serialization::dump(opdef.qmin)}, {"qmax", serialization::dump(opdef.qmax)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("qmin"); if (iter != state.end()) { opdef.qmin = serialization::load(iter->second); } } { auto&& iter = state.find("qmax"); if (iter != state.end()) { opdef.qmax = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(LSQ) int PyOp(LSQ)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"qmin", "qmax", "scope", NULL}; PyObject *qmin = NULL, *qmax = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &qmin, &qmax, &scope)) return -1; if (qmin) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().qmin = py::cast(py::handle(qmin)); } CATCH_ALL(-1) } if (qmax) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().qmax = py::cast(py::handle(qmax)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(LSQ)::py_getsetters[] = { {const_cast("qmin"), py_get_generic(LSQ, qmin), py_set_generic(LSQ, qmin), const_cast("qmin"), NULL}, {const_cast("qmax"), py_get_generic(LSQ, qmax), py_set_generic(LSQ, qmax), const_cast("qmax"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(LSQ)::tp_methods[] = { {const_cast("__getstate__"), PyOp(LSQ)::getstate, METH_NOARGS, "LSQ getstate"}, {const_cast("__setstate__"), PyOp(LSQ)::setstate, METH_VARARGS, "LSQ setstate"}, {NULL} /* Sentinel */ }; void _init_py_LSQ(py::module m) { using py_op = PyOp(LSQ); auto& py_type = PyOpType(LSQ); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.LSQ"; py_type.tp_basicsize = sizeof(PyOp(LSQ)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "LSQ"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("LSQ", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(LSQ::typeinfo(), &py_type).second); } void _init_py_LSTM_FwdMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "FwdMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(LSTM) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"num_layers", serialization::dump(opdef.num_layers)}, {"bidirectional", serialization::dump(opdef.bidirectional)}, {"bias", serialization::dump(opdef.bias)}, {"hidden_size", serialization::dump(opdef.hidden_size)}, {"proj_size", serialization::dump(opdef.proj_size)}, {"dropout", serialization::dump(opdef.dropout)}, {"fwd_mode", serialization::dump(opdef.fwd_mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("num_layers"); if (iter != state.end()) { opdef.num_layers = serialization::load(iter->second); } } { auto&& iter = state.find("bidirectional"); if (iter != state.end()) { opdef.bidirectional = serialization::load(iter->second); } } { auto&& iter = state.find("bias"); if (iter != state.end()) { opdef.bias = serialization::load(iter->second); } } { auto&& iter = state.find("hidden_size"); if (iter != state.end()) { opdef.hidden_size = serialization::load(iter->second); } } { auto&& iter = state.find("proj_size"); if (iter != state.end()) { opdef.proj_size = serialization::load(iter->second); } } { auto&& iter = state.find("dropout"); if (iter != state.end()) { opdef.dropout = serialization::load(iter->second); } } { auto&& iter = state.find("fwd_mode"); if (iter != state.end()) { opdef.fwd_mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(LSTM) int PyOp(LSTM)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"num_layers", "bidirectional", "bias", "hidden_size", "proj_size", "dropout", "fwd_mode", "scope", NULL}; PyObject *num_layers = NULL, *bidirectional = NULL, *bias = NULL, *hidden_size = NULL, *proj_size = NULL, *dropout = NULL, *fwd_mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOO", const_cast(kwlist), &num_layers, &bidirectional, &bias, &hidden_size, &proj_size, &dropout, &fwd_mode, &scope)) return -1; if (num_layers) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().num_layers = py::cast(py::handle(num_layers)); } CATCH_ALL(-1) } if (bidirectional) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bidirectional = py::cast(py::handle(bidirectional)); } CATCH_ALL(-1) } if (bias) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bias = py::cast(py::handle(bias)); } CATCH_ALL(-1) } if (hidden_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().hidden_size = py::cast(py::handle(hidden_size)); } CATCH_ALL(-1) } if (proj_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().proj_size = py::cast(py::handle(proj_size)); } CATCH_ALL(-1) } if (dropout) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dropout = py::cast(py::handle(dropout)); } CATCH_ALL(-1) } if (fwd_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().fwd_mode = py::cast(py::handle(fwd_mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(LSTM)::py_getsetters[] = { {const_cast("num_layers"), py_get_generic(LSTM, num_layers), py_set_generic(LSTM, num_layers), const_cast("num_layers"), NULL}, {const_cast("bidirectional"), py_get_generic(LSTM, bidirectional), py_set_generic(LSTM, bidirectional), const_cast("bidirectional"), NULL}, {const_cast("bias"), py_get_generic(LSTM, bias), py_set_generic(LSTM, bias), const_cast("bias"), NULL}, {const_cast("hidden_size"), py_get_generic(LSTM, hidden_size), py_set_generic(LSTM, hidden_size), const_cast("hidden_size"), NULL}, {const_cast("proj_size"), py_get_generic(LSTM, proj_size), py_set_generic(LSTM, proj_size), const_cast("proj_size"), NULL}, {const_cast("dropout"), py_get_generic(LSTM, dropout), py_set_generic(LSTM, dropout), const_cast("dropout"), NULL}, {const_cast("fwd_mode"), py_get_generic(LSTM, fwd_mode), py_set_generic(LSTM, fwd_mode), const_cast("fwd_mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(LSTM)::tp_methods[] = { {const_cast("__getstate__"), PyOp(LSTM)::getstate, METH_NOARGS, "LSTM getstate"}, {const_cast("__setstate__"), PyOp(LSTM)::setstate, METH_VARARGS, "LSTM setstate"}, {NULL} /* Sentinel */ }; void _init_py_LSTM(py::module m) { using py_op = PyOp(LSTM); auto& py_type = PyOpType(LSTM); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.LSTM"; py_type.tp_basicsize = sizeof(PyOp(LSTM)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "LSTM"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_LSTM_FwdMode(py_type); PyType_Modified(&py_type); m.add_object("LSTM", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(LSTM::typeinfo(), &py_type).second); } PyOpDefBegin(LSTMCell) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(LSTMCell) int PyOp(LSTMCell)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(LSTMCell)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(LSTMCell)::tp_methods[] = { {const_cast("__getstate__"), PyOp(LSTMCell)::getstate, METH_NOARGS, "LSTMCell getstate"}, {const_cast("__setstate__"), PyOp(LSTMCell)::setstate, METH_VARARGS, "LSTMCell setstate"}, {NULL} /* Sentinel */ }; void _init_py_LSTMCell(py::module m) { using py_op = PyOp(LSTMCell); auto& py_type = PyOpType(LSTMCell); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.LSTMCell"; py_type.tp_basicsize = sizeof(PyOp(LSTMCell)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "LSTMCell"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("LSTMCell", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(LSTMCell::typeinfo(), &py_type).second); } PyOpDefBegin(LayerNorm) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"affine", serialization::dump(opdef.affine)}, {"eps", serialization::dump(opdef.eps)}, {"normalized_dim", serialization::dump(opdef.normalized_dim)}, {"normalized_size", serialization::dump(opdef.normalized_size)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("affine"); if (iter != state.end()) { opdef.affine = serialization::load(iter->second); } } { auto&& iter = state.find("eps"); if (iter != state.end()) { opdef.eps = serialization::load(iter->second); } } { auto&& iter = state.find("normalized_dim"); if (iter != state.end()) { opdef.normalized_dim = serialization::load(iter->second); } } { auto&& iter = state.find("normalized_size"); if (iter != state.end()) { opdef.normalized_size = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(LayerNorm) int PyOp(LayerNorm)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"affine", "eps", "normalized_dim", "normalized_size", "scope", NULL}; PyObject *affine = NULL, *eps = NULL, *normalized_dim = NULL, *normalized_size = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &affine, &eps, &normalized_dim, &normalized_size, &scope)) return -1; if (affine) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().affine = py::cast(py::handle(affine)); } CATCH_ALL(-1) } if (eps) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().eps = py::cast(py::handle(eps)); } CATCH_ALL(-1) } if (normalized_dim) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().normalized_dim = py::cast(py::handle(normalized_dim)); } CATCH_ALL(-1) } if (normalized_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().normalized_size = py::cast(py::handle(normalized_size)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(LayerNorm)::py_getsetters[] = { {const_cast("affine"), py_get_generic(LayerNorm, affine), py_set_generic(LayerNorm, affine), const_cast("affine"), NULL}, {const_cast("eps"), py_get_generic(LayerNorm, eps), py_set_generic(LayerNorm, eps), const_cast("eps"), NULL}, {const_cast("normalized_dim"), py_get_generic(LayerNorm, normalized_dim), py_set_generic(LayerNorm, normalized_dim), const_cast("normalized_dim"), NULL}, {const_cast("normalized_size"), py_get_generic(LayerNorm, normalized_size), py_set_generic(LayerNorm, normalized_size), const_cast("normalized_size"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(LayerNorm)::tp_methods[] = { {const_cast("__getstate__"), PyOp(LayerNorm)::getstate, METH_NOARGS, "LayerNorm getstate"}, {const_cast("__setstate__"), PyOp(LayerNorm)::setstate, METH_VARARGS, "LayerNorm setstate"}, {NULL} /* Sentinel */ }; void _init_py_LayerNorm(py::module m) { using py_op = PyOp(LayerNorm); auto& py_type = PyOpType(LayerNorm); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.LayerNorm"; py_type.tp_basicsize = sizeof(PyOp(LayerNorm)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "LayerNorm"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("LayerNorm", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(LayerNorm::typeinfo(), &py_type).second); } PyOpDefBegin(Linspace) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"endpoint", serialization::dump(opdef.endpoint)}, {"comp_node", serialization::dump(opdef.comp_node)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("endpoint"); if (iter != state.end()) { opdef.endpoint = serialization::load(iter->second); } } { auto&& iter = state.find("comp_node"); if (iter != state.end()) { opdef.comp_node = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Linspace) int PyOp(Linspace)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"endpoint", "comp_node", "scope", NULL}; PyObject *endpoint = NULL, *comp_node = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &endpoint, &comp_node, &scope)) return -1; if (endpoint) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().endpoint = py::cast(py::handle(endpoint)); } CATCH_ALL(-1) } if (comp_node) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().comp_node = py::cast(py::handle(comp_node)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Linspace)::py_getsetters[] = { {const_cast("endpoint"), py_get_generic(Linspace, endpoint), py_set_generic(Linspace, endpoint), const_cast("endpoint"), NULL}, {const_cast("comp_node"), py_get_generic(Linspace, comp_node), py_set_generic(Linspace, comp_node), const_cast("comp_node"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Linspace)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Linspace)::getstate, METH_NOARGS, "Linspace getstate"}, {const_cast("__setstate__"), PyOp(Linspace)::setstate, METH_VARARGS, "Linspace setstate"}, {NULL} /* Sentinel */ }; void _init_py_Linspace(py::module m) { using py_op = PyOp(Linspace); auto& py_type = PyOpType(Linspace); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Linspace"; py_type.tp_basicsize = sizeof(PyOp(Linspace)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Linspace"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Linspace", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Linspace::typeinfo(), &py_type).second); } PyOpDefBegin(MagicMindRuntime) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"buf", serialization::dump(opdef.buf)}, {"buf_size", serialization::dump(opdef.buf_size)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("buf"); if (iter != state.end()) { opdef.buf = serialization::load(iter->second); } } { auto&& iter = state.find("buf_size"); if (iter != state.end()) { opdef.buf_size = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(MagicMindRuntime) int PyOp(MagicMindRuntime)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"buf", "buf_size", "scope", NULL}; PyObject *buf = NULL, *buf_size = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &buf, &buf_size, &scope)) return -1; if (buf) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf = py::cast(py::handle(buf)); } CATCH_ALL(-1) } if (buf_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf_size = py::cast(py::handle(buf_size)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(MagicMindRuntime)::py_getsetters[] = { {const_cast("buf"), py_get_generic(MagicMindRuntime, buf), py_set_generic(MagicMindRuntime, buf), const_cast("buf"), NULL}, {const_cast("buf_size"), py_get_generic(MagicMindRuntime, buf_size), py_set_generic(MagicMindRuntime, buf_size), const_cast("buf_size"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(MagicMindRuntime)::tp_methods[] = { {const_cast("__getstate__"), PyOp(MagicMindRuntime)::getstate, METH_NOARGS, "MagicMindRuntime getstate"}, {const_cast("__setstate__"), PyOp(MagicMindRuntime)::setstate, METH_VARARGS, "MagicMindRuntime setstate"}, {NULL} /* Sentinel */ }; void _init_py_MagicMindRuntime(py::module m) { using py_op = PyOp(MagicMindRuntime); auto& py_type = PyOpType(MagicMindRuntime); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.MagicMindRuntime"; py_type.tp_basicsize = sizeof(PyOp(MagicMindRuntime)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "MagicMindRuntime"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("MagicMindRuntime", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(MagicMindRuntime::typeinfo(), &py_type).second); } PyOpDefBegin(MatrixInverse) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(MatrixInverse) int PyOp(MatrixInverse)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { return 0; } PyGetSetDef PyOp(MatrixInverse)::py_getsetters[] = { {NULL} /* Sentinel */ }; PyMethodDef PyOp(MatrixInverse)::tp_methods[] = { {const_cast("__getstate__"), PyOp(MatrixInverse)::getstate, METH_NOARGS, "MatrixInverse getstate"}, {const_cast("__setstate__"), PyOp(MatrixInverse)::setstate, METH_VARARGS, "MatrixInverse setstate"}, {NULL} /* Sentinel */ }; void _init_py_MatrixInverse(py::module m) { using py_op = PyOp(MatrixInverse); auto& py_type = PyOpType(MatrixInverse); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.MatrixInverse"; py_type.tp_basicsize = sizeof(PyOp(MatrixInverse)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "MatrixInverse"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("MatrixInverse", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(MatrixInverse::typeinfo(), &py_type).second); } void _init_py_MatrixMul_ComputeMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "ComputeMode", reinterpret_cast(e_type)) >= 0); } void _init_py_MatrixMul_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_MatrixMul_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(MatrixMul) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"transposeA", serialization::dump(opdef.transposeA)}, {"transposeB", serialization::dump(opdef.transposeB)}, {"compute_mode", serialization::dump(opdef.compute_mode)}, {"format", serialization::dump(opdef.format)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)}, {"dimA", serialization::dump(opdef.dimA)}, {"dimB", serialization::dump(opdef.dimB)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("transposeA"); if (iter != state.end()) { opdef.transposeA = serialization::load(iter->second); } } { auto&& iter = state.find("transposeB"); if (iter != state.end()) { opdef.transposeB = serialization::load(iter->second); } } { auto&& iter = state.find("compute_mode"); if (iter != state.end()) { opdef.compute_mode = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } { auto&& iter = state.find("dimA"); if (iter != state.end()) { opdef.dimA = serialization::load(iter->second); } } { auto&& iter = state.find("dimB"); if (iter != state.end()) { opdef.dimB = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(MatrixMul) int PyOp(MatrixMul)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"transposeA", "transposeB", "compute_mode", "format", "strategy", "workspace_limit", "dimA", "dimB", "scope", NULL}; PyObject *transposeA = NULL, *transposeB = NULL, *compute_mode = NULL, *format = NULL, *strategy = NULL, *workspace_limit = NULL, *dimA = NULL, *dimB = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOO", const_cast(kwlist), &transposeA, &transposeB, &compute_mode, &format, &strategy, &workspace_limit, &dimA, &dimB, &scope)) return -1; if (transposeA) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().transposeA = py::cast(py::handle(transposeA)); } CATCH_ALL(-1) } if (transposeB) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().transposeB = py::cast(py::handle(transposeB)); } CATCH_ALL(-1) } if (compute_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_mode = py::cast(py::handle(compute_mode)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (dimA) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dimA = py::cast(py::handle(dimA)); } CATCH_ALL(-1) } if (dimB) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dimB = py::cast(py::handle(dimB)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(MatrixMul)::py_getsetters[] = { {const_cast("transposeA"), py_get_generic(MatrixMul, transposeA), py_set_generic(MatrixMul, transposeA), const_cast("transposeA"), NULL}, {const_cast("transposeB"), py_get_generic(MatrixMul, transposeB), py_set_generic(MatrixMul, transposeB), const_cast("transposeB"), NULL}, {const_cast("compute_mode"), py_get_generic(MatrixMul, compute_mode), py_set_generic(MatrixMul, compute_mode), const_cast("compute_mode"), NULL}, {const_cast("format"), py_get_generic(MatrixMul, format), py_set_generic(MatrixMul, format), const_cast("format"), NULL}, {const_cast("strategy"), py_get_generic(MatrixMul, strategy), py_set_generic(MatrixMul, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(MatrixMul, workspace_limit), py_set_generic(MatrixMul, workspace_limit), const_cast("workspace_limit"), NULL}, {const_cast("dimA"), py_get_generic(MatrixMul, dimA), py_set_generic(MatrixMul, dimA), const_cast("dimA"), NULL}, {const_cast("dimB"), py_get_generic(MatrixMul, dimB), py_set_generic(MatrixMul, dimB), const_cast("dimB"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(MatrixMul)::tp_methods[] = { {const_cast("__getstate__"), PyOp(MatrixMul)::getstate, METH_NOARGS, "MatrixMul getstate"}, {const_cast("__setstate__"), PyOp(MatrixMul)::setstate, METH_VARARGS, "MatrixMul setstate"}, {NULL} /* Sentinel */ }; void _init_py_MatrixMul(py::module m) { using py_op = PyOp(MatrixMul); auto& py_type = PyOpType(MatrixMul); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.MatrixMul"; py_type.tp_basicsize = sizeof(PyOp(MatrixMul)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "MatrixMul"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_MatrixMul_ComputeMode(py_type); _init_py_MatrixMul_Format(py_type); _init_py_MatrixMul_Strategy(py_type); PyType_Modified(&py_type); m.add_object("MatrixMul", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(MatrixMul::typeinfo(), &py_type).second); } PyOpDefBegin(MeshIndexing) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(MeshIndexing) int PyOp(MeshIndexing)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(MeshIndexing)::py_getsetters[] = { {const_cast("items"), py_get_generic(MeshIndexing, items), py_set_generic(MeshIndexing, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(MeshIndexing)::tp_methods[] = { {const_cast("__getstate__"), PyOp(MeshIndexing)::getstate, METH_NOARGS, "MeshIndexing getstate"}, {const_cast("__setstate__"), PyOp(MeshIndexing)::setstate, METH_VARARGS, "MeshIndexing setstate"}, {NULL} /* Sentinel */ }; void _init_py_MeshIndexing(py::module m) { using py_op = PyOp(MeshIndexing); auto& py_type = PyOpType(MeshIndexing); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.MeshIndexing"; py_type.tp_basicsize = sizeof(PyOp(MeshIndexing)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "MeshIndexing"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("MeshIndexing", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(MeshIndexing::typeinfo(), &py_type).second); } PyOpDefBegin(NMSKeep) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"iou_thresh", serialization::dump(opdef.iou_thresh)}, {"max_output", serialization::dump(opdef.max_output)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("iou_thresh"); if (iter != state.end()) { opdef.iou_thresh = serialization::load(iter->second); } } { auto&& iter = state.find("max_output"); if (iter != state.end()) { opdef.max_output = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(NMSKeep) int PyOp(NMSKeep)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"iou_thresh", "max_output", "scope", NULL}; PyObject *iou_thresh = NULL, *max_output = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &iou_thresh, &max_output, &scope)) return -1; if (iou_thresh) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().iou_thresh = py::cast(py::handle(iou_thresh)); } CATCH_ALL(-1) } if (max_output) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().max_output = py::cast(py::handle(max_output)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(NMSKeep)::py_getsetters[] = { {const_cast("iou_thresh"), py_get_generic(NMSKeep, iou_thresh), py_set_generic(NMSKeep, iou_thresh), const_cast("iou_thresh"), NULL}, {const_cast("max_output"), py_get_generic(NMSKeep, max_output), py_set_generic(NMSKeep, max_output), const_cast("max_output"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(NMSKeep)::tp_methods[] = { {const_cast("__getstate__"), PyOp(NMSKeep)::getstate, METH_NOARGS, "NMSKeep getstate"}, {const_cast("__setstate__"), PyOp(NMSKeep)::setstate, METH_VARARGS, "NMSKeep setstate"}, {NULL} /* Sentinel */ }; void _init_py_NMSKeep(py::module m) { using py_op = PyOp(NMSKeep); auto& py_type = PyOpType(NMSKeep); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.NMSKeep"; py_type.tp_basicsize = sizeof(PyOp(NMSKeep)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "NMSKeep"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("NMSKeep", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(NMSKeep::typeinfo(), &py_type).second); } PyOpDefBegin(NvOf) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"precision", serialization::dump(opdef.precision)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("precision"); if (iter != state.end()) { opdef.precision = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(NvOf) int PyOp(NvOf)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"precision", "scope", NULL}; PyObject *precision = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &precision, &scope)) return -1; if (precision) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().precision = py::cast(py::handle(precision)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(NvOf)::py_getsetters[] = { {const_cast("precision"), py_get_generic(NvOf, precision), py_set_generic(NvOf, precision), const_cast("precision"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(NvOf)::tp_methods[] = { {const_cast("__getstate__"), PyOp(NvOf)::getstate, METH_NOARGS, "NvOf getstate"}, {const_cast("__setstate__"), PyOp(NvOf)::setstate, METH_VARARGS, "NvOf setstate"}, {NULL} /* Sentinel */ }; void _init_py_NvOf(py::module m) { using py_op = PyOp(NvOf); auto& py_type = PyOpType(NvOf); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.NvOf"; py_type.tp_basicsize = sizeof(PyOp(NvOf)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "NvOf"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("NvOf", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(NvOf::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Padding.PaddingMode"; static constexpr std::underlying_type_t max = 3 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"REPLICATE", "REFLECT", "CONSTANT"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("REPLICATE"), Padding::PaddingMode::REPLICATE}, {normalize_enum("REFLECT"), Padding::PaddingMode::REFLECT}, {normalize_enum("CONSTANT"), Padding::PaddingMode::CONSTANT}}; template<> PyObject* EnumWrapper::pyobj_insts[3] = {nullptr}; void _init_py_Padding_PaddingMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Padding.PaddingMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("PaddingMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Padding.PaddingMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Padding::PaddingMode::REPLICATE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REPLICATE", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Padding::PaddingMode::REFLECT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REFLECT", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Padding::PaddingMode::CONSTANT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CONSTANT", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "PaddingMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Padding) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"front_offset_dim0", serialization::dump(opdef.front_offset_dim0)}, {"front_offset_dim1", serialization::dump(opdef.front_offset_dim1)}, {"front_offset_dim2", serialization::dump(opdef.front_offset_dim2)}, {"front_offset_dim3", serialization::dump(opdef.front_offset_dim3)}, {"front_offset_dim4", serialization::dump(opdef.front_offset_dim4)}, {"front_offset_dim5", serialization::dump(opdef.front_offset_dim5)}, {"front_offset_dim6", serialization::dump(opdef.front_offset_dim6)}, {"back_offset_dim0", serialization::dump(opdef.back_offset_dim0)}, {"back_offset_dim1", serialization::dump(opdef.back_offset_dim1)}, {"back_offset_dim2", serialization::dump(opdef.back_offset_dim2)}, {"back_offset_dim3", serialization::dump(opdef.back_offset_dim3)}, {"back_offset_dim4", serialization::dump(opdef.back_offset_dim4)}, {"back_offset_dim5", serialization::dump(opdef.back_offset_dim5)}, {"back_offset_dim6", serialization::dump(opdef.back_offset_dim6)}, {"padding_val", serialization::dump(opdef.padding_val)}, {"padding_mode", serialization::dump(opdef.padding_mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("front_offset_dim0"); if (iter != state.end()) { opdef.front_offset_dim0 = serialization::load(iter->second); } } { auto&& iter = state.find("front_offset_dim1"); if (iter != state.end()) { opdef.front_offset_dim1 = serialization::load(iter->second); } } { auto&& iter = state.find("front_offset_dim2"); if (iter != state.end()) { opdef.front_offset_dim2 = serialization::load(iter->second); } } { auto&& iter = state.find("front_offset_dim3"); if (iter != state.end()) { opdef.front_offset_dim3 = serialization::load(iter->second); } } { auto&& iter = state.find("front_offset_dim4"); if (iter != state.end()) { opdef.front_offset_dim4 = serialization::load(iter->second); } } { auto&& iter = state.find("front_offset_dim5"); if (iter != state.end()) { opdef.front_offset_dim5 = serialization::load(iter->second); } } { auto&& iter = state.find("front_offset_dim6"); if (iter != state.end()) { opdef.front_offset_dim6 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim0"); if (iter != state.end()) { opdef.back_offset_dim0 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim1"); if (iter != state.end()) { opdef.back_offset_dim1 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim2"); if (iter != state.end()) { opdef.back_offset_dim2 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim3"); if (iter != state.end()) { opdef.back_offset_dim3 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim4"); if (iter != state.end()) { opdef.back_offset_dim4 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim5"); if (iter != state.end()) { opdef.back_offset_dim5 = serialization::load(iter->second); } } { auto&& iter = state.find("back_offset_dim6"); if (iter != state.end()) { opdef.back_offset_dim6 = serialization::load(iter->second); } } { auto&& iter = state.find("padding_val"); if (iter != state.end()) { opdef.padding_val = serialization::load(iter->second); } } { auto&& iter = state.find("padding_mode"); if (iter != state.end()) { opdef.padding_mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Padding) int PyOp(Padding)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"front_offset_dim0", "front_offset_dim1", "front_offset_dim2", "front_offset_dim3", "front_offset_dim4", "front_offset_dim5", "front_offset_dim6", "back_offset_dim0", "back_offset_dim1", "back_offset_dim2", "back_offset_dim3", "back_offset_dim4", "back_offset_dim5", "back_offset_dim6", "padding_val", "padding_mode", "scope", NULL}; PyObject *front_offset_dim0 = NULL, *front_offset_dim1 = NULL, *front_offset_dim2 = NULL, *front_offset_dim3 = NULL, *front_offset_dim4 = NULL, *front_offset_dim5 = NULL, *front_offset_dim6 = NULL, *back_offset_dim0 = NULL, *back_offset_dim1 = NULL, *back_offset_dim2 = NULL, *back_offset_dim3 = NULL, *back_offset_dim4 = NULL, *back_offset_dim5 = NULL, *back_offset_dim6 = NULL, *padding_val = NULL, *padding_mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOOOOOOOO", const_cast(kwlist), &front_offset_dim0, &front_offset_dim1, &front_offset_dim2, &front_offset_dim3, &front_offset_dim4, &front_offset_dim5, &front_offset_dim6, &back_offset_dim0, &back_offset_dim1, &back_offset_dim2, &back_offset_dim3, &back_offset_dim4, &back_offset_dim5, &back_offset_dim6, &padding_val, &padding_mode, &scope)) return -1; if (front_offset_dim0) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim0 = py::cast(py::handle(front_offset_dim0)); } CATCH_ALL(-1) } if (front_offset_dim1) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim1 = py::cast(py::handle(front_offset_dim1)); } CATCH_ALL(-1) } if (front_offset_dim2) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim2 = py::cast(py::handle(front_offset_dim2)); } CATCH_ALL(-1) } if (front_offset_dim3) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim3 = py::cast(py::handle(front_offset_dim3)); } CATCH_ALL(-1) } if (front_offset_dim4) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim4 = py::cast(py::handle(front_offset_dim4)); } CATCH_ALL(-1) } if (front_offset_dim5) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim5 = py::cast(py::handle(front_offset_dim5)); } CATCH_ALL(-1) } if (front_offset_dim6) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().front_offset_dim6 = py::cast(py::handle(front_offset_dim6)); } CATCH_ALL(-1) } if (back_offset_dim0) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim0 = py::cast(py::handle(back_offset_dim0)); } CATCH_ALL(-1) } if (back_offset_dim1) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim1 = py::cast(py::handle(back_offset_dim1)); } CATCH_ALL(-1) } if (back_offset_dim2) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim2 = py::cast(py::handle(back_offset_dim2)); } CATCH_ALL(-1) } if (back_offset_dim3) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim3 = py::cast(py::handle(back_offset_dim3)); } CATCH_ALL(-1) } if (back_offset_dim4) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim4 = py::cast(py::handle(back_offset_dim4)); } CATCH_ALL(-1) } if (back_offset_dim5) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim5 = py::cast(py::handle(back_offset_dim5)); } CATCH_ALL(-1) } if (back_offset_dim6) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().back_offset_dim6 = py::cast(py::handle(back_offset_dim6)); } CATCH_ALL(-1) } if (padding_val) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().padding_val = py::cast(py::handle(padding_val)); } CATCH_ALL(-1) } if (padding_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().padding_mode = py::cast(py::handle(padding_mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Padding)::py_getsetters[] = { {const_cast("front_offset_dim0"), py_get_generic(Padding, front_offset_dim0), py_set_generic(Padding, front_offset_dim0), const_cast("front_offset_dim0"), NULL}, {const_cast("front_offset_dim1"), py_get_generic(Padding, front_offset_dim1), py_set_generic(Padding, front_offset_dim1), const_cast("front_offset_dim1"), NULL}, {const_cast("front_offset_dim2"), py_get_generic(Padding, front_offset_dim2), py_set_generic(Padding, front_offset_dim2), const_cast("front_offset_dim2"), NULL}, {const_cast("front_offset_dim3"), py_get_generic(Padding, front_offset_dim3), py_set_generic(Padding, front_offset_dim3), const_cast("front_offset_dim3"), NULL}, {const_cast("front_offset_dim4"), py_get_generic(Padding, front_offset_dim4), py_set_generic(Padding, front_offset_dim4), const_cast("front_offset_dim4"), NULL}, {const_cast("front_offset_dim5"), py_get_generic(Padding, front_offset_dim5), py_set_generic(Padding, front_offset_dim5), const_cast("front_offset_dim5"), NULL}, {const_cast("front_offset_dim6"), py_get_generic(Padding, front_offset_dim6), py_set_generic(Padding, front_offset_dim6), const_cast("front_offset_dim6"), NULL}, {const_cast("back_offset_dim0"), py_get_generic(Padding, back_offset_dim0), py_set_generic(Padding, back_offset_dim0), const_cast("back_offset_dim0"), NULL}, {const_cast("back_offset_dim1"), py_get_generic(Padding, back_offset_dim1), py_set_generic(Padding, back_offset_dim1), const_cast("back_offset_dim1"), NULL}, {const_cast("back_offset_dim2"), py_get_generic(Padding, back_offset_dim2), py_set_generic(Padding, back_offset_dim2), const_cast("back_offset_dim2"), NULL}, {const_cast("back_offset_dim3"), py_get_generic(Padding, back_offset_dim3), py_set_generic(Padding, back_offset_dim3), const_cast("back_offset_dim3"), NULL}, {const_cast("back_offset_dim4"), py_get_generic(Padding, back_offset_dim4), py_set_generic(Padding, back_offset_dim4), const_cast("back_offset_dim4"), NULL}, {const_cast("back_offset_dim5"), py_get_generic(Padding, back_offset_dim5), py_set_generic(Padding, back_offset_dim5), const_cast("back_offset_dim5"), NULL}, {const_cast("back_offset_dim6"), py_get_generic(Padding, back_offset_dim6), py_set_generic(Padding, back_offset_dim6), const_cast("back_offset_dim6"), NULL}, {const_cast("padding_val"), py_get_generic(Padding, padding_val), py_set_generic(Padding, padding_val), const_cast("padding_val"), NULL}, {const_cast("padding_mode"), py_get_generic(Padding, padding_mode), py_set_generic(Padding, padding_mode), const_cast("padding_mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Padding)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Padding)::getstate, METH_NOARGS, "Padding getstate"}, {const_cast("__setstate__"), PyOp(Padding)::setstate, METH_VARARGS, "Padding setstate"}, {NULL} /* Sentinel */ }; void _init_py_Padding(py::module m) { using py_op = PyOp(Padding); auto& py_type = PyOpType(Padding); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Padding"; py_type.tp_basicsize = sizeof(PyOp(Padding)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Padding"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Padding_PaddingMode(py_type); PyType_Modified(&py_type); m.add_object("Padding", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Padding::typeinfo(), &py_type).second); } PyOpDefBegin(ParamPackConcat) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"offsets", serialization::dump(opdef.offsets)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("offsets"); if (iter != state.end()) { opdef.offsets = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ParamPackConcat) int PyOp(ParamPackConcat)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"offsets", "scope", NULL}; PyObject *offsets = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &offsets, &scope)) return -1; if (offsets) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().offsets = py::cast(py::handle(offsets)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ParamPackConcat)::py_getsetters[] = { {const_cast("offsets"), py_get_generic(ParamPackConcat, offsets), py_set_generic(ParamPackConcat, offsets), const_cast("offsets"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ParamPackConcat)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ParamPackConcat)::getstate, METH_NOARGS, "ParamPackConcat getstate"}, {const_cast("__setstate__"), PyOp(ParamPackConcat)::setstate, METH_VARARGS, "ParamPackConcat setstate"}, {NULL} /* Sentinel */ }; void _init_py_ParamPackConcat(py::module m) { using py_op = PyOp(ParamPackConcat); auto& py_type = PyOpType(ParamPackConcat); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ParamPackConcat"; py_type.tp_basicsize = sizeof(PyOp(ParamPackConcat)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ParamPackConcat"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("ParamPackConcat", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ParamPackConcat::typeinfo(), &py_type).second); } PyOpDefBegin(ParamPackSplit) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"offsets", serialization::dump(opdef.offsets)}, {"shapes", serialization::dump(opdef.shapes)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("offsets"); if (iter != state.end()) { opdef.offsets = serialization::load(iter->second); } } { auto&& iter = state.find("shapes"); if (iter != state.end()) { opdef.shapes = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ParamPackSplit) int PyOp(ParamPackSplit)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"offsets", "shapes", "scope", NULL}; PyObject *offsets = NULL, *shapes = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &offsets, &shapes, &scope)) return -1; if (offsets) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().offsets = py::cast(py::handle(offsets)); } CATCH_ALL(-1) } if (shapes) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().shapes = py::cast(py::handle(shapes)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ParamPackSplit)::py_getsetters[] = { {const_cast("offsets"), py_get_generic(ParamPackSplit, offsets), py_set_generic(ParamPackSplit, offsets), const_cast("offsets"), NULL}, {const_cast("shapes"), py_get_generic(ParamPackSplit, shapes), py_set_generic(ParamPackSplit, shapes), const_cast("shapes"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ParamPackSplit)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ParamPackSplit)::getstate, METH_NOARGS, "ParamPackSplit getstate"}, {const_cast("__setstate__"), PyOp(ParamPackSplit)::setstate, METH_VARARGS, "ParamPackSplit setstate"}, {NULL} /* Sentinel */ }; void _init_py_ParamPackSplit(py::module m) { using py_op = PyOp(ParamPackSplit); auto& py_type = PyOpType(ParamPackSplit); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ParamPackSplit"; py_type.tp_basicsize = sizeof(PyOp(ParamPackSplit)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ParamPackSplit"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("ParamPackSplit", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ParamPackSplit::typeinfo(), &py_type).second); } PyOpDefBegin(PermutationRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"dtype", serialization::dump(opdef.dtype)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(PermutationRNG) int PyOp(PermutationRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "dtype", "handle", "scope", NULL}; PyObject *seed = NULL, *dtype = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOO", const_cast(kwlist), &seed, &dtype, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(PermutationRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(PermutationRNG, seed), py_set_generic(PermutationRNG, seed), const_cast("seed"), NULL}, {const_cast("dtype"), py_get_generic(PermutationRNG, dtype), py_set_generic(PermutationRNG, dtype), const_cast("dtype"), NULL}, {const_cast("handle"), py_get_generic(PermutationRNG, handle), py_set_generic(PermutationRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(PermutationRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(PermutationRNG)::getstate, METH_NOARGS, "PermutationRNG getstate"}, {const_cast("__setstate__"), PyOp(PermutationRNG)::setstate, METH_VARARGS, "PermutationRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_PermutationRNG(py::module m) { using py_op = PyOp(PermutationRNG); auto& py_type = PyOpType(PermutationRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.PermutationRNG"; py_type.tp_basicsize = sizeof(PyOp(PermutationRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "PermutationRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("PermutationRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(PermutationRNG::typeinfo(), &py_type).second); } PyOpDefBegin(PixelShuffle) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"factor", serialization::dump(opdef.factor)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("factor"); if (iter != state.end()) { opdef.factor = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(PixelShuffle) int PyOp(PixelShuffle)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"factor", "scope", NULL}; PyObject *factor = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &factor, &scope)) return -1; if (factor) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().factor = py::cast(py::handle(factor)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(PixelShuffle)::py_getsetters[] = { {const_cast("factor"), py_get_generic(PixelShuffle, factor), py_set_generic(PixelShuffle, factor), const_cast("factor"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(PixelShuffle)::tp_methods[] = { {const_cast("__getstate__"), PyOp(PixelShuffle)::getstate, METH_NOARGS, "PixelShuffle getstate"}, {const_cast("__setstate__"), PyOp(PixelShuffle)::setstate, METH_VARARGS, "PixelShuffle setstate"}, {NULL} /* Sentinel */ }; void _init_py_PixelShuffle(py::module m) { using py_op = PyOp(PixelShuffle); auto& py_type = PyOpType(PixelShuffle); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.PixelShuffle"; py_type.tp_basicsize = sizeof(PyOp(PixelShuffle)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "PixelShuffle"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("PixelShuffle", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(PixelShuffle::typeinfo(), &py_type).second); } PyOpDefBegin(PixelShuffleBackward) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"factor", serialization::dump(opdef.factor)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("factor"); if (iter != state.end()) { opdef.factor = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(PixelShuffleBackward) int PyOp(PixelShuffleBackward)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"factor", "scope", NULL}; PyObject *factor = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &factor, &scope)) return -1; if (factor) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().factor = py::cast(py::handle(factor)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(PixelShuffleBackward)::py_getsetters[] = { {const_cast("factor"), py_get_generic(PixelShuffleBackward, factor), py_set_generic(PixelShuffleBackward, factor), const_cast("factor"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(PixelShuffleBackward)::tp_methods[] = { {const_cast("__getstate__"), PyOp(PixelShuffleBackward)::getstate, METH_NOARGS, "PixelShuffleBackward getstate"}, {const_cast("__setstate__"), PyOp(PixelShuffleBackward)::setstate, METH_VARARGS, "PixelShuffleBackward setstate"}, {NULL} /* Sentinel */ }; void _init_py_PixelShuffleBackward(py::module m) { using py_op = PyOp(PixelShuffleBackward); auto& py_type = PyOpType(PixelShuffleBackward); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.PixelShuffleBackward"; py_type.tp_basicsize = sizeof(PyOp(PixelShuffleBackward)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "PixelShuffleBackward"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("PixelShuffleBackward", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(PixelShuffleBackward::typeinfo(), &py_type).second); } PyOpDefBegin(PoissonRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(PoissonRNG) int PyOp(PoissonRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "handle", "scope", NULL}; PyObject *seed = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &seed, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(PoissonRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(PoissonRNG, seed), py_set_generic(PoissonRNG, seed), const_cast("seed"), NULL}, {const_cast("handle"), py_get_generic(PoissonRNG, handle), py_set_generic(PoissonRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(PoissonRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(PoissonRNG)::getstate, METH_NOARGS, "PoissonRNG getstate"}, {const_cast("__setstate__"), PyOp(PoissonRNG)::setstate, METH_VARARGS, "PoissonRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_PoissonRNG(py::module m) { using py_op = PyOp(PoissonRNG); auto& py_type = PyOpType(PoissonRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.PoissonRNG"; py_type.tp_basicsize = sizeof(PyOp(PoissonRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "PoissonRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("PoissonRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(PoissonRNG::typeinfo(), &py_type).second); } void _init_py_Pooling_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_Pooling_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } void _init_py_Pooling_Strategy(PyTypeObject& py_type) { auto& e_type = BitCombinedEnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Strategy", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Pooling) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"window_h", serialization::dump(opdef.window_h)}, {"window_w", serialization::dump(opdef.window_w)}, {"format", serialization::dump(opdef.format)}, {"strategy", serialization::dump(opdef.strategy)}, {"workspace_limit", serialization::dump(opdef.workspace_limit)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("window_h"); if (iter != state.end()) { opdef.window_h = serialization::load(iter->second); } } { auto&& iter = state.find("window_w"); if (iter != state.end()) { opdef.window_w = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("strategy"); if (iter != state.end()) { opdef.strategy = serialization::load(iter->second); } } { auto&& iter = state.find("workspace_limit"); if (iter != state.end()) { opdef.workspace_limit = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Pooling) int PyOp(Pooling)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "pad_h", "pad_w", "stride_h", "stride_w", "window_h", "window_w", "format", "strategy", "workspace_limit", "scope", NULL}; PyObject *mode = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *window_h = NULL, *window_w = NULL, *format = NULL, *strategy = NULL, *workspace_limit = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOO", const_cast(kwlist), &mode, &pad_h, &pad_w, &stride_h, &stride_w, &window_h, &window_w, &format, &strategy, &workspace_limit, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (window_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().window_h = py::cast(py::handle(window_h)); } CATCH_ALL(-1) } if (window_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().window_w = py::cast(py::handle(window_w)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (strategy) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().strategy = py::cast(py::handle(strategy)); } CATCH_ALL(-1) } if (workspace_limit) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().workspace_limit = py::cast(py::handle(workspace_limit)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Pooling)::py_getsetters[] = { {const_cast("mode"), py_get_generic(Pooling, mode), py_set_generic(Pooling, mode), const_cast("mode"), NULL}, {const_cast("pad_h"), py_get_generic(Pooling, pad_h), py_set_generic(Pooling, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(Pooling, pad_w), py_set_generic(Pooling, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(Pooling, stride_h), py_set_generic(Pooling, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(Pooling, stride_w), py_set_generic(Pooling, stride_w), const_cast("stride_w"), NULL}, {const_cast("window_h"), py_get_generic(Pooling, window_h), py_set_generic(Pooling, window_h), const_cast("window_h"), NULL}, {const_cast("window_w"), py_get_generic(Pooling, window_w), py_set_generic(Pooling, window_w), const_cast("window_w"), NULL}, {const_cast("format"), py_get_generic(Pooling, format), py_set_generic(Pooling, format), const_cast("format"), NULL}, {const_cast("strategy"), py_get_generic(Pooling, strategy), py_set_generic(Pooling, strategy), const_cast("strategy"), NULL}, {const_cast("workspace_limit"), py_get_generic(Pooling, workspace_limit), py_set_generic(Pooling, workspace_limit), const_cast("workspace_limit"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Pooling)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Pooling)::getstate, METH_NOARGS, "Pooling getstate"}, {const_cast("__setstate__"), PyOp(Pooling)::setstate, METH_VARARGS, "Pooling setstate"}, {NULL} /* Sentinel */ }; void _init_py_Pooling(py::module m) { using py_op = PyOp(Pooling); auto& py_type = PyOpType(Pooling); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Pooling"; py_type.tp_basicsize = sizeof(PyOp(Pooling)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Pooling"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Pooling_Mode(py_type); _init_py_Pooling_Format(py_type); _init_py_Pooling_Strategy(py_type); PyType_Modified(&py_type); m.add_object("Pooling", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Pooling::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "RNN.NonlineMode"; static constexpr std::underlying_type_t max = 3 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"IDENTITY", "RELU", "TANH"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("IDENTITY"), RNN::NonlineMode::IDENTITY}, {normalize_enum("RELU"), RNN::NonlineMode::RELU}, {normalize_enum("TANH"), RNN::NonlineMode::TANH}}; template<> PyObject* EnumWrapper::pyobj_insts[3] = {nullptr}; void _init_py_RNN_NonlineMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.RNN.NonlineMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("NonlineMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("RNN.NonlineMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = RNN::NonlineMode::IDENTITY; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "IDENTITY", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = RNN::NonlineMode::RELU; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "RELU", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = RNN::NonlineMode::TANH; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "TANH", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "NonlineMode", reinterpret_cast(e_type)) >= 0); } void _init_py_RNN_FwdMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "FwdMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(RNN) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"num_layers", serialization::dump(opdef.num_layers)}, {"bidirectional", serialization::dump(opdef.bidirectional)}, {"bias", serialization::dump(opdef.bias)}, {"hidden_size", serialization::dump(opdef.hidden_size)}, {"dropout", serialization::dump(opdef.dropout)}, {"nonlineMode", serialization::dump(opdef.nonlineMode)}, {"fwd_mode", serialization::dump(opdef.fwd_mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("num_layers"); if (iter != state.end()) { opdef.num_layers = serialization::load(iter->second); } } { auto&& iter = state.find("bidirectional"); if (iter != state.end()) { opdef.bidirectional = serialization::load(iter->second); } } { auto&& iter = state.find("bias"); if (iter != state.end()) { opdef.bias = serialization::load(iter->second); } } { auto&& iter = state.find("hidden_size"); if (iter != state.end()) { opdef.hidden_size = serialization::load(iter->second); } } { auto&& iter = state.find("dropout"); if (iter != state.end()) { opdef.dropout = serialization::load(iter->second); } } { auto&& iter = state.find("nonlineMode"); if (iter != state.end()) { opdef.nonlineMode = serialization::load(iter->second); } } { auto&& iter = state.find("fwd_mode"); if (iter != state.end()) { opdef.fwd_mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(RNN) int PyOp(RNN)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"num_layers", "bidirectional", "bias", "hidden_size", "dropout", "nonlineMode", "fwd_mode", "scope", NULL}; PyObject *num_layers = NULL, *bidirectional = NULL, *bias = NULL, *hidden_size = NULL, *dropout = NULL, *nonlineMode = NULL, *fwd_mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOO", const_cast(kwlist), &num_layers, &bidirectional, &bias, &hidden_size, &dropout, &nonlineMode, &fwd_mode, &scope)) return -1; if (num_layers) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().num_layers = py::cast(py::handle(num_layers)); } CATCH_ALL(-1) } if (bidirectional) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bidirectional = py::cast(py::handle(bidirectional)); } CATCH_ALL(-1) } if (bias) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bias = py::cast(py::handle(bias)); } CATCH_ALL(-1) } if (hidden_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().hidden_size = py::cast(py::handle(hidden_size)); } CATCH_ALL(-1) } if (dropout) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dropout = py::cast(py::handle(dropout)); } CATCH_ALL(-1) } if (nonlineMode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nonlineMode = py::cast(py::handle(nonlineMode)); } CATCH_ALL(-1) } if (fwd_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().fwd_mode = py::cast(py::handle(fwd_mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(RNN)::py_getsetters[] = { {const_cast("num_layers"), py_get_generic(RNN, num_layers), py_set_generic(RNN, num_layers), const_cast("num_layers"), NULL}, {const_cast("bidirectional"), py_get_generic(RNN, bidirectional), py_set_generic(RNN, bidirectional), const_cast("bidirectional"), NULL}, {const_cast("bias"), py_get_generic(RNN, bias), py_set_generic(RNN, bias), const_cast("bias"), NULL}, {const_cast("hidden_size"), py_get_generic(RNN, hidden_size), py_set_generic(RNN, hidden_size), const_cast("hidden_size"), NULL}, {const_cast("dropout"), py_get_generic(RNN, dropout), py_set_generic(RNN, dropout), const_cast("dropout"), NULL}, {const_cast("nonlineMode"), py_get_generic(RNN, nonlineMode), py_set_generic(RNN, nonlineMode), const_cast("nonlineMode"), NULL}, {const_cast("fwd_mode"), py_get_generic(RNN, fwd_mode), py_set_generic(RNN, fwd_mode), const_cast("fwd_mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(RNN)::tp_methods[] = { {const_cast("__getstate__"), PyOp(RNN)::getstate, METH_NOARGS, "RNN getstate"}, {const_cast("__setstate__"), PyOp(RNN)::setstate, METH_VARARGS, "RNN setstate"}, {NULL} /* Sentinel */ }; void _init_py_RNN(py::module m) { using py_op = PyOp(RNN); auto& py_type = PyOpType(RNN); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.RNN"; py_type.tp_basicsize = sizeof(PyOp(RNN)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "RNN"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_RNN_NonlineMode(py_type); _init_py_RNN_FwdMode(py_type); PyType_Modified(&py_type); m.add_object("RNN", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(RNN::typeinfo(), &py_type).second); } void _init_py_RNNCell_NonlineMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "NonlineMode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(RNNCell) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"nonlineMode", serialization::dump(opdef.nonlineMode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("nonlineMode"); if (iter != state.end()) { opdef.nonlineMode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(RNNCell) int PyOp(RNNCell)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"nonlineMode", "scope", NULL}; PyObject *nonlineMode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &nonlineMode, &scope)) return -1; if (nonlineMode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nonlineMode = py::cast(py::handle(nonlineMode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(RNNCell)::py_getsetters[] = { {const_cast("nonlineMode"), py_get_generic(RNNCell, nonlineMode), py_set_generic(RNNCell, nonlineMode), const_cast("nonlineMode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(RNNCell)::tp_methods[] = { {const_cast("__getstate__"), PyOp(RNNCell)::getstate, METH_NOARGS, "RNNCell getstate"}, {const_cast("__setstate__"), PyOp(RNNCell)::setstate, METH_VARARGS, "RNNCell setstate"}, {NULL} /* Sentinel */ }; void _init_py_RNNCell(py::module m) { using py_op = PyOp(RNNCell); auto& py_type = PyOpType(RNNCell); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.RNNCell"; py_type.tp_basicsize = sizeof(PyOp(RNNCell)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "RNNCell"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_RNNCell_NonlineMode(py_type); PyType_Modified(&py_type); m.add_object("RNNCell", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(RNNCell::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "ROIAlign.Mode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"MAX", "AVERAGE"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("MAX"), ROIAlign::Mode::MAX}, {normalize_enum("AVERAGE"), ROIAlign::Mode::AVERAGE}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_ROIAlign_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.ROIAlign.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("ROIAlign.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ROIAlign::Mode::MAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MAX", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ROIAlign::Mode::AVERAGE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "AVERAGE", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } void _init_py_ROIAlign_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(ROIAlign) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"format", serialization::dump(opdef.format)}, {"spatial_scale", serialization::dump(opdef.spatial_scale)}, {"offset", serialization::dump(opdef.offset)}, {"pooled_height", serialization::dump(opdef.pooled_height)}, {"pooled_width", serialization::dump(opdef.pooled_width)}, {"sample_height", serialization::dump(opdef.sample_height)}, {"sample_width", serialization::dump(opdef.sample_width)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("spatial_scale"); if (iter != state.end()) { opdef.spatial_scale = serialization::load(iter->second); } } { auto&& iter = state.find("offset"); if (iter != state.end()) { opdef.offset = serialization::load(iter->second); } } { auto&& iter = state.find("pooled_height"); if (iter != state.end()) { opdef.pooled_height = serialization::load(iter->second); } } { auto&& iter = state.find("pooled_width"); if (iter != state.end()) { opdef.pooled_width = serialization::load(iter->second); } } { auto&& iter = state.find("sample_height"); if (iter != state.end()) { opdef.sample_height = serialization::load(iter->second); } } { auto&& iter = state.find("sample_width"); if (iter != state.end()) { opdef.sample_width = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ROIAlign) int PyOp(ROIAlign)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "format", "spatial_scale", "offset", "pooled_height", "pooled_width", "sample_height", "sample_width", "scope", NULL}; PyObject *mode = NULL, *format = NULL, *spatial_scale = NULL, *offset = NULL, *pooled_height = NULL, *pooled_width = NULL, *sample_height = NULL, *sample_width = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOO", const_cast(kwlist), &mode, &format, &spatial_scale, &offset, &pooled_height, &pooled_width, &sample_height, &sample_width, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (spatial_scale) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().spatial_scale = py::cast(py::handle(spatial_scale)); } CATCH_ALL(-1) } if (offset) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().offset = py::cast(py::handle(offset)); } CATCH_ALL(-1) } if (pooled_height) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pooled_height = py::cast(py::handle(pooled_height)); } CATCH_ALL(-1) } if (pooled_width) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pooled_width = py::cast(py::handle(pooled_width)); } CATCH_ALL(-1) } if (sample_height) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sample_height = py::cast(py::handle(sample_height)); } CATCH_ALL(-1) } if (sample_width) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().sample_width = py::cast(py::handle(sample_width)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ROIAlign)::py_getsetters[] = { {const_cast("mode"), py_get_generic(ROIAlign, mode), py_set_generic(ROIAlign, mode), const_cast("mode"), NULL}, {const_cast("format"), py_get_generic(ROIAlign, format), py_set_generic(ROIAlign, format), const_cast("format"), NULL}, {const_cast("spatial_scale"), py_get_generic(ROIAlign, spatial_scale), py_set_generic(ROIAlign, spatial_scale), const_cast("spatial_scale"), NULL}, {const_cast("offset"), py_get_generic(ROIAlign, offset), py_set_generic(ROIAlign, offset), const_cast("offset"), NULL}, {const_cast("pooled_height"), py_get_generic(ROIAlign, pooled_height), py_set_generic(ROIAlign, pooled_height), const_cast("pooled_height"), NULL}, {const_cast("pooled_width"), py_get_generic(ROIAlign, pooled_width), py_set_generic(ROIAlign, pooled_width), const_cast("pooled_width"), NULL}, {const_cast("sample_height"), py_get_generic(ROIAlign, sample_height), py_set_generic(ROIAlign, sample_height), const_cast("sample_height"), NULL}, {const_cast("sample_width"), py_get_generic(ROIAlign, sample_width), py_set_generic(ROIAlign, sample_width), const_cast("sample_width"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ROIAlign)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ROIAlign)::getstate, METH_NOARGS, "ROIAlign getstate"}, {const_cast("__setstate__"), PyOp(ROIAlign)::setstate, METH_VARARGS, "ROIAlign setstate"}, {NULL} /* Sentinel */ }; void _init_py_ROIAlign(py::module m) { using py_op = PyOp(ROIAlign); auto& py_type = PyOpType(ROIAlign); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ROIAlign"; py_type.tp_basicsize = sizeof(PyOp(ROIAlign)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ROIAlign"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_ROIAlign_Mode(py_type); _init_py_ROIAlign_Format(py_type); PyType_Modified(&py_type); m.add_object("ROIAlign", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ROIAlign::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "ROIPooling.Mode"; static constexpr std::underlying_type_t max = 2 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"MAX", "AVERAGE"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("MAX"), ROIPooling::Mode::MAX}, {normalize_enum("AVERAGE"), ROIPooling::Mode::AVERAGE}}; template<> PyObject* EnumWrapper::pyobj_insts[2] = {nullptr}; void _init_py_ROIPooling_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.ROIPooling.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("ROIPooling.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ROIPooling::Mode::MAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MAX", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = ROIPooling::Mode::AVERAGE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "AVERAGE", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(ROIPooling) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"scale", serialization::dump(opdef.scale)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("scale"); if (iter != state.end()) { opdef.scale = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ROIPooling) int PyOp(ROIPooling)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "scale", "scope", NULL}; PyObject *mode = NULL, *scale = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &mode, &scale, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (scale) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().scale = py::cast(py::handle(scale)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ROIPooling)::py_getsetters[] = { {const_cast("mode"), py_get_generic(ROIPooling, mode), py_set_generic(ROIPooling, mode), const_cast("mode"), NULL}, {const_cast("scale"), py_get_generic(ROIPooling, scale), py_set_generic(ROIPooling, scale), const_cast("scale"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ROIPooling)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ROIPooling)::getstate, METH_NOARGS, "ROIPooling getstate"}, {const_cast("__setstate__"), PyOp(ROIPooling)::setstate, METH_VARARGS, "ROIPooling setstate"}, {NULL} /* Sentinel */ }; void _init_py_ROIPooling(py::module m) { using py_op = PyOp(ROIPooling); auto& py_type = PyOpType(ROIPooling); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ROIPooling"; py_type.tp_basicsize = sizeof(PyOp(ROIPooling)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ROIPooling"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_ROIPooling_Mode(py_type); PyType_Modified(&py_type); m.add_object("ROIPooling", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ROIPooling::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Reduce.Mode"; static constexpr std::underlying_type_t max = 6 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"SUM", "SUM_SQR", "PRODUCT", "MIN", "MAX", "MEAN"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("SUM"), Reduce::Mode::SUM}, {normalize_enum("SUM_SQR"), Reduce::Mode::SUM_SQR}, {normalize_enum("PRODUCT"), Reduce::Mode::PRODUCT}, {normalize_enum("MIN"), Reduce::Mode::MIN}, {normalize_enum("MAX"), Reduce::Mode::MAX}, {normalize_enum("MEAN"), Reduce::Mode::MEAN}}; template<> PyObject* EnumWrapper::pyobj_insts[6] = {nullptr}; void _init_py_Reduce_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Reduce.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Reduce.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::Mode::SUM; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SUM", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::Mode::SUM_SQR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "SUM_SQR", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::Mode::PRODUCT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "PRODUCT", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::Mode::MIN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MIN", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::Mode::MAX; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MAX", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::Mode::MEAN; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "MEAN", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "Reduce.DataType"; static constexpr std::underlying_type_t max = 6 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"DEFAULT", "FLOAT_IO16xC32", "FLOAT_O32xC32", "FLOAT_O16xC32", "QUINT_I8xO32", "QINT_I8xO32"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("DEFAULT"), Reduce::DataType::DEFAULT}, {normalize_enum("FLOAT_IO16xC32"), Reduce::DataType::FLOAT_IO16xC32}, {normalize_enum("FLOAT_O32xC32"), Reduce::DataType::FLOAT_O32xC32}, {normalize_enum("FLOAT_O16xC32"), Reduce::DataType::FLOAT_O16xC32}, {normalize_enum("QUINT_I8xO32"), Reduce::DataType::QUINT_I8xO32}, {normalize_enum("QINT_I8xO32"), Reduce::DataType::QINT_I8xO32}}; template<> PyObject* EnumWrapper::pyobj_insts[6] = {nullptr}; void _init_py_Reduce_DataType(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Reduce.DataType", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("DataType").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Reduce.DataType").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::DataType::DEFAULT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "DEFAULT", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::DataType::FLOAT_IO16xC32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT_IO16xC32", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::DataType::FLOAT_O32xC32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT_O32xC32", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::DataType::FLOAT_O16xC32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "FLOAT_O16xC32", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::DataType::QUINT_I8xO32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QUINT_I8xO32", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Reduce::DataType::QINT_I8xO32; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "QINT_I8xO32", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "DataType", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Reduce) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)}, {"axis", serialization::dump(opdef.axis)}, {"data_type", serialization::dump(opdef.data_type)}, {"keepdim", serialization::dump(opdef.keepdim)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("data_type"); if (iter != state.end()) { opdef.data_type = serialization::load(iter->second); } } { auto&& iter = state.find("keepdim"); if (iter != state.end()) { opdef.keepdim = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Reduce) int PyOp(Reduce)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "axis", "data_type", "keepdim", "scope", NULL}; PyObject *mode = NULL, *axis = NULL, *data_type = NULL, *keepdim = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &mode, &axis, &data_type, &keepdim, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (data_type) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().data_type = py::cast(py::handle(data_type)); } CATCH_ALL(-1) } if (keepdim) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().keepdim = py::cast(py::handle(keepdim)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Reduce)::py_getsetters[] = { {const_cast("mode"), py_get_generic(Reduce, mode), py_set_generic(Reduce, mode), const_cast("mode"), NULL}, {const_cast("axis"), py_get_generic(Reduce, axis), py_set_generic(Reduce, axis), const_cast("axis"), NULL}, {const_cast("data_type"), py_get_generic(Reduce, data_type), py_set_generic(Reduce, data_type), const_cast("data_type"), NULL}, {const_cast("keepdim"), py_get_generic(Reduce, keepdim), py_set_generic(Reduce, keepdim), const_cast("keepdim"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Reduce)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Reduce)::getstate, METH_NOARGS, "Reduce getstate"}, {const_cast("__setstate__"), PyOp(Reduce)::setstate, METH_VARARGS, "Reduce setstate"}, {NULL} /* Sentinel */ }; void _init_py_Reduce(py::module m) { using py_op = PyOp(Reduce); auto& py_type = PyOpType(Reduce); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Reduce"; py_type.tp_basicsize = sizeof(PyOp(Reduce)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Reduce"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Reduce_Mode(py_type); _init_py_Reduce_DataType(py_type); PyType_Modified(&py_type); m.add_object("Reduce", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Reduce::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "Remap.InterpolationMode"; static constexpr std::underlying_type_t max = 5 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"NEAREST", "LINEAR", "AREA", "CUBIC", "LANCZOS4"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("NEAREST"), Remap::InterpolationMode::NEAREST}, {normalize_enum("LINEAR"), Remap::InterpolationMode::LINEAR}, {normalize_enum("AREA"), Remap::InterpolationMode::AREA}, {normalize_enum("CUBIC"), Remap::InterpolationMode::CUBIC}, {normalize_enum("LANCZOS4"), Remap::InterpolationMode::LANCZOS4}}; template<> PyObject* EnumWrapper::pyobj_insts[5] = {nullptr}; void _init_py_Remap_InterpolationMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Remap.InterpolationMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("InterpolationMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Remap.InterpolationMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::InterpolationMode::NEAREST; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "NEAREST", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::InterpolationMode::LINEAR; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LINEAR", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::InterpolationMode::AREA; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "AREA", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::InterpolationMode::CUBIC; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CUBIC", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::InterpolationMode::LANCZOS4; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "LANCZOS4", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "InterpolationMode", reinterpret_cast(e_type)) >= 0); } template<> struct EnumTrait { static constexpr const char *name = "Remap.BorderMode"; static constexpr std::underlying_type_t max = 7 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"REPLICATE", "REFLECT", "REFLECT_101", "WRAP", "CONSTANT", "TRANSPARENT", "ISOLATED"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("REPLICATE"), Remap::BorderMode::REPLICATE}, {normalize_enum("REFLECT"), Remap::BorderMode::REFLECT}, {normalize_enum("REFLECT_101"), Remap::BorderMode::REFLECT_101}, {normalize_enum("WRAP"), Remap::BorderMode::WRAP}, {normalize_enum("CONSTANT"), Remap::BorderMode::CONSTANT}, {normalize_enum("TRANSPARENT"), Remap::BorderMode::TRANSPARENT}, {normalize_enum("ISOLATED"), Remap::BorderMode::ISOLATED}}; template<> PyObject* EnumWrapper::pyobj_insts[7] = {nullptr}; void _init_py_Remap_BorderMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.Remap.BorderMode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("BorderMode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("Remap.BorderMode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::REPLICATE; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REPLICATE", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::REFLECT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REFLECT", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::REFLECT_101; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "REFLECT_101", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::WRAP; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "WRAP", inst) >= 0); EnumWrapper::pyobj_insts[3] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::CONSTANT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "CONSTANT", inst) >= 0); EnumWrapper::pyobj_insts[4] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::TRANSPARENT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "TRANSPARENT", inst) >= 0); EnumWrapper::pyobj_insts[5] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = Remap::BorderMode::ISOLATED; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "ISOLATED", inst) >= 0); EnumWrapper::pyobj_insts[6] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "BorderMode", reinterpret_cast(e_type)) >= 0); } void _init_py_Remap_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Remap) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"imode", serialization::dump(opdef.imode)}, {"border_type", serialization::dump(opdef.border_type)}, {"format", serialization::dump(opdef.format)}, {"scalar", serialization::dump(opdef.scalar)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("imode"); if (iter != state.end()) { opdef.imode = serialization::load(iter->second); } } { auto&& iter = state.find("border_type"); if (iter != state.end()) { opdef.border_type = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("scalar"); if (iter != state.end()) { opdef.scalar = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Remap) int PyOp(Remap)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"imode", "border_type", "format", "scalar", "scope", NULL}; PyObject *imode = NULL, *border_type = NULL, *format = NULL, *scalar = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &imode, &border_type, &format, &scalar, &scope)) return -1; if (imode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().imode = py::cast(py::handle(imode)); } CATCH_ALL(-1) } if (border_type) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().border_type = py::cast(py::handle(border_type)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (scalar) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().scalar = py::cast(py::handle(scalar)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Remap)::py_getsetters[] = { {const_cast("imode"), py_get_generic(Remap, imode), py_set_generic(Remap, imode), const_cast("imode"), NULL}, {const_cast("border_type"), py_get_generic(Remap, border_type), py_set_generic(Remap, border_type), const_cast("border_type"), NULL}, {const_cast("format"), py_get_generic(Remap, format), py_set_generic(Remap, format), const_cast("format"), NULL}, {const_cast("scalar"), py_get_generic(Remap, scalar), py_set_generic(Remap, scalar), const_cast("scalar"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Remap)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Remap)::getstate, METH_NOARGS, "Remap getstate"}, {const_cast("__setstate__"), PyOp(Remap)::setstate, METH_VARARGS, "Remap setstate"}, {NULL} /* Sentinel */ }; void _init_py_Remap(py::module m) { using py_op = PyOp(Remap); auto& py_type = PyOpType(Remap); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Remap"; py_type.tp_basicsize = sizeof(PyOp(Remap)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Remap"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Remap_InterpolationMode(py_type); _init_py_Remap_BorderMode(py_type); _init_py_Remap_Format(py_type); PyType_Modified(&py_type); m.add_object("Remap", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Remap::typeinfo(), &py_type).second); } PyOpDefBegin(RemoteRecv) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"key", serialization::dump(opdef.key)}, {"addr", serialization::dump(opdef.addr)}, {"port", serialization::dump(opdef.port)}, {"rank_from", serialization::dump(opdef.rank_from)}, {"cn", serialization::dump(opdef.cn)}, {"shape", serialization::dump(opdef.shape)}, {"dtype", serialization::dump(opdef.dtype)}, {"backend", serialization::dump(opdef.backend)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("key"); if (iter != state.end()) { opdef.key = serialization::load(iter->second); } } { auto&& iter = state.find("addr"); if (iter != state.end()) { opdef.addr = serialization::load(iter->second); } } { auto&& iter = state.find("port"); if (iter != state.end()) { opdef.port = serialization::load(iter->second); } } { auto&& iter = state.find("rank_from"); if (iter != state.end()) { opdef.rank_from = serialization::load(iter->second); } } { auto&& iter = state.find("cn"); if (iter != state.end()) { opdef.cn = serialization::load(iter->second); } } { auto&& iter = state.find("shape"); if (iter != state.end()) { opdef.shape = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } { auto&& iter = state.find("backend"); if (iter != state.end()) { opdef.backend = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(RemoteRecv) int PyOp(RemoteRecv)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"key", "addr", "port", "rank_from", "cn", "shape", "dtype", "backend", "scope", NULL}; PyObject *key = NULL, *addr = NULL, *port = NULL, *rank_from = NULL, *cn = NULL, *shape = NULL, *dtype = NULL, *backend = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOO", const_cast(kwlist), &key, &addr, &port, &rank_from, &cn, &shape, &dtype, &backend, &scope)) return -1; if (key) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().key = py::cast(py::handle(key)); } CATCH_ALL(-1) } if (addr) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().addr = py::cast(py::handle(addr)); } CATCH_ALL(-1) } if (port) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().port = py::cast(py::handle(port)); } CATCH_ALL(-1) } if (rank_from) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().rank_from = py::cast(py::handle(rank_from)); } CATCH_ALL(-1) } if (cn) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().cn = py::cast(py::handle(cn)); } CATCH_ALL(-1) } if (shape) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().shape = py::cast(py::handle(shape)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (backend) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().backend = py::cast(py::handle(backend)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(RemoteRecv)::py_getsetters[] = { {const_cast("key"), py_get_generic(RemoteRecv, key), py_set_generic(RemoteRecv, key), const_cast("key"), NULL}, {const_cast("addr"), py_get_generic(RemoteRecv, addr), py_set_generic(RemoteRecv, addr), const_cast("addr"), NULL}, {const_cast("port"), py_get_generic(RemoteRecv, port), py_set_generic(RemoteRecv, port), const_cast("port"), NULL}, {const_cast("rank_from"), py_get_generic(RemoteRecv, rank_from), py_set_generic(RemoteRecv, rank_from), const_cast("rank_from"), NULL}, {const_cast("cn"), py_get_generic(RemoteRecv, cn), py_set_generic(RemoteRecv, cn), const_cast("cn"), NULL}, {const_cast("shape"), py_get_generic(RemoteRecv, shape), py_set_generic(RemoteRecv, shape), const_cast("shape"), NULL}, {const_cast("dtype"), py_get_generic(RemoteRecv, dtype), py_set_generic(RemoteRecv, dtype), const_cast("dtype"), NULL}, {const_cast("backend"), py_get_generic(RemoteRecv, backend), py_set_generic(RemoteRecv, backend), const_cast("backend"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(RemoteRecv)::tp_methods[] = { {const_cast("__getstate__"), PyOp(RemoteRecv)::getstate, METH_NOARGS, "RemoteRecv getstate"}, {const_cast("__setstate__"), PyOp(RemoteRecv)::setstate, METH_VARARGS, "RemoteRecv setstate"}, {NULL} /* Sentinel */ }; void _init_py_RemoteRecv(py::module m) { using py_op = PyOp(RemoteRecv); auto& py_type = PyOpType(RemoteRecv); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.RemoteRecv"; py_type.tp_basicsize = sizeof(PyOp(RemoteRecv)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "RemoteRecv"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("RemoteRecv", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(RemoteRecv::typeinfo(), &py_type).second); } PyOpDefBegin(RemoteSend) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"key", serialization::dump(opdef.key)}, {"addr", serialization::dump(opdef.addr)}, {"port", serialization::dump(opdef.port)}, {"rank_to", serialization::dump(opdef.rank_to)}, {"backend", serialization::dump(opdef.backend)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("key"); if (iter != state.end()) { opdef.key = serialization::load(iter->second); } } { auto&& iter = state.find("addr"); if (iter != state.end()) { opdef.addr = serialization::load(iter->second); } } { auto&& iter = state.find("port"); if (iter != state.end()) { opdef.port = serialization::load(iter->second); } } { auto&& iter = state.find("rank_to"); if (iter != state.end()) { opdef.rank_to = serialization::load(iter->second); } } { auto&& iter = state.find("backend"); if (iter != state.end()) { opdef.backend = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(RemoteSend) int PyOp(RemoteSend)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"key", "addr", "port", "rank_to", "backend", "scope", NULL}; PyObject *key = NULL, *addr = NULL, *port = NULL, *rank_to = NULL, *backend = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOO", const_cast(kwlist), &key, &addr, &port, &rank_to, &backend, &scope)) return -1; if (key) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().key = py::cast(py::handle(key)); } CATCH_ALL(-1) } if (addr) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().addr = py::cast(py::handle(addr)); } CATCH_ALL(-1) } if (port) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().port = py::cast(py::handle(port)); } CATCH_ALL(-1) } if (rank_to) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().rank_to = py::cast(py::handle(rank_to)); } CATCH_ALL(-1) } if (backend) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().backend = py::cast(py::handle(backend)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(RemoteSend)::py_getsetters[] = { {const_cast("key"), py_get_generic(RemoteSend, key), py_set_generic(RemoteSend, key), const_cast("key"), NULL}, {const_cast("addr"), py_get_generic(RemoteSend, addr), py_set_generic(RemoteSend, addr), const_cast("addr"), NULL}, {const_cast("port"), py_get_generic(RemoteSend, port), py_set_generic(RemoteSend, port), const_cast("port"), NULL}, {const_cast("rank_to"), py_get_generic(RemoteSend, rank_to), py_set_generic(RemoteSend, rank_to), const_cast("rank_to"), NULL}, {const_cast("backend"), py_get_generic(RemoteSend, backend), py_set_generic(RemoteSend, backend), const_cast("backend"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(RemoteSend)::tp_methods[] = { {const_cast("__getstate__"), PyOp(RemoteSend)::getstate, METH_NOARGS, "RemoteSend getstate"}, {const_cast("__setstate__"), PyOp(RemoteSend)::setstate, METH_VARARGS, "RemoteSend setstate"}, {NULL} /* Sentinel */ }; void _init_py_RemoteSend(py::module m) { using py_op = PyOp(RemoteSend); auto& py_type = PyOpType(RemoteSend); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.RemoteSend"; py_type.tp_basicsize = sizeof(PyOp(RemoteSend)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "RemoteSend"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("RemoteSend", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(RemoteSend::typeinfo(), &py_type).second); } PyOpDefBegin(RemoveAxis) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(RemoveAxis) int PyOp(RemoveAxis)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "scope", NULL}; PyObject *axis = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &axis, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(RemoveAxis)::py_getsetters[] = { {const_cast("axis"), py_get_generic(RemoveAxis, axis), py_set_generic(RemoveAxis, axis), const_cast("axis"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(RemoveAxis)::tp_methods[] = { {const_cast("__getstate__"), PyOp(RemoveAxis)::getstate, METH_NOARGS, "RemoveAxis getstate"}, {const_cast("__setstate__"), PyOp(RemoveAxis)::setstate, METH_VARARGS, "RemoveAxis setstate"}, {NULL} /* Sentinel */ }; void _init_py_RemoveAxis(py::module m) { using py_op = PyOp(RemoveAxis); auto& py_type = PyOpType(RemoveAxis); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.RemoveAxis"; py_type.tp_basicsize = sizeof(PyOp(RemoveAxis)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "RemoveAxis"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("RemoveAxis", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(RemoveAxis::typeinfo(), &py_type).second); } PyOpDefBegin(Reshape) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)}, {"shape", serialization::dump(opdef.shape)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("shape"); if (iter != state.end()) { opdef.shape = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Reshape) int PyOp(Reshape)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "shape", "scope", NULL}; PyObject *axis = NULL, *shape = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &axis, &shape, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (shape) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().shape = py::cast(py::handle(shape)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Reshape)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Reshape, axis), py_set_generic(Reshape, axis), const_cast("axis"), NULL}, {const_cast("shape"), py_get_generic(Reshape, shape), py_set_generic(Reshape, shape), const_cast("shape"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Reshape)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Reshape)::getstate, METH_NOARGS, "Reshape getstate"}, {const_cast("__setstate__"), PyOp(Reshape)::setstate, METH_VARARGS, "Reshape setstate"}, {NULL} /* Sentinel */ }; void _init_py_Reshape(py::module m) { using py_op = PyOp(Reshape); auto& py_type = PyOpType(Reshape); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Reshape"; py_type.tp_basicsize = sizeof(PyOp(Reshape)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Reshape"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Reshape", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Reshape::typeinfo(), &py_type).second); } void _init_py_Resize_InterpolationMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "InterpolationMode", reinterpret_cast(e_type)) >= 0); } void _init_py_Resize_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(Resize) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"imode", serialization::dump(opdef.imode)}, {"format", serialization::dump(opdef.format)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("imode"); if (iter != state.end()) { opdef.imode = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Resize) int PyOp(Resize)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"imode", "format", "scope", NULL}; PyObject *imode = NULL, *format = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &imode, &format, &scope)) return -1; if (imode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().imode = py::cast(py::handle(imode)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Resize)::py_getsetters[] = { {const_cast("imode"), py_get_generic(Resize, imode), py_set_generic(Resize, imode), const_cast("imode"), NULL}, {const_cast("format"), py_get_generic(Resize, format), py_set_generic(Resize, format), const_cast("format"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Resize)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Resize)::getstate, METH_NOARGS, "Resize getstate"}, {const_cast("__setstate__"), PyOp(Resize)::setstate, METH_VARARGS, "Resize setstate"}, {NULL} /* Sentinel */ }; void _init_py_Resize(py::module m) { using py_op = PyOp(Resize); auto& py_type = PyOpType(Resize); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Resize"; py_type.tp_basicsize = sizeof(PyOp(Resize)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Resize"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_Resize_InterpolationMode(py_type); _init_py_Resize_Format(py_type); PyType_Modified(&py_type); m.add_object("Resize", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Resize::typeinfo(), &py_type).second); } PyOpDefBegin(SVD) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"full_matrices", serialization::dump(opdef.full_matrices)}, {"compute_uv", serialization::dump(opdef.compute_uv)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("full_matrices"); if (iter != state.end()) { opdef.full_matrices = serialization::load(iter->second); } } { auto&& iter = state.find("compute_uv"); if (iter != state.end()) { opdef.compute_uv = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(SVD) int PyOp(SVD)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"full_matrices", "compute_uv", "scope", NULL}; PyObject *full_matrices = NULL, *compute_uv = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &full_matrices, &compute_uv, &scope)) return -1; if (full_matrices) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().full_matrices = py::cast(py::handle(full_matrices)); } CATCH_ALL(-1) } if (compute_uv) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().compute_uv = py::cast(py::handle(compute_uv)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(SVD)::py_getsetters[] = { {const_cast("full_matrices"), py_get_generic(SVD, full_matrices), py_set_generic(SVD, full_matrices), const_cast("full_matrices"), NULL}, {const_cast("compute_uv"), py_get_generic(SVD, compute_uv), py_set_generic(SVD, compute_uv), const_cast("compute_uv"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(SVD)::tp_methods[] = { {const_cast("__getstate__"), PyOp(SVD)::getstate, METH_NOARGS, "SVD getstate"}, {const_cast("__setstate__"), PyOp(SVD)::setstate, METH_VARARGS, "SVD setstate"}, {NULL} /* Sentinel */ }; void _init_py_SVD(py::module m) { using py_op = PyOp(SVD); auto& py_type = PyOpType(SVD); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.SVD"; py_type.tp_basicsize = sizeof(PyOp(SVD)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "SVD"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("SVD", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(SVD::typeinfo(), &py_type).second); } PyOpDefBegin(SetMeshIndexing) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(SetMeshIndexing) int PyOp(SetMeshIndexing)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(SetMeshIndexing)::py_getsetters[] = { {const_cast("items"), py_get_generic(SetMeshIndexing, items), py_set_generic(SetMeshIndexing, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(SetMeshIndexing)::tp_methods[] = { {const_cast("__getstate__"), PyOp(SetMeshIndexing)::getstate, METH_NOARGS, "SetMeshIndexing getstate"}, {const_cast("__setstate__"), PyOp(SetMeshIndexing)::setstate, METH_VARARGS, "SetMeshIndexing setstate"}, {NULL} /* Sentinel */ }; void _init_py_SetMeshIndexing(py::module m) { using py_op = PyOp(SetMeshIndexing); auto& py_type = PyOpType(SetMeshIndexing); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.SetMeshIndexing"; py_type.tp_basicsize = sizeof(PyOp(SetMeshIndexing)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "SetMeshIndexing"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("SetMeshIndexing", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(SetMeshIndexing::typeinfo(), &py_type).second); } PyOpDefBegin(SetSubtensor) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(SetSubtensor) int PyOp(SetSubtensor)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(SetSubtensor)::py_getsetters[] = { {const_cast("items"), py_get_generic(SetSubtensor, items), py_set_generic(SetSubtensor, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(SetSubtensor)::tp_methods[] = { {const_cast("__getstate__"), PyOp(SetSubtensor)::getstate, METH_NOARGS, "SetSubtensor getstate"}, {const_cast("__setstate__"), PyOp(SetSubtensor)::setstate, METH_VARARGS, "SetSubtensor setstate"}, {NULL} /* Sentinel */ }; void _init_py_SetSubtensor(py::module m) { using py_op = PyOp(SetSubtensor); auto& py_type = PyOpType(SetSubtensor); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.SetSubtensor"; py_type.tp_basicsize = sizeof(PyOp(SetSubtensor)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "SetSubtensor"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("SetSubtensor", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(SetSubtensor::typeinfo(), &py_type).second); } PyOpDefBegin(ShuffleRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(ShuffleRNG) int PyOp(ShuffleRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "handle", "scope", NULL}; PyObject *seed = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &seed, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(ShuffleRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(ShuffleRNG, seed), py_set_generic(ShuffleRNG, seed), const_cast("seed"), NULL}, {const_cast("handle"), py_get_generic(ShuffleRNG, handle), py_set_generic(ShuffleRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(ShuffleRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(ShuffleRNG)::getstate, METH_NOARGS, "ShuffleRNG getstate"}, {const_cast("__setstate__"), PyOp(ShuffleRNG)::setstate, METH_VARARGS, "ShuffleRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_ShuffleRNG(py::module m) { using py_op = PyOp(ShuffleRNG); auto& py_type = PyOpType(ShuffleRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.ShuffleRNG"; py_type.tp_basicsize = sizeof(PyOp(ShuffleRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "ShuffleRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("ShuffleRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(ShuffleRNG::typeinfo(), &py_type).second); } PyOpDefBegin(SlidingWindowTranspose) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"out_h", serialization::dump(opdef.out_h)}, {"out_w", serialization::dump(opdef.out_w)}, {"pad_h", serialization::dump(opdef.pad_h)}, {"pad_w", serialization::dump(opdef.pad_w)}, {"stride_h", serialization::dump(opdef.stride_h)}, {"stride_w", serialization::dump(opdef.stride_w)}, {"dilate_h", serialization::dump(opdef.dilate_h)}, {"dilate_w", serialization::dump(opdef.dilate_w)}, {"window_h", serialization::dump(opdef.window_h)}, {"window_w", serialization::dump(opdef.window_w)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("out_h"); if (iter != state.end()) { opdef.out_h = serialization::load(iter->second); } } { auto&& iter = state.find("out_w"); if (iter != state.end()) { opdef.out_w = serialization::load(iter->second); } } { auto&& iter = state.find("pad_h"); if (iter != state.end()) { opdef.pad_h = serialization::load(iter->second); } } { auto&& iter = state.find("pad_w"); if (iter != state.end()) { opdef.pad_w = serialization::load(iter->second); } } { auto&& iter = state.find("stride_h"); if (iter != state.end()) { opdef.stride_h = serialization::load(iter->second); } } { auto&& iter = state.find("stride_w"); if (iter != state.end()) { opdef.stride_w = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_h"); if (iter != state.end()) { opdef.dilate_h = serialization::load(iter->second); } } { auto&& iter = state.find("dilate_w"); if (iter != state.end()) { opdef.dilate_w = serialization::load(iter->second); } } { auto&& iter = state.find("window_h"); if (iter != state.end()) { opdef.window_h = serialization::load(iter->second); } } { auto&& iter = state.find("window_w"); if (iter != state.end()) { opdef.window_w = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(SlidingWindowTranspose) int PyOp(SlidingWindowTranspose)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"out_h", "out_w", "pad_h", "pad_w", "stride_h", "stride_w", "dilate_h", "dilate_w", "window_h", "window_w", "scope", NULL}; PyObject *out_h = NULL, *out_w = NULL, *pad_h = NULL, *pad_w = NULL, *stride_h = NULL, *stride_w = NULL, *dilate_h = NULL, *dilate_w = NULL, *window_h = NULL, *window_w = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOOOOOOOO", const_cast(kwlist), &out_h, &out_w, &pad_h, &pad_w, &stride_h, &stride_w, &dilate_h, &dilate_w, &window_h, &window_w, &scope)) return -1; if (out_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().out_h = py::cast(py::handle(out_h)); } CATCH_ALL(-1) } if (out_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().out_w = py::cast(py::handle(out_w)); } CATCH_ALL(-1) } if (pad_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_h = py::cast(py::handle(pad_h)); } CATCH_ALL(-1) } if (pad_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().pad_w = py::cast(py::handle(pad_w)); } CATCH_ALL(-1) } if (stride_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_h = py::cast(py::handle(stride_h)); } CATCH_ALL(-1) } if (stride_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().stride_w = py::cast(py::handle(stride_w)); } CATCH_ALL(-1) } if (dilate_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_h = py::cast(py::handle(dilate_h)); } CATCH_ALL(-1) } if (dilate_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dilate_w = py::cast(py::handle(dilate_w)); } CATCH_ALL(-1) } if (window_h) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().window_h = py::cast(py::handle(window_h)); } CATCH_ALL(-1) } if (window_w) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().window_w = py::cast(py::handle(window_w)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(SlidingWindowTranspose)::py_getsetters[] = { {const_cast("out_h"), py_get_generic(SlidingWindowTranspose, out_h), py_set_generic(SlidingWindowTranspose, out_h), const_cast("out_h"), NULL}, {const_cast("out_w"), py_get_generic(SlidingWindowTranspose, out_w), py_set_generic(SlidingWindowTranspose, out_w), const_cast("out_w"), NULL}, {const_cast("pad_h"), py_get_generic(SlidingWindowTranspose, pad_h), py_set_generic(SlidingWindowTranspose, pad_h), const_cast("pad_h"), NULL}, {const_cast("pad_w"), py_get_generic(SlidingWindowTranspose, pad_w), py_set_generic(SlidingWindowTranspose, pad_w), const_cast("pad_w"), NULL}, {const_cast("stride_h"), py_get_generic(SlidingWindowTranspose, stride_h), py_set_generic(SlidingWindowTranspose, stride_h), const_cast("stride_h"), NULL}, {const_cast("stride_w"), py_get_generic(SlidingWindowTranspose, stride_w), py_set_generic(SlidingWindowTranspose, stride_w), const_cast("stride_w"), NULL}, {const_cast("dilate_h"), py_get_generic(SlidingWindowTranspose, dilate_h), py_set_generic(SlidingWindowTranspose, dilate_h), const_cast("dilate_h"), NULL}, {const_cast("dilate_w"), py_get_generic(SlidingWindowTranspose, dilate_w), py_set_generic(SlidingWindowTranspose, dilate_w), const_cast("dilate_w"), NULL}, {const_cast("window_h"), py_get_generic(SlidingWindowTranspose, window_h), py_set_generic(SlidingWindowTranspose, window_h), const_cast("window_h"), NULL}, {const_cast("window_w"), py_get_generic(SlidingWindowTranspose, window_w), py_set_generic(SlidingWindowTranspose, window_w), const_cast("window_w"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(SlidingWindowTranspose)::tp_methods[] = { {const_cast("__getstate__"), PyOp(SlidingWindowTranspose)::getstate, METH_NOARGS, "SlidingWindowTranspose getstate"}, {const_cast("__setstate__"), PyOp(SlidingWindowTranspose)::setstate, METH_VARARGS, "SlidingWindowTranspose setstate"}, {NULL} /* Sentinel */ }; void _init_py_SlidingWindowTranspose(py::module m) { using py_op = PyOp(SlidingWindowTranspose); auto& py_type = PyOpType(SlidingWindowTranspose); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.SlidingWindowTranspose"; py_type.tp_basicsize = sizeof(PyOp(SlidingWindowTranspose)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "SlidingWindowTranspose"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("SlidingWindowTranspose", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(SlidingWindowTranspose::typeinfo(), &py_type).second); } PyOpDefBegin(Softmax) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Softmax) int PyOp(Softmax)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "scope", NULL}; PyObject *axis = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &axis, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Softmax)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Softmax, axis), py_set_generic(Softmax, axis), const_cast("axis"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Softmax)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Softmax)::getstate, METH_NOARGS, "Softmax getstate"}, {const_cast("__setstate__"), PyOp(Softmax)::setstate, METH_VARARGS, "Softmax setstate"}, {NULL} /* Sentinel */ }; void _init_py_Softmax(py::module m) { using py_op = PyOp(Softmax); auto& py_type = PyOpType(Softmax); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Softmax"; py_type.tp_basicsize = sizeof(PyOp(Softmax)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Softmax"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Softmax", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Softmax::typeinfo(), &py_type).second); } PyOpDefBegin(Split) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"axis", serialization::dump(opdef.axis)}, {"nsections", serialization::dump(opdef.nsections)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("axis"); if (iter != state.end()) { opdef.axis = serialization::load(iter->second); } } { auto&& iter = state.find("nsections"); if (iter != state.end()) { opdef.nsections = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Split) int PyOp(Split)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"axis", "nsections", "scope", NULL}; PyObject *axis = NULL, *nsections = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &axis, &nsections, &scope)) return -1; if (axis) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().axis = py::cast(py::handle(axis)); } CATCH_ALL(-1) } if (nsections) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().nsections = py::cast(py::handle(nsections)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Split)::py_getsetters[] = { {const_cast("axis"), py_get_generic(Split, axis), py_set_generic(Split, axis), const_cast("axis"), NULL}, {const_cast("nsections"), py_get_generic(Split, nsections), py_set_generic(Split, nsections), const_cast("nsections"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Split)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Split)::getstate, METH_NOARGS, "Split getstate"}, {const_cast("__setstate__"), PyOp(Split)::setstate, METH_VARARGS, "Split setstate"}, {NULL} /* Sentinel */ }; void _init_py_Split(py::module m) { using py_op = PyOp(Split); auto& py_type = PyOpType(Split); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Split"; py_type.tp_basicsize = sizeof(PyOp(Split)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Split"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Split", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Split::typeinfo(), &py_type).second); } PyOpDefBegin(Subtensor) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"items", serialization::dump(opdef.items)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("items"); if (iter != state.end()) { opdef.items = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(Subtensor) int PyOp(Subtensor)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"items", "scope", NULL}; PyObject *items = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &items, &scope)) return -1; if (items) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().items = py::cast(py::handle(items)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(Subtensor)::py_getsetters[] = { {const_cast("items"), py_get_generic(Subtensor, items), py_set_generic(Subtensor, items), const_cast("items"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(Subtensor)::tp_methods[] = { {const_cast("__getstate__"), PyOp(Subtensor)::getstate, METH_NOARGS, "Subtensor getstate"}, {const_cast("__setstate__"), PyOp(Subtensor)::setstate, METH_VARARGS, "Subtensor setstate"}, {NULL} /* Sentinel */ }; void _init_py_Subtensor(py::module m) { using py_op = PyOp(Subtensor); auto& py_type = PyOpType(Subtensor); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.Subtensor"; py_type.tp_basicsize = sizeof(PyOp(Subtensor)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "Subtensor"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("Subtensor", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(Subtensor::typeinfo(), &py_type).second); } PyOpDefBegin(TQT) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"qmin", serialization::dump(opdef.qmin)}, {"qmax", serialization::dump(opdef.qmax)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("qmin"); if (iter != state.end()) { opdef.qmin = serialization::load(iter->second); } } { auto&& iter = state.find("qmax"); if (iter != state.end()) { opdef.qmax = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(TQT) int PyOp(TQT)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"qmin", "qmax", "scope", NULL}; PyObject *qmin = NULL, *qmax = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &qmin, &qmax, &scope)) return -1; if (qmin) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().qmin = py::cast(py::handle(qmin)); } CATCH_ALL(-1) } if (qmax) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().qmax = py::cast(py::handle(qmax)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(TQT)::py_getsetters[] = { {const_cast("qmin"), py_get_generic(TQT, qmin), py_set_generic(TQT, qmin), const_cast("qmin"), NULL}, {const_cast("qmax"), py_get_generic(TQT, qmax), py_set_generic(TQT, qmax), const_cast("qmax"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(TQT)::tp_methods[] = { {const_cast("__getstate__"), PyOp(TQT)::getstate, METH_NOARGS, "TQT getstate"}, {const_cast("__setstate__"), PyOp(TQT)::setstate, METH_VARARGS, "TQT setstate"}, {NULL} /* Sentinel */ }; void _init_py_TQT(py::module m) { using py_op = PyOp(TQT); auto& py_type = PyOpType(TQT); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.TQT"; py_type.tp_basicsize = sizeof(PyOp(TQT)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "TQT"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("TQT", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(TQT::typeinfo(), &py_type).second); } PyOpDefBegin(TensorRTRuntime) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"buf", serialization::dump(opdef.buf)}, {"buf_size", serialization::dump(opdef.buf_size)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("buf"); if (iter != state.end()) { opdef.buf = serialization::load(iter->second); } } { auto&& iter = state.find("buf_size"); if (iter != state.end()) { opdef.buf_size = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(TensorRTRuntime) int PyOp(TensorRTRuntime)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"buf", "buf_size", "scope", NULL}; PyObject *buf = NULL, *buf_size = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOO", const_cast(kwlist), &buf, &buf_size, &scope)) return -1; if (buf) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf = py::cast(py::handle(buf)); } CATCH_ALL(-1) } if (buf_size) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().buf_size = py::cast(py::handle(buf_size)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(TensorRTRuntime)::py_getsetters[] = { {const_cast("buf"), py_get_generic(TensorRTRuntime, buf), py_set_generic(TensorRTRuntime, buf), const_cast("buf"), NULL}, {const_cast("buf_size"), py_get_generic(TensorRTRuntime, buf_size), py_set_generic(TensorRTRuntime, buf_size), const_cast("buf_size"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(TensorRTRuntime)::tp_methods[] = { {const_cast("__getstate__"), PyOp(TensorRTRuntime)::getstate, METH_NOARGS, "TensorRTRuntime getstate"}, {const_cast("__setstate__"), PyOp(TensorRTRuntime)::setstate, METH_VARARGS, "TensorRTRuntime setstate"}, {NULL} /* Sentinel */ }; void _init_py_TensorRTRuntime(py::module m) { using py_op = PyOp(TensorRTRuntime); auto& py_type = PyOpType(TensorRTRuntime); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.TensorRTRuntime"; py_type.tp_basicsize = sizeof(PyOp(TensorRTRuntime)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "TensorRTRuntime"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("TensorRTRuntime", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(TensorRTRuntime::typeinfo(), &py_type).second); } template<> struct EnumTrait { static constexpr const char *name = "TopK.Mode"; static constexpr std::underlying_type_t max = 3 - 1; }; template<> PyTypeObject* EnumWrapper::type = nullptr; template<> const char* EnumWrapper::members[] = {"KTH_ONLY", "VALUE_IDX_NOSORT", "VALUE_IDX_SORTED"}; template<> std::unordered_map EnumWrapper::mem2value = {{normalize_enum("KTH_ONLY"), TopK::Mode::KTH_ONLY}, {normalize_enum("VALUE_IDX_NOSORT"), TopK::Mode::VALUE_IDX_NOSORT}, {normalize_enum("VALUE_IDX_SORTED"), TopK::Mode::VALUE_IDX_SORTED}}; template<> PyObject* EnumWrapper::pyobj_insts[3] = {nullptr}; void _init_py_TopK_Mode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; static PyMethodDef tp_methods[] = { {const_cast("dump"), (PyCFunction)EnumWrapper::py_dump, METH_NOARGS, NULL}, {NULL} /* Sentinel */ }; static PyType_Slot slots[] = { {Py_tp_repr, (void*)EnumWrapper::py_repr}, {Py_tp_richcompare, (void*)EnumWrapper::tp_richcompare}, {Py_tp_methods, tp_methods}, {0, NULL} }; static PyType_Spec spec = { // name "megengine.core._imperative_rt.ops.TopK.Mode", // basicsize sizeof(EnumWrapper), // itemsize 0, // flags Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE, // slots slots }; e_type = reinterpret_cast(PyType_FromSpec(&spec)); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__name__").release().ptr(), py::cast("Mode").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__module__").release().ptr(), py::cast("megengine.core._imperative_rt.ops").release().ptr()) >= 0); mgb_assert( e_type->tp_setattro( reinterpret_cast(e_type), py::cast("__qualname__").release().ptr(), py::cast("TopK.Mode").release().ptr()) >= 0); { PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = TopK::Mode::KTH_ONLY; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "KTH_ONLY", inst) >= 0); EnumWrapper::pyobj_insts[0] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = TopK::Mode::VALUE_IDX_NOSORT; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "VALUE_IDX_NOSORT", inst) >= 0); EnumWrapper::pyobj_insts[1] = inst; }{ PyObject* inst = e_type->tp_alloc(e_type, 0); reinterpret_cast*>(inst)->value = TopK::Mode::VALUE_IDX_SORTED; mgb_assert(PyDict_SetItemString(e_type->tp_dict, "VALUE_IDX_SORTED", inst) >= 0); EnumWrapper::pyobj_insts[2] = inst; } Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Mode", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(TopK) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"mode", serialization::dump(opdef.mode)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("mode"); if (iter != state.end()) { opdef.mode = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(TopK) int PyOp(TopK)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"mode", "scope", NULL}; PyObject *mode = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &mode, &scope)) return -1; if (mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().mode = py::cast(py::handle(mode)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(TopK)::py_getsetters[] = { {const_cast("mode"), py_get_generic(TopK, mode), py_set_generic(TopK, mode), const_cast("mode"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(TopK)::tp_methods[] = { {const_cast("__getstate__"), PyOp(TopK)::getstate, METH_NOARGS, "TopK getstate"}, {const_cast("__setstate__"), PyOp(TopK)::setstate, METH_VARARGS, "TopK setstate"}, {NULL} /* Sentinel */ }; void _init_py_TopK(py::module m) { using py_op = PyOp(TopK); auto& py_type = PyOpType(TopK); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.TopK"; py_type.tp_basicsize = sizeof(PyOp(TopK)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "TopK"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_TopK_Mode(py_type); PyType_Modified(&py_type); m.add_object("TopK", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(TopK::typeinfo(), &py_type).second); } PyOpDefBegin(TypeCvt) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"dtype", serialization::dump(opdef.dtype)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(TypeCvt) int PyOp(TypeCvt)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"dtype", "scope", NULL}; PyObject *dtype = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO", const_cast(kwlist), &dtype, &scope)) return -1; if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(TypeCvt)::py_getsetters[] = { {const_cast("dtype"), py_get_generic(TypeCvt, dtype), py_set_generic(TypeCvt, dtype), const_cast("dtype"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(TypeCvt)::tp_methods[] = { {const_cast("__getstate__"), PyOp(TypeCvt)::getstate, METH_NOARGS, "TypeCvt getstate"}, {const_cast("__setstate__"), PyOp(TypeCvt)::setstate, METH_VARARGS, "TypeCvt setstate"}, {NULL} /* Sentinel */ }; void _init_py_TypeCvt(py::module m) { using py_op = PyOp(TypeCvt); auto& py_type = PyOpType(TypeCvt); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.TypeCvt"; py_type.tp_basicsize = sizeof(PyOp(TypeCvt)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "TypeCvt"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("TypeCvt", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(TypeCvt::typeinfo(), &py_type).second); } PyOpDefBegin(UniformRNG) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"seed", serialization::dump(opdef.seed)}, {"dtype", serialization::dump(opdef.dtype)}, {"handle", serialization::dump(opdef.handle)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("seed"); if (iter != state.end()) { opdef.seed = serialization::load(iter->second); } } { auto&& iter = state.find("dtype"); if (iter != state.end()) { opdef.dtype = serialization::load(iter->second); } } { auto&& iter = state.find("handle"); if (iter != state.end()) { opdef.handle = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(UniformRNG) int PyOp(UniformRNG)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"seed", "dtype", "handle", "scope", NULL}; PyObject *seed = NULL, *dtype = NULL, *handle = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOO", const_cast(kwlist), &seed, &dtype, &handle, &scope)) return -1; if (seed) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().seed = py::cast(py::handle(seed)); } CATCH_ALL(-1) } if (dtype) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().dtype = py::cast(py::handle(dtype)); } CATCH_ALL(-1) } if (handle) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().handle = py::cast(py::handle(handle)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(UniformRNG)::py_getsetters[] = { {const_cast("seed"), py_get_generic(UniformRNG, seed), py_set_generic(UniformRNG, seed), const_cast("seed"), NULL}, {const_cast("dtype"), py_get_generic(UniformRNG, dtype), py_set_generic(UniformRNG, dtype), const_cast("dtype"), NULL}, {const_cast("handle"), py_get_generic(UniformRNG, handle), py_set_generic(UniformRNG, handle), const_cast("handle"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(UniformRNG)::tp_methods[] = { {const_cast("__getstate__"), PyOp(UniformRNG)::getstate, METH_NOARGS, "UniformRNG getstate"}, {const_cast("__setstate__"), PyOp(UniformRNG)::setstate, METH_VARARGS, "UniformRNG setstate"}, {NULL} /* Sentinel */ }; void _init_py_UniformRNG(py::module m) { using py_op = PyOp(UniformRNG); auto& py_type = PyOpType(UniformRNG); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.UniformRNG"; py_type.tp_basicsize = sizeof(PyOp(UniformRNG)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "UniformRNG"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); PyType_Modified(&py_type); m.add_object("UniformRNG", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(UniformRNG::typeinfo(), &py_type).second); } void _init_py_WarpAffine_InterpolationMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "InterpolationMode", reinterpret_cast(e_type)) >= 0); } void _init_py_WarpAffine_BorderMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "BorderMode", reinterpret_cast(e_type)) >= 0); } void _init_py_WarpAffine_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(WarpAffine) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"imode", serialization::dump(opdef.imode)}, {"border_mode", serialization::dump(opdef.border_mode)}, {"border_val", serialization::dump(opdef.border_val)}, {"format", serialization::dump(opdef.format)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("imode"); if (iter != state.end()) { opdef.imode = serialization::load(iter->second); } } { auto&& iter = state.find("border_mode"); if (iter != state.end()) { opdef.border_mode = serialization::load(iter->second); } } { auto&& iter = state.find("border_val"); if (iter != state.end()) { opdef.border_val = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(WarpAffine) int PyOp(WarpAffine)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"imode", "border_mode", "border_val", "format", "scope", NULL}; PyObject *imode = NULL, *border_mode = NULL, *border_val = NULL, *format = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &imode, &border_mode, &border_val, &format, &scope)) return -1; if (imode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().imode = py::cast(py::handle(imode)); } CATCH_ALL(-1) } if (border_mode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().border_mode = py::cast(py::handle(border_mode)); } CATCH_ALL(-1) } if (border_val) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().border_val = py::cast(py::handle(border_val)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(WarpAffine)::py_getsetters[] = { {const_cast("imode"), py_get_generic(WarpAffine, imode), py_set_generic(WarpAffine, imode), const_cast("imode"), NULL}, {const_cast("border_mode"), py_get_generic(WarpAffine, border_mode), py_set_generic(WarpAffine, border_mode), const_cast("border_mode"), NULL}, {const_cast("border_val"), py_get_generic(WarpAffine, border_val), py_set_generic(WarpAffine, border_val), const_cast("border_val"), NULL}, {const_cast("format"), py_get_generic(WarpAffine, format), py_set_generic(WarpAffine, format), const_cast("format"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(WarpAffine)::tp_methods[] = { {const_cast("__getstate__"), PyOp(WarpAffine)::getstate, METH_NOARGS, "WarpAffine getstate"}, {const_cast("__setstate__"), PyOp(WarpAffine)::setstate, METH_VARARGS, "WarpAffine setstate"}, {NULL} /* Sentinel */ }; void _init_py_WarpAffine(py::module m) { using py_op = PyOp(WarpAffine); auto& py_type = PyOpType(WarpAffine); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.WarpAffine"; py_type.tp_basicsize = sizeof(PyOp(WarpAffine)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "WarpAffine"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_WarpAffine_InterpolationMode(py_type); _init_py_WarpAffine_BorderMode(py_type); _init_py_WarpAffine_Format(py_type); PyType_Modified(&py_type); m.add_object("WarpAffine", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(WarpAffine::typeinfo(), &py_type).second); } void _init_py_WarpPerspective_InterpolationMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "InterpolationMode", reinterpret_cast(e_type)) >= 0); } void _init_py_WarpPerspective_BorderMode(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "BorderMode", reinterpret_cast(e_type)) >= 0); } void _init_py_WarpPerspective_Format(PyTypeObject& py_type) { auto& e_type = EnumWrapper::type; Py_INCREF(e_type); mgb_assert(PyDict_SetItemString( py_type.tp_dict, "Format", reinterpret_cast(e_type)) >= 0); } PyOpDefBegin(WarpPerspective) // { static PyGetSetDef py_getsetters[]; static PyMethodDef tp_methods[]; static PyObject* getstate(PyObject* self, PyObject*) { auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); std::unordered_map state { {"imode", serialization::dump(opdef.imode)}, {"bmode", serialization::dump(opdef.bmode)}, {"format", serialization::dump(opdef.format)}, {"border_val", serialization::dump(opdef.border_val)} }; return py::cast(state).release().ptr(); } static PyObject* setstate(PyObject* self, PyObject* args) { PyObject* dict = PyTuple_GetItem(args, 0); if (!dict) return NULL; auto state = py::cast>(dict); auto& opdef = reinterpret_cast(self)->inst(); static_cast(opdef); { auto&& iter = state.find("imode"); if (iter != state.end()) { opdef.imode = serialization::load(iter->second); } } { auto&& iter = state.find("bmode"); if (iter != state.end()) { opdef.bmode = serialization::load(iter->second); } } { auto&& iter = state.find("format"); if (iter != state.end()) { opdef.format = serialization::load(iter->second); } } { auto&& iter = state.find("border_val"); if (iter != state.end()) { opdef.border_val = serialization::load(iter->second); } } Py_RETURN_NONE; } static int py_init(PyObject *self, PyObject *args, PyObject *kwds); // }; PyOpDefEnd(WarpPerspective) int PyOp(WarpPerspective)::py_init(PyObject *self, PyObject *args, PyObject *kwds) { static const char* kwlist[] = {"imode", "bmode", "format", "border_val", "scope", NULL}; PyObject *imode = NULL, *bmode = NULL, *format = NULL, *border_val = NULL, *scope = NULL; if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOOOO", const_cast(kwlist), &imode, &bmode, &format, &border_val, &scope)) return -1; if (imode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().imode = py::cast(py::handle(imode)); } CATCH_ALL(-1) } if (bmode) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().bmode = py::cast(py::handle(bmode)); } CATCH_ALL(-1) } if (format) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().format = py::cast(py::handle(format)); } CATCH_ALL(-1) } if (border_val) { try { // TODO: remove this guard which is used for pybind11 implicit conversion py::detail::loader_life_support guard{}; reinterpret_cast(self)->inst().border_val = py::cast(py::handle(border_val)); } CATCH_ALL(-1) } if (scope) { try { reinterpret_cast(self)->op ->set_scope(py::cast(py::handle(scope))); } CATCH_ALL(-1) } return 0; } PyGetSetDef PyOp(WarpPerspective)::py_getsetters[] = { {const_cast("imode"), py_get_generic(WarpPerspective, imode), py_set_generic(WarpPerspective, imode), const_cast("imode"), NULL}, {const_cast("bmode"), py_get_generic(WarpPerspective, bmode), py_set_generic(WarpPerspective, bmode), const_cast("bmode"), NULL}, {const_cast("format"), py_get_generic(WarpPerspective, format), py_set_generic(WarpPerspective, format), const_cast("format"), NULL}, {const_cast("border_val"), py_get_generic(WarpPerspective, border_val), py_set_generic(WarpPerspective, border_val), const_cast("border_val"), NULL}, {NULL} /* Sentinel */ }; PyMethodDef PyOp(WarpPerspective)::tp_methods[] = { {const_cast("__getstate__"), PyOp(WarpPerspective)::getstate, METH_NOARGS, "WarpPerspective getstate"}, {const_cast("__setstate__"), PyOp(WarpPerspective)::setstate, METH_VARARGS, "WarpPerspective setstate"}, {NULL} /* Sentinel */ }; void _init_py_WarpPerspective(py::module m) { using py_op = PyOp(WarpPerspective); auto& py_type = PyOpType(WarpPerspective); py_type = {PyVarObject_HEAD_INIT(NULL, 0)}; py_type.tp_name = "megengine.core._imperative_rt.ops.WarpPerspective"; py_type.tp_basicsize = sizeof(PyOp(WarpPerspective)); py_type.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE; py_type.tp_doc = "WarpPerspective"; py_type.tp_base = &PyOpType(OpDef); py_type.tp_dealloc = py_dealloc_generic; py_type.tp_new = py_new_generic; py_type.tp_init = py_op::py_init; py_type.tp_methods = py_op::tp_methods; py_type.tp_getset = py_op::py_getsetters; mgb_assert(PyType_Ready(&py_type) >= 0); _init_py_WarpPerspective_InterpolationMode(py_type); _init_py_WarpPerspective_BorderMode(py_type); _init_py_WarpPerspective_Format(py_type); PyType_Modified(&py_type); m.add_object("WarpPerspective", reinterpret_cast(&py_type)); mgb_assert(PyOp(OpDef)::ctype2pytype.emplace(WarpPerspective::typeinfo(), &py_type).second); } #define INIT_ALL_OP(m) \ _init_py_AdaptivePooling(m); \ _init_py_AddAxis(m); \ _init_py_Argmax(m); \ _init_py_Argmin(m); \ _init_py_Argsort(m); \ _init_py_AssertEqual(m); \ _init_py_AtlasRuntime(m); \ _init_py_Barrier(m); \ _init_py_BatchConvBias(m); \ _init_py_BatchNorm(m); \ _init_py_BatchNormBackward(m); \ _init_py_BatchedIncrMeshIndexing(m); \ _init_py_BatchedMatrixMul(m); \ _init_py_BatchedMeshIndexing(m); \ _init_py_BatchedSetMeshIndexing(m); \ _init_py_BetaRNG(m); \ _init_py_Borrow(m); \ _init_py_Broadcast(m); \ _init_py_CambriconRuntime(m); \ _init_py_CheckNonFinite(m); \ _init_py_CollectiveComm(m); \ _init_py_Concat(m); \ _init_py_CondTake(m); \ _init_py_ConvBias(m); \ _init_py_Convolution(m); \ _init_py_Convolution3D(m); \ _init_py_Convolution3DBackwardData(m); \ _init_py_ConvolutionBackwardData(m); \ _init_py_Copy(m); \ _init_py_Correlation(m); \ _init_py_Cumsum(m); \ _init_py_CvtColor(m); \ _init_py_DeformableConv(m); \ _init_py_DeformablePSROIPooling(m); \ _init_py_Diag(m); \ _init_py_Dimshuffle(m); \ _init_py_Dot(m); \ _init_py_Dropout(m); \ _init_py_Elemwise(m); \ _init_py_ElemwiseMultiType(m); \ _init_py_ExternOpr(m); \ _init_py_Eye(m); \ _init_py_FakeQuant(m); \ _init_py_FastpathCopy(m); \ _init_py_GammaRNG(m); \ _init_py_GaussianRNG(m); \ _init_py_GetVarShape(m); \ _init_py_GroupLocal(m); \ _init_py_Identity(m); \ _init_py_Images2Neibs(m); \ _init_py_IncrMeshIndexing(m); \ _init_py_IncrSubtensor(m); \ _init_py_IndexingIncrMultiAxisVec(m); \ _init_py_IndexingMultiAxisVec(m); \ _init_py_IndexingOneHot(m); \ _init_py_IndexingSetMultiAxisVec(m); \ _init_py_IndexingSetOneHot(m); \ _init_py_InplaceAdd(m); \ _init_py_LAMBUpdate(m); \ _init_py_LRN(m); \ _init_py_LSQ(m); \ _init_py_LSTM(m); \ _init_py_LSTMCell(m); \ _init_py_LayerNorm(m); \ _init_py_Linspace(m); \ _init_py_MagicMindRuntime(m); \ _init_py_MatrixInverse(m); \ _init_py_MatrixMul(m); \ _init_py_MeshIndexing(m); \ _init_py_NMSKeep(m); \ _init_py_NvOf(m); \ _init_py_Padding(m); \ _init_py_ParamPackConcat(m); \ _init_py_ParamPackSplit(m); \ _init_py_PermutationRNG(m); \ _init_py_PixelShuffle(m); \ _init_py_PixelShuffleBackward(m); \ _init_py_PoissonRNG(m); \ _init_py_Pooling(m); \ _init_py_RNN(m); \ _init_py_RNNCell(m); \ _init_py_ROIAlign(m); \ _init_py_ROIPooling(m); \ _init_py_Reduce(m); \ _init_py_Remap(m); \ _init_py_RemoteRecv(m); \ _init_py_RemoteSend(m); \ _init_py_RemoveAxis(m); \ _init_py_Reshape(m); \ _init_py_Resize(m); \ _init_py_SVD(m); \ _init_py_SetMeshIndexing(m); \ _init_py_SetSubtensor(m); \ _init_py_ShuffleRNG(m); \ _init_py_SlidingWindowTranspose(m); \ _init_py_Softmax(m); \ _init_py_Split(m); \ _init_py_Subtensor(m); \ _init_py_TQT(m); \ _init_py_TensorRTRuntime(m); \ _init_py_TopK(m); \ _init_py_TypeCvt(m); \ _init_py_UniformRNG(m); \ _init_py_WarpAffine(m); \ _init_py_WarpPerspective(m); // clang-format on