torchtnt.utils.device.maybe_enable_tf32

torchtnt.utils.device.maybe_enable_tf32(precision: str = 'high') → None

Conditionally sets the precision of float32 matrix multiplications and conv operations.

For more information, see the PyTorch docs

Parameters:precision – The setting to determine which datatypes to use for matrix multiplication.