From a1a75097f51844771e4a48a79292a6e0add2d3f1 Mon Sep 17 00:00:00 2001 From: lijiaojiao Date: Mon, 30 Sep 2024 09:39:48 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=AE=89=E5=85=A8=E3=80=91=E6=95=B0?= =?UTF-8?q?=E7=BB=84=E8=B6=8A=E7=95=8C=E9=97=AE=E9=A2=98=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../msprobe/pytorch/online_dispatch/dispatch.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/debug/accuracy_tools/msprobe/pytorch/online_dispatch/dispatch.py b/debug/accuracy_tools/msprobe/pytorch/online_dispatch/dispatch.py index 441f2b99289..483dc2143c0 100644 --- a/debug/accuracy_tools/msprobe/pytorch/online_dispatch/dispatch.py +++ b/debug/accuracy_tools/msprobe/pytorch/online_dispatch/dispatch.py @@ -102,7 +102,10 @@ class PtdbgDispatch(TorchDispatchMode): if len(json_line_data) == 0: break msg = json.loads(json_line_data) - self.all_summary[msg[0]] = msg[1] + try: + self.all_summary[msg[0]] = msg[1] + except IndexError as e: + logger.error(f"Please check summary json, An exception may have occurred! {e}") fp_handle.close() if self.debug_flag: -- Gitee