Rate this Page
★ ★ ★ ★ ★

torch.cuda.is_bf16_supported#

torch.cuda.is_bf16_supported(including_emulation=True)[source]#

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

On this page
Show Source
PyTorch Libraries