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 72f3bca37453cbdec02cc858db5c4427a6a27d72..28e824e517df874b208a7da553c0cabb69b111dc 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 3d4afb9cfc76805c706dbda57dae6ed9af8114e2..6c49fb6825113d939f65bf1a52e7045b1da7a5e8 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 9673c292ba20cef94b926986ddac270f748645e9..bc57866930a81ecac7ff12044817714680b972ab 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,))