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

Method zero_use_cpu_optimizer

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

Source from the content-addressed store, hash-verified

1183 return self._config.zero_config.offload_param
1184
1185 def zero_use_cpu_optimizer(self):
1186 if self._config.zero_config.offload_optimizer is not None:
1187 return self._config.zero_config.offload_optimizer.device in [OffloadDeviceEnum.cpu, OffloadDeviceEnum.nvme]
1188 return False
1189
1190 def zero_cpu_offload(self):
1191 if self._config.zero_config.offload_optimizer is not None:

Callers 4

_configure_optimizerMethod · 0.95
init_z3Function · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected