From d611ffdb0d79216bec17fd239d8e121a9e26e9ab Mon Sep 17 00:00:00 2001
From: zhouxianqi <13165993773@163.com>
Date: Wed, 6 Sep 2023 19:13:09 +0800
Subject: [PATCH] bug_fix_kernel_lost

---
 profiler/compare_tools/utils/profiling_parser.py | 8 ++++----
 profiler/compare_tools/utils/torch_op_node.py    | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/profiler/compare_tools/utils/profiling_parser.py b/profiler/compare_tools/utils/profiling_parser.py
index 88a02b4dfa..0bc7c4985f 100644
--- a/profiler/compare_tools/utils/profiling_parser.py
+++ b/profiler/compare_tools/utils/profiling_parser.py
@@ -1,7 +1,7 @@
 from abc import ABCMeta, abstractmethod
 from math import ceil
 
-from utils.compare_event import KernelEvent, MemoryEvent
+from utils.compare_event import KernelEvent
 from utils.constant import Constant
 from utils.file_reader import FileReader
 from utils.trace_event_data import TraceEventData
@@ -75,7 +75,7 @@ class GPUProfilingParser(ProfilingParser):
         json_data = FileReader.read_trace_file(self._json_path)
         total_events = json_data.get("traceEvents", [])
         for event in total_events:
-            if event.get("cat") == "cpu_op":
+            if event.get("cat").lower() == "cpu_op":
                 torch_op_list.append(event)
         self._torch_op_data = torch_op_list
 
@@ -91,7 +91,7 @@ class GPUProfilingParser(ProfilingParser):
                 flow_start_dict[event.get("id")] = event
             elif event.get("cat") in flow_cat and event.get("ph") == "f":
                 flow_end_dict[event.get("id")] = event
-            elif event.get("cat", "").lower() == "kernel":
+            elif event.get("cat", "").lower() == "kernel" and event.get("name", "").split("_")[0].lower() != "ncclkernel":
                 kernel_dict["{}-{}-{}".format(event.get("pid"), event.get("tid"), event.get("ts"))] = event
 
         for flow_id, start_flow in flow_start_dict.items():
@@ -111,7 +111,7 @@ class GPUProfilingParser(ProfilingParser):
         json_data = FileReader.read_trace_file(self._json_path)
         total_events = json_data.get("traceEvents", [])
         for event in total_events:
-            if event.get("name", "") == "[memory]":
+            if event.get("name", "").lower() == "[memory]":
                 memory_events.append(event)
         memory_events.sort(key=lambda x: x.get("ts", 0))
         addr_dict = {}
diff --git a/profiler/compare_tools/utils/torch_op_node.py b/profiler/compare_tools/utils/torch_op_node.py
index 8995dafc1e..2554873ec8 100644
--- a/profiler/compare_tools/utils/torch_op_node.py
+++ b/profiler/compare_tools/utils/torch_op_node.py
@@ -65,7 +65,7 @@ class TorchOpNode:
         self._child_nodes.append(child_node)
 
     def set_kernel_list(self, kernel_list: list):
-        self._kernel_list = kernel_list
+        self._kernel_list.extend(kernel_list)
 
     def add_kernel_num(self, kernel_num: int):
         self._kernel_num += kernel_num
-- 
Gitee