get_precision_context

composer.core.get_precision_context(precision, precision_config=None, fp8_autocast_enabled=True)

Returns a context manager to automatically cast to a specific precision.

Parameters

• precision (str | Precision) – Precision for the context

• precision_config (Optional[dict[str, Any]]) – Config for FP8 scaling strategy. See parameters for DelayedScaling.

• fp8_autocast_enabled (bool) – Whether to enable FP8 autocast. Defaults to True.