From d0fbef5353cc0b19498a895df736d927aa50963d Mon Sep 17 00:00:00 2001 From: lcw Date: Thu, 20 Mar 2025 17:30:38 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90bugfix=E3=80=91mindspore=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3=E5=8F=98=E6=9B=B4=E5=AF=BC=E8=87=B4=E5=BC=82=E6=AD=A5?= =?UTF-8?q?dump=20bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- debug/accuracy_tools/msprobe/core/data_dump/json_writer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py b/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py index 2d83647917..738e36afc8 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py @@ -157,8 +157,6 @@ class DataWriter: logger.warning("Some bad data in async dump") else: tensor_stat_index, tensor_stat_data = tensor_stat[0], tensor_stat[1] - if hasattr(tensor_stat_data, "device") and tensor_stat_data.device != Const.CPU_LOWERCASE: - tensor_stat_data = tensor_stat_data.cpu() for index, stat in zip(tensor_stat_index, tensor_stat_data): data.update({index: stat.item()}) del data["tensor_stat"] -- Gitee