File tree Expand file tree Collapse file tree 2 files changed +3
-0
lines changed Expand file tree Collapse file tree 2 files changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -91,10 +91,12 @@ def __init__(
9191 )
9292 self ._start_cpu_time_ns = 0
9393 self ._start_wall_time_ns = 0
94+ self ._start_epoch_time_usec = 0
9495
9596 def __enter__ (self ) -> "log_event" :
9697 self ._start_cpu_time_ns = time .process_time_ns ()
9798 self ._start_wall_time_ns = time .perf_counter_ns ()
99+ self ._start_epoch_time_usec = int (time .time () * 1000 )
98100 return self
99101
100102 def __exit__ (
Original file line number Diff line number Diff line change @@ -46,6 +46,7 @@ class TorchxEvent:
4646 source : SourceType = SourceType .UNKNOWN
4747 cpu_time_usec : Optional [int ] = None
4848 wall_time_usec : Optional [int ] = None
49+ start_epoch_time_usec : Optional [int ] = None
4950
5051 def __str__ (self ) -> str :
5152 return self .serialize ()
You can’t perform that action at this time.
0 commit comments