From 4980e5b46d1ce66ae8ed2cc4ccc69264496ff585 Mon Sep 17 00:00:00 2001
From: zhouxianqi <13165993773@163.com>
Date: Tue, 23 Jul 2024 15:35:54 +0800
Subject: [PATCH] update_overall_interface

---
 .../disaggregate/overall_perf_interface.py    | 28 ++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/profiler/compare_tools/compare_backend/disaggregate/overall_perf_interface.py b/profiler/compare_tools/compare_backend/disaggregate/overall_perf_interface.py
index c89e8451930..7bac2b03353 100644
--- a/profiler/compare_tools/compare_backend/disaggregate/overall_perf_interface.py
+++ b/profiler/compare_tools/compare_backend/disaggregate/overall_perf_interface.py
@@ -31,4 +31,30 @@ class OverallPerfInterface:
 
     def _generate_result(self):
         overall_data = self._profiling_data.overall_metrics
-        self._result_data = getattr(overall_data, "__dict__", {})
+
+        self._result_data = {
+            "profiling_type": overall_data.profiling_type,
+            "minimal_profiling": overall_data.minimal_profiling,
+            "overall": {"e2e_time_ms": overall_data.e2e_time_ms,
+                        "computing_time_ms": overall_data.compute_time_ms,
+                        "uncovered_communication_time_ms": overall_data.communication_not_overlapped_ms,
+                        "free_time_ms": overall_data.free_time_ms},
+            "computing_time_disaggregate": {"fa_time_ms": overall_data.fa_total_time,
+                                            "conv_time_ms": overall_data.conv_total_time,
+                                            "matmul_time_ms": overall_data.mm_total_time,
+                                            "page_attention_time_ms": overall_data.page_attention_time,
+                                            "vector_time_ms": overall_data.vector_total_time,
+                                            "tensor_move_time_ms": overall_data.sdma_time_tensor_move,
+                                            "other_cube_time_ms": overall_data.other_cube_time},
+            "computing_num_disaggregate": {"fa_num": overall_data.fa_total_num,
+                                           "conv_num": overall_data.conv_total_num,
+                                           "matmul_num": overall_data.mm_total_num,
+                                           "page_attention_num": overall_data.page_attention_num,
+                                           "vector_num": overall_data.vector_total_num,
+                                           "tensor_move_num": overall_data.sdma_num_tensor_move,
+                                           "other_cube_num": overall_data.other_cube_num},
+            "communication_time_disaggregate": {"wait_time_ms": overall_data.wait_time_ms,
+                                                "transmit_time_ms": overall_data.transmit_time_ms},
+            "free_time_disaggregate": {"sdma_time_ms": overall_data.sdma_time_stream,
+                                       "free_ms": overall_data.free_time_ms - overall_data.sdma_time_stream}
+        }
-- 
Gitee