Exporting start_time in InstructionEvent to Inspector (#10295)

Differential Revision: D72740782

Pull Request resolved: #10344

Author: Gasoonjia

devtools/inspector/_inspector.py

@@ -334,6 +334,7 @@ class Event:
     _delegate_time_scale_converter: Optional[
         Callable[[Union[int, str], Union[int, float]], Union[int, float]]
     ] = None
+    _start_time: Optional[List[Union[int, float]]] = None
 
     @cached_property
     def delegate_debug_metadatas(self) -> Union[List[str], Dict[str, Any]]:
@@ -352,6 +353,13 @@ def raw_delegate_debug_metadatas(self) -> List[str]:
         """
         return self._delegate_debug_metadatas
 
+    @property
+    def start_time(self) -> Optional[List[Union[int, float]]]:
+        """
+        Returns the start time of the event.
+        """
+        return self._start_time
+
     def to_dataframe(self, _units="") -> pd.DataFrame:
         """
         Convert the Event into a pandas DataFrame
@@ -402,6 +410,7 @@ def truncated_list(long_list: List[str]) -> str:
             "is_delegated_op": self.is_delegated_op,
             "delegate_backend_name": self.delegate_backend_name,
             "debug_data": [self.debug_data],
+            "start_time": [self._start_time],
         }
 
     @staticmethod
@@ -530,6 +539,7 @@ def _populate_profiling_related_fields(
 
         # Fill out fields from profile event
         data = []
+        stime = []
         delegate_debug_metadatas = []
         for event in events:
             if (profile_events := event.profile_events) is not None:
@@ -571,6 +581,7 @@ def _populate_profiling_related_fields(
                     )
 
                 data.append(scaled_time)
+                stime.append(profile_event.start_time)
                 delegate_debug_metadatas.append(
                     profile_event.delegate_debug_metadata
                     if profile_event.delegate_debug_metadata
@@ -583,6 +594,10 @@ def _populate_profiling_related_fields(
         if any(delegate_debug_metadatas):
             ret_event._delegate_debug_metadatas = delegate_debug_metadatas
 
+        # add _start_time to the event
+        if len(stime) > 0:
+            ret_event._start_time = stime
+
     @staticmethod
     def _populate_debugging_related_fields(
         ret_event: "Event",

devtools/inspector/tests/event_blocks_test.py

@@ -518,6 +518,7 @@ def _test_profile_event_generation(
                     delegate_debug_metadatas if is_delegated else []
                 ),
                 _instruction_id=event_signature.instruction_id,
+                _start_time=[0, 0, 0],
             )
             self.assertEqual(event, expected_event)