Rate this Page
★ ★ ★ ★ ★

torch.xpu.is_bf16_supported#

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

Return a bool indicating if the current XPU device supports dtype bfloat16.

Return type:

bool

On this page
Show Source
PyTorch Libraries