From 806dd7dd6d329ab1e7da30614c1f374a528a486f Mon Sep 17 00:00:00 2001
From: sunboquan <sunboquan@huawei.com>
Date: Sat, 9 Dec 2023 11:54:25 +0800
Subject: [PATCH] bugfix

---
 profiler/compare_tools/profiling_analysis/npu_parser.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/profiler/compare_tools/profiling_analysis/npu_parser.py b/profiler/compare_tools/profiling_analysis/npu_parser.py
index e17a1f605e..47634389a1 100644
--- a/profiler/compare_tools/profiling_analysis/npu_parser.py
+++ b/profiler/compare_tools/profiling_analysis/npu_parser.py
@@ -209,7 +209,7 @@ class NpuProfilingParser:
             name = info.loc[i, 'Name']
             aiv_vec_time = info.loc[i, 'aiv_vec_time(us)'] if info.get('aiv_vec_time(us)') is not None else None
             mac_time = info.loc[i, 'mac_time(us)'] if info.get('mac_time(us)') is not None else None
-            if (aiv_vec_time is None or mac_time is None) and pd.isna(op_type):
+            if pd.isna(aiv_vec_time) and pd.isna(mac_time):
                 continue
             task_durations = info.loc[i, 'Duration(us)']
             if self.FLASH_ATTENTION in op_type.lower():
-- 
Gitee