From b872c0d1d266c2592cc5b7927acfbd432186d2b8 Mon Sep 17 00:00:00 2001 From: fanxiaotong Date: Tue, 24 Oct 2023 19:41:43 +0800 Subject: [PATCH] fix bug --- .../cluster_kernels_analysis/cluster_op_summary_analysis.py | 4 ++-- 1 file changed, 2 insertions(+), 2 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 68232b24e..9eccd54f7 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 @@ -62,7 +62,7 @@ class FormDataProcessor: file = self.files[0] df = pd.read_csv(file) if 'aiv_time(us)' in df.columns: - return "ASCEND_910B" + return "ASCEND_NEW" return "ASCEND_OTHER" def getDeviceId(self, dir_path): @@ -87,7 +87,7 @@ class ViewInfoManager: def setOpSummaryColumnsParams(self): # 有些数据除了用表格的列进行分组之外,还添加了其他属性对数据进行分类,这部分数据放在extend_attr_to_group里面 self.op_summary_columns_dict = { - 'ASCEND_910B': { + 'ASCEND_NEW': { 'TimeToCsvAnalyzer': {'columns_to_group': ["Op Name", "Input Shapes", "Input Data Types", "Output Shapes"], 'extend_attr_to_group': ["device_id", "node_id"], -- Gitee