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

Method zero_use_cpu_optimizer

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

Source from the content-addressed store, hash-verified

1129 return self._config.zero_config.offload_param
1130
1131 def zero_use_cpu_optimizer(self):
1132 if self._config.zero_config.offload_optimizer is not None:
1133 return self._config.zero_config.offload_optimizer.device in [OffloadDeviceEnum.cpu, OffloadDeviceEnum.nvme]
1134 return False
1135
1136 def zero_cpu_offload(self):
1137 if self._config.zero_config.offload_optimizer is not None:

Callers 3

_configure_optimizerMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected