From 6acc56eb8aa5c9ef7f8a8da87fb30f900ed408e1 Mon Sep 17 00:00:00 2001 From: RanZheng <364167184@qq.com> Date: Thu, 14 Aug 2025 15:14:25 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=89=E5=85=A8=E9=97=AE=E9=A2=98=E6=95=B4?= =?UTF-8?q?=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit (cherry picked from commit 3341ba6408fc6540acce03bd86379eafffcf9226) --- debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py | 2 +- .../msprobe/pytorch/parse_tool/lib/visualization.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py b/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py index 48f3f27cb..a67a258cb 100644 --- a/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py +++ b/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py @@ -80,7 +80,7 @@ def write_step(output_dirpath, parse_step_result, rank, data_type): for op, value in ops.items(): tag = f"{vpp_name}/{op}" writer.add_scalar(tag, value, step) - writer.flush() + writer.close() @recursion_depth_decorator("update_dict", max_depth=50) diff --git a/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py b/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py index 5b53831b1..5e54a45f9 100644 --- a/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py +++ b/debug/accuracy_tools/msprobe/pytorch/parse_tool/lib/visualization.py @@ -83,4 +83,3 @@ class Visualization: self.util.log.info("\nStatistic Info:") title_printed = True self.util.log.info(summery_info) - pkl_handle.close() -- Gitee