torch.set_float32_matmul_precision
torch.set_float32_matmul_precision(precision)
Sets the internal precision of float32 matrix multiplications.
Running float32 matrix multiplications in lower precision may
   significantly increase performance, and in some programs the loss
   of precision has a negligible impact.
Supports three settings:
  * "highest", float32 matrix multiplications use the float32
    datatype for internal computations.

  * "high", float32 matrix multiplications use the TensorFloat32
    or bfloat16_3x datatypes for internal computations, if fast
    matrix multiplication algorithms using those datatypes
    internally are available. Otherwise float32 matrix
    multiplications are computed as if the precision is "highest".

  * "medium", float32 matrix multiplications use the bfloat16
    datatype for internal computations, if a fast matrix
