Method fp16_enabled

deepspeed/runtime/engine.py:1243–1244 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

1241	return self._config.graph_harvesting
1242
1243	def fp16_enabled(self):
1244	return self._config.float16_config.enabled
1245
1246	def bfloat16_enabled(self):
1247	return self._config.bfloat16_config.enabled

communication_data_typeMethod · 0.95

get_data_typesMethod · 0.95

_assert_valid_mixed_precision_configMethod · 0.95

_mixed_precision_dtypesMethod · 0.95

_do_sanity_checkMethod · 0.95

_configure_distributed_modelMethod · 0.95

_configure_basic_optimizerMethod · 0.95

_configure_quantizationMethod · 0.95

_configure_zero_optimizerMethod · 0.95

_forward_prologueMethod · 0.95

_take_model_stepMethod · 0.95

stepMethod · 0.95

no outgoing calls

no test coverage detected