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

Method _stop_timers

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

Source from the content-addressed store, hash-verified

3487 self.timers(name).start()
3488
3489 def _stop_timers(self, timer_names):
3490 record = self.is_gradient_accumulation_boundary() and \
3491 self.flops_profiler_enabled() and \
3492 (self.global_steps >= self.flops_profiler_profile_step())
3493 for name in timer_names:
3494 self.timers(name).stop(record=record)
3495
3496 def _update_wall_clock_timers(self):
3497 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