From 374d8e0c95428f56c790761a78484fe11c9d151e Mon Sep 17 00:00:00 2001 From: yang-minghai22 Date: Mon, 19 Feb 2024 17:20:50 +0800 Subject: [PATCH] update norm calculate useless result --- .../ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py | 1 + .../ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py | 4 ++-- debug/accuracy_tools/ptdbg_ascend/test/ut/test_dump.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py index 72f3bca37..28e824e51 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py @@ -76,6 +76,7 @@ class Const: BACKWARD = 'backward' FORWARD = 'forward' PRE_FORWARD = "pre_forward" + NA = 'N/A' # dump mode ALL = "all" diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py index 3d4afb9cf..6c49fb682 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py @@ -33,7 +33,7 @@ else: from .utils import (DumpUtil, check_if_in_api_list, make_dump_data_dir, get_tensor_rank, create_dirs_if_not_exist, CompareException) -from ..common.utils import (print_warn_log, Const, print_info_log, modify_dump_path, check_inplace_op, CompareConst, +from ..common.utils import (print_warn_log, Const, print_info_log, modify_dump_path, check_inplace_op, get_md5_for_tensor, print_error_log) from ..dump.utils import check_writable from ..common.file_check_util import FileOpen, change_mode, FileCheckConst, check_path_pattern_vaild, check_path_length @@ -106,7 +106,7 @@ def get_not_float_tensor_info(data): tensor_max = torch._C._VariableFunctionsClass.max(data).cpu().detach().float().numpy().tolist() tensor_min = torch._C._VariableFunctionsClass.min(data).cpu().detach().float().numpy().tolist() tensor_mean = torch._C._VariableFunctionsClass.mean(data.float()).cpu().detach().float().numpy().tolist() - return get_tensor_data_info(data, tensor_max, tensor_min, tensor_mean, CompareConst.NAN) + return get_tensor_data_info(data, tensor_max, tensor_min, tensor_mean, Const.NA) def get_scalar_data_info(data): diff --git a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_dump.py b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_dump.py index 9673c292b..bc5786693 100644 --- a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_dump.py +++ b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_dump.py @@ -15,7 +15,7 @@ class TestDump(unittest.TestCase): def test_get_not_float_tensor_info(self): data_info = get_not_float_tensor_info(self.tensor) self.assertEqual(data_info.save_data.tolist(), self.tensor.numpy().tolist()) - self.assertEqual(data_info.summary_data, [3.0, 1.0, 2.0, 'Nan']) + self.assertEqual(data_info.summary_data, [3.0, 1.0, 2.0, 'N/A']) self.assertEqual(data_info.dtype, 'torch.float32') self.assertEqual(data_info.shape, (3,)) -- Gitee