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__(
91
91
)
92
92
self ._start_cpu_time_ns = 0
93
93
self ._start_wall_time_ns = 0
94
+ self ._start_epoch_time_usec = 0
94
95
95
96
def __enter__ (self ) -> "log_event" :
96
97
self ._start_cpu_time_ns = time .process_time_ns ()
97
98
self ._start_wall_time_ns = time .perf_counter_ns ()
99
+ self ._start_epoch_time_usec = int (time .time () * 1000 )
98
100
return self
99
101
100
102
def __exit__ (
Original file line number Diff line number Diff line change @@ -46,6 +46,7 @@ class TorchxEvent:
46
46
source : SourceType = SourceType .UNKNOWN
47
47
cpu_time_usec : Optional [int ] = None
48
48
wall_time_usec : Optional [int ] = None
49
+ start_epoch_time_usec : Optional [int ] = None
49
50
50
51
def __str__ (self ) -> str :
51
52
return self .serialize ()
You can’t perform that action at this time.
0 commit comments