From 46b7c3275eb9a4e2af9c116a2f3078d0bb52f2f9 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 15 Oct 2021 19:03:48 +0800 Subject: [PATCH] feat(lite): add get static mem info function in pylite GitOrigin-RevId: d938eee65a1ec4f40fc83a92a64b6c01b41d9b32 --- lite/pylite/megenginelite/network.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/lite/pylite/megenginelite/network.py b/lite/pylite/megenginelite/network.py index 6808621d..4edcc960 100644 --- a/lite/pylite/megenginelite/network.py +++ b/lite/pylite/megenginelite/network.py @@ -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)