MCPcopy Create free account
hub / github.com/AIS-SNU/Smart-Infinity / zero_use_cpu_optimizer

Method zero_use_cpu_optimizer

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

Source from the content-addressed store, hash-verified

687 return self._config.zero_config.offload_param
688
689 def zero_use_cpu_optimizer(self):
690 if self._config.zero_config.offload_optimizer is not None:
691 return self._config.zero_config.offload_optimizer.device in [OffloadDeviceEnum.cpu, OffloadDeviceEnum.nvme]
692 return False
693
694 def zero_cpu_offload(self):
695 if self._config.zero_config.offload_optimizer is not None:

Callers 2

_configure_optimizerMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected