Rate this Page

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