提交 3840c1f4 编写于 作者: M Megvii Engine Team

fix(mge/imperative): use logical reprentation of compnode

GitOrigin-RevId: 9d104de3aa3f5894b8d2081c18dce7bb4d86628b
上级 227ec04b
...@@ -16,7 +16,7 @@ void init_common(py::module m) { ...@@ -16,7 +16,7 @@ void init_common(py::module m) {
py::class_<CompNode>(m, "CompNode") py::class_<CompNode>(m, "CompNode")
.def(py::init()) .def(py::init())
.def(py::init(py::overload_cast<const std::string&>(&CompNode::load))) .def(py::init(py::overload_cast<const std::string&>(&CompNode::load)))
.def("__str__", &CompNode::to_string) .def("__str__", &CompNode::to_string_logical)
.def_static("_sync_all", &CompNode::sync_all) .def_static("_sync_all", &CompNode::sync_all)
.def(py::self == py::self) .def(py::self == py::self)
.def_static("_get_device_count", &CompNode::get_device_count, .def_static("_get_device_count", &CompNode::get_device_count,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册