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

Method bfloat16_enabled

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

Source from the content-addressed store, hash-verified

1298 return self._config.float16_config.enabled
1299
1300 def bfloat16_enabled(self):
1301 return self._config.bfloat16_config.enabled
1302
1303 def fp16_master_weights_and_gradients(self):
1304 return self._config.float16_config.fp16_master_weights_and_grads

Callers 15

__init__Method · 0.95
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_zero_optimizerMethod · 0.95
_take_model_stepMethod · 0.95
_get_zero_ckpt_nameMethod · 0.95
_load_checkpointMethod · 0.95
_get_all_zero_checkpointsMethod · 0.95

Calls

no outgoing calls

Tested by 2

engine_input_dtypeFunction · 0.64
test_grad_clip_and_norm_updateMethod · 0.64