From cbfb67bd0fe7fa4b0130679a5585659612c329bc Mon Sep 17 00:00:00 2001
From: RanZheng <364167184@qq.com>
Date: Fri, 5 Sep 2025 10:33:51 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=AE=89=E5=85=A8=E3=80=91monitor?=
 =?UTF-8?q?=E5=92=8Cparse=5Ftool=E5=AE=89=E5=85=A8=E9=97=AE=E9=A2=98?=
 =?UTF-8?q?=E6=95=B4=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 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 a807f7e27..0f015dceb 100644
--- a/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py
+++ b/debug/accuracy_tools/msprobe/pytorch/monitor/csv2tb.py
@@ -79,7 +79,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