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

Method _stop_timers

deepspeed/runtime/engine.py:3364–3369  ·  view source on GitHub ↗
(self, timer_names)

Source from the content-addressed store, hash-verified

3362 self.timers(name).start()
3363
3364 def _stop_timers(self, timer_names):
3365 record = self.is_gradient_accumulation_boundary() and \
3366 self.flops_profiler_enabled() and \
3367 (self.global_steps >= self.flops_profiler_profile_step())
3368 for name in timer_names:
3369 self.timers(name).stop(record=record)
3370
3371 def _update_wall_clock_timers(self):
3372 self.engine_timers_cache = {}

Callers 3

_forward_epilogueMethod · 0.95
_backward_epilogueMethod · 0.95
stepMethod · 0.95

Calls 4

stopMethod · 0.45

Tested by

no test coverage detected