From 6cfc96cd214dde961f0b8336390df1d0bd62ff33 Mon Sep 17 00:00:00 2001 From: fanglanyue Date: Fri, 15 Aug 2025 16:33:01 +0800 Subject: [PATCH] fix safety for compare --- .../compare_backend/compare_bean/kernel_compare_bean.py | 2 +- .../compare_tools/compare_backend/view/work_sheet_creator.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/profiler/msprof_analyze/compare_tools/compare_backend/compare_bean/kernel_compare_bean.py b/profiler/msprof_analyze/compare_tools/compare_backend/compare_bean/kernel_compare_bean.py index 3f64f28ec..b652cdc20 100644 --- a/profiler/msprof_analyze/compare_tools/compare_backend/compare_bean/kernel_compare_bean.py +++ b/profiler/msprof_analyze/compare_tools/compare_backend/compare_bean/kernel_compare_bean.py @@ -27,7 +27,7 @@ class KernelCompareInfo: self._number = None self._max_dur = None self._min_dur = None - if not data_list: + if len(data_list) < 6: return self._kernel_type = data_list[0] self._input_shapes = data_list[1] diff --git a/profiler/msprof_analyze/compare_tools/compare_backend/view/work_sheet_creator.py b/profiler/msprof_analyze/compare_tools/compare_backend/view/work_sheet_creator.py index b73f6df97..d10e35f49 100644 --- a/profiler/msprof_analyze/compare_tools/compare_backend/view/work_sheet_creator.py +++ b/profiler/msprof_analyze/compare_tools/compare_backend/view/work_sheet_creator.py @@ -44,7 +44,7 @@ class WorkSheetCreator: com_header_format = self._work_book.add_format(CellFormatType.YELLOW_BOLD) com_index_range = [-1, -1] overhead = self._data.get("overhead", []) - if overhead: + if len(overhead) >= 2: base_path = f"Base Profiling: {self._args.base_profiling_path}" self._work_sheet.merge_range(overhead[0], base_path, base_header_format) com_index_range = [self._col_ids.index(overhead[1].split(":")[0][0]), -- Gitee