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 3878d95182bbb81fc7023bc9a4ccde8bbeb74db0..1e82f62fde16206f64229df1ada275f3ca630771 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 @@ -103,6 +103,8 @@ class Const: FILE_NAME_LENGTH = 255 DIRECTORY_LENGTH = 4096 DISTRIBUTED_PREFIX_LENGTH = 60 + SUMMARY_COLUMN_NUM = 6 + STACK_COLUMN_NUM = 2 # env dump path ASCEND_WORK_PATH = "ASCEND_WORK_PATH" DUMP_DIR = "dump_data" diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/compare/acc_compare.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/compare/acc_compare.py index bb39809891611ea0b3d17e660f5841849195b1ff..68fe6c0d7ea6442bc01dfea19cbc97295e174016 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/compare/acc_compare.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/compare/acc_compare.py @@ -237,9 +237,15 @@ def merge_tensor(tensor_list): for tensor in tensor_list: if tensor[0].find("stack_info") != -1: + if len(tensor) != Const.STACK_COLUMN_NUM: + print_error_log(f"This stack_info data is not complete. {tensor}") + raise CompareException(CompareException.INVALID_DATA_ERROR) op_dict["stack_info"].append(tensor[1]) break op_dict["op_name"].append(tensor[0]) + if len(tensor) != Const.SUMMARY_COLUMN_NUM: + print_error_log(f"This summary data is not complete. {tensor}") + raise CompareException(CompareException.INVALID_DATA_ERROR) if tensor[0].find("input") != -1: op_dict["input_struct"].append((tensor[3], tensor[4], tensor[2])) elif tensor[0].find("output") != -1: