get_precision_context#

composer.core.get_precision_context(precision, precision_config=None)[source]#

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.