Method bfloat16_enabled

deepspeed/runtime/engine.py:759–760 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

757	return self._config.fp16_enabled
758
759	def bfloat16_enabled(self):
760	return self._config.bfloat16_enabled
761
762	def fp16_master_weights_and_gradients(self):
763	return self._config.fp16_master_weights_and_gradients

__init__Method · 0.95

communication_data_typeMethod · 0.95

get_data_typesMethod · 0.95

_configure_checkpointingMethod · 0.95

_configure_distributed_modelMethod · 0.95

allreduce_gradientsMethod · 0.95

backwardMethod · 0.95

_take_model_stepMethod · 0.95

_get_zero_ckpt_nameMethod · 0.95

load_checkpointMethod · 0.95

_load_checkpointMethod · 0.95

_get_all_zero_checkpointsMethod · 0.95

no outgoing calls

no test coverage detected