Browse Source

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

GitOrigin-RevId: 9d104de3aa
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
3840c1f433
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/src/common.cpp

+ 1
- 1
imperative/python/src/common.cpp View File

@@ -16,7 +16,7 @@ void init_common(py::module m) {
py::class_<CompNode>(m, "CompNode")
.def(py::init())
.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(py::self == py::self)
.def_static("_get_device_count", &CompNode::get_device_count,


Loading…
Cancel
Save