提交 46b7c327 编写于 作者: M Megvii Engine Team

feat(lite): add get static mem info function in pylite

GitOrigin-RevId: d938eee65a1ec4f40fc83a92a64b6c01b41d9b32
上级 5e8a9a95
......@@ -290,6 +290,7 @@ class _NetworkAPI(_LiteCObjBase):
("LITE_set_async_callback", [_Cnetwork, LiteAsyncCallback]),
("LITE_set_start_callback", [_Cnetwork]),
("LITE_set_finish_callback", [_Cnetwork]),
("LITE_get_static_memory_alloc_info", [_Cnetwork, c_char_p]),
]
......@@ -532,3 +533,7 @@ class LiteNetwork(object):
def io_bin_dump(self, bin_dir):
c_dir = bin_dir.encode("utf-8")
self._api.LITE_enable_io_bin_dump(self._network, c_dir)
def get_static_memory_alloc_info(self, log_dir="logs/test"):
c_log_dir = log_dir.encode("utf-8")
self._api.LITE_get_static_memory_alloc_info(self._network, c_log_dir)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册