MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / stop

Method stop

deepspeed/utils/timer.py:71–84  ·  view source on GitHub ↗

Stop the timer.

(self, reset=False, record=False)

Source from the content-addressed store, hash-verified

69 self.started_ = True
70
71 def stop(self, reset=False, record=False):
72 """Stop the timer."""
73 assert self.started_, "timer is not started"
74 event_class = get_accelerator().Event
75 if self.use_host_timer:
76 self.end_time = time.time()
77 self.event_timers.append(self.end_time - self.start_time)
78 else:
79 event_class = get_accelerator().Event
80 end_event = event_class(enable_timing=True)
81 end_event.record()
82 self.event_timers.append(CudaEventTimer(self.start_event, end_event))
83 self.start_event = None
84 self.started_ = False
85
86 def _get_elapsed_msec(self):
87 if self.use_host_timer:

Callers

nothing calls this directly

Calls 4

get_acceleratorFunction · 0.90
CudaEventTimerClass · 0.85
appendMethod · 0.80
recordMethod · 0.45

Tested by

no test coverage detected