torch.cuda.is_tf32_supported

torch.cuda.is_tf32_supported() [source]

Return a bool indicating if the current CUDA/ROCm device supports dtype tf32.

Return type

bool