Simplify the parse of nsys log.

Author: Xreki

api/common/launch.py

@@ -190,47 +190,38 @@ def _parse_logs(self, logs):
                         break
 
         parse_status = False
-        kernel_gpu_time = 0.0
+        time_unit = TimeUnit()
+
         if kernel_line_from is not None and kernel_line_to is not None:
             for i in range(kernel_line_from, kernel_line_to):
                 print(logs[i])
+                if i >= kernel_line_from + 4:
+                    gpu_time, percent, function = self._parse_line(logs[i])
+                    time_unit.add_info(gpu_time, function)
             print("")
             parse_status = True
-            kernel_gpu_time = self._parse_gpu_time(logs[kernel_line_from + 4])
 
-        memcpy_gpu_time = 0.0
         if memcpy_line_from is not None and memcpy_line_to is not None:
             for i in range(memcpy_line_from, memcpy_line_to):
                 print(logs[i])
+                if i >= memcpy_line_from + 4:
+                    gpu_time, percent, function = self._parse_line(logs[i])
+                    time_unit.add_info(gpu_time, function)
             print("")
             parse_status = True
-            memcpy_gpu_time = self._parse_gpu_time(logs[memcpy_line_from + 4])
-
-        total_gpu_time = kernel_gpu_time + memcpy_gpu_time
-        if total_gpu_time != 0.0:
-            print(
-                "total gpu_time: {:.4f} ms (kernel: {:.4f} ms ({:.2f}%); memcpy: {:.4f} ms ({:.2f}%))".
-                format(total_gpu_time, kernel_gpu_time, kernel_gpu_time * 100 /
-                       total_gpu_time, memcpy_gpu_time, memcpy_gpu_time * 100 /
-                       total_gpu_time))
-        else:
-            print(
-                "total gpu_time: {:.4f} ms (kernel: {:.4f} ms; memcpy: {:.4f} ms)".
-                format(total_gpu_time, kernel_gpu_time, memcpy_gpu_time))
-        print("")
-        return parse_status, total_gpu_time
 
-    def _parse_gpu_time(self, line):
+        print(time_unit)
+        return parse_status, time_unit.total()
+
+    def _parse_line(self, line):
         infos = line.strip().split()
         percent = float(infos[0].replace("%", "")) * 0.01
         gpu_time = float(infos[1].replace(",", "")) * 1E-6
         calls = int(infos[2].replace(",", ""))
         function = infos[7]
         for i in range(8, len(infos)):
             function = function + " " + infos[i]
-        #print("percent: %.2f; gpu_time: %.4f ms; calls: %d; function: %s" %
-        #      (percent, gpu_time, calls, function))
-        return gpu_time / percent
+        return gpu_time, percent, function
 
 
 def launch(benchmark_script,