From b65109be2a1be58e45851a7d52f13a78f66ba0b1 Mon Sep 17 00:00:00 2001 From: l30036321 Date: Thu, 14 Mar 2024 19:37:18 +0800 Subject: [PATCH] Add array length validation for pkl content --- .../ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py | 2 ++ .../src/python/ptdbg_ascend/compare/acc_compare.py | 6 ++++++ 2 files changed, 8 insertions(+) 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 3878d95182..1e82f62fde 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 bb39809891..68fe6c0d7e 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: -- Gitee