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

Method memory_breakdown

deepspeed/runtime/engine.py:1093–1094  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

1091 return self._config.flops_profiler_config.output_file
1092
1093 def memory_breakdown(self):
1094 return self._config.memory_breakdown
1095
1096 def autotuning_enabled(self):
1097 return self._config.autotuning_config.enabled

Callers 6

__init__Method · 0.95
_forward_prologueMethod · 0.95
_forward_epilogueMethod · 0.95
_backward_prologueMethod · 0.95
_backward_epilogueMethod · 0.95
stepMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected