add start epoch time to TorchXEvent

Differential Revision: D56037935 Pull Request resolved: #882
pytorch · Apr 15, 2024 · a38a1f8 · a38a1f8
1 parent 8a18b70
commit a38a1f8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/torchx/runner/events/__init__.py b/torchx/runner/events/__init__.py
@@ -91,10 +91,12 @@ def __init__(
         )
         self._start_cpu_time_ns = 0
         self._start_wall_time_ns = 0
+        self._start_epoch_time_usec = 0
 
     def __enter__(self) -> "log_event":
         self._start_cpu_time_ns = time.process_time_ns()
         self._start_wall_time_ns = time.perf_counter_ns()
+        self._start_epoch_time_usec = int(time.time() * 1000)
         return self
 
     def __exit__(

diff --git a/torchx/runner/events/api.py b/torchx/runner/events/api.py
@@ -46,6 +46,7 @@ class TorchxEvent:
     source: SourceType = SourceType.UNKNOWN
     cpu_time_usec: Optional[int] = None
     wall_time_usec: Optional[int] = None
+    start_epoch_time_usec: Optional[int] = None
 
     def __str__(self) -> str:
         return self.serialize()