From d5773de9dff293b2da863ee3f276611292190b60 Mon Sep 17 00:00:00 2001 From: fanxiaotong Date: Sat, 5 Aug 2023 17:02:07 +0800 Subject: [PATCH 1/2] fix bug --- .../cluster_kernels_analysis/cluster_op_summary_analysis.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py b/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py index 04dbfdfe9..70409edd3 100644 --- a/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py +++ b/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py @@ -269,3 +269,5 @@ def main(): if __name__ == "__main__": main() + + -- Gitee From 232d4deaae11833b90bfad137edc2a43ef61bbee Mon Sep 17 00:00:00 2001 From: fanxiaotong Date: Sat, 5 Aug 2023 17:18:49 +0800 Subject: [PATCH 2/2] fix bug --- .../cluster_op_summary_analysis.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py b/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py index 70409edd3..38d66a234 100644 --- a/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py +++ b/profiler/cluster_analyse/cluster_kernels_analysis/cluster_op_summary_analysis.py @@ -45,8 +45,11 @@ class FormDataProcessor: # 读取CSV文件 df = pd.read_csv(f) # 保留需要的列 - df = df[columns_to_keep] - + try: + df = df[columns_to_keep] + except KeyError: + print(f"{f}文件没有所需的列，请确认profiling数据的正确性:\n,以下列可能不存在{columns_to_keep}\n") + continue # 从文件名提取设备ID # 添加新列 "device_id" df['device_id'] = self.getDeviceId(f) @@ -106,7 +109,7 @@ class ViewInfoManager: 'TimeToCsvAnalyzer': {'columns_to_group': ["Op Name", "Input Shapes", "Input Data Types", "Output Shapes"], 'extend_attr_to_group': ["device_id", "node_id"], - "columns_to_view": ["aicore_time(us)"], + "columns_to_view": ["Task Duration(us)"], 'calculate_fun': ['mean', 'var', 'max', 'min'] }, 'StatisticalInfoToHtmlAnalyzer': @@ -270,4 +273,4 @@ def main(): if __name__ == "__main__": main() - + -- Gitee