diff --git a/debug/accuracy_tools/msprobe/pytorch/online_dispatch/dispatch.py b/debug/accuracy_tools/msprobe/pytorch/online_dispatch/dispatch.py index 441f2b99289b3311af40601fd75ca862b271c96a..483dc2143c01f8d7c6ce12e899fe2a7541f7a76d 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: